touch_later.rb 1.6 KB
Newer Older
A
Arthur Neves 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18
module ActiveRecord
  # = Active Record Touch Later
  module TouchLater
    extend ActiveSupport::Concern

    included do
      before_commit_without_transaction_enrollment :touch_deferred_attributes
    end

    def touch_later(*names) # :nodoc:
      raise ActiveRecordError, "cannot touch on a new record object" unless persisted?

      @_defer_touch_attrs ||= timestamp_attributes_for_update_in_model
      @_defer_touch_attrs |= names
      @_touch_time = current_time_from_proper_timezone

      surreptitiously_touch @_defer_touch_attrs
      self.class.connection.add_transaction_record self
19 20 21 22 23 24 25

      # touch the parents as we are not calling the after_save callbacks
      self.class.reflect_on_all_associations(:belongs_to).each do |r|
        if touch = r.options[:touch]
          ActiveRecord::Associations::Builder::BelongsTo.touch_record(self, r.foreign_key, r.name, touch, :touch_later)
        end
      end
A
Arthur Neves 已提交
26 27 28 29 30 31 32 33 34 35
    end

    def touch(*names, time: nil) # :nodoc:
      if has_defer_touch_attrs?
        names |= @_defer_touch_attrs
      end
      super(*names, time: time)
    end

    private
36

A
Arthur Neves 已提交
37 38 39 40 41 42 43 44
      def surreptitiously_touch(attrs)
        attrs.each { |attr| write_attribute attr, @_touch_time }
        clear_attribute_changes attrs
      end

      def touch_deferred_attributes
        if has_defer_touch_attrs? && persisted?
          touch(*@_defer_touch_attrs, time: @_touch_time)
45
          @_defer_touch_attrs, @_touch_time = nil, nil
A
Arthur Neves 已提交
46 47 48 49 50 51 52
        end
      end

      def has_defer_touch_attrs?
        defined?(@_defer_touch_attrs) && @_defer_touch_attrs.present?
      end

53 54
      def belongs_to_touch_method
        :touch_later
A
Arthur Neves 已提交
55
      end
56

A
Arthur Neves 已提交
57 58
  end
end