mongodb/mongoid

# frozen_string_literal: true
# rubocop:todo all

module Mongoid
  module Persistable

    # Defines behavior for $inc operations.
    module Incrementable
      extend ActiveSupport::Concern

      # Increment the provided fields by the corresponding values. Values can
      # be positive or negative, and if no value exists for the field it will
      # be set with the provided value.
      #
      # @example Increment the fields.
      #   document.inc(score: 10, place: 1, lives: -10)
      #
      # @param [ Hash ] increments The field/inc increment pairs.
      #
      # @return [ Document ] The document.
      def inc(increments)
        prepare_atomic_operation do |ops|
          process_atomic_operations(increments) do |field, value|
            increment = value.is_a?(BigDecimal) ? value.to_f : value
            current = attributes[field]
            new_value = (current || 0) + increment
            process_attribute field, new_value if executing_atomically?
            attributes[field] = new_value
            ops[atomic_attribute_name(field)] = increment
          end
          { "$inc" => ops } unless ops.empty?
        end
      end
    end
  end
end