Module: GuardAgainstPhysicalDelete::SupportCounterCache::Associations::Builder::BelongsTo

Defined in:
lib/guard_against_physical_delete/support_counter_cache/associations/builder/belongs_to.rb,
lib/guard_against_physical_delete/support_counter_cache/3.2_and_4.0/associations/builder/belongs_to.rb

Class Method Summary collapse

Class Method Details

.included(obj) ⇒ Object 



6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
 
# File 'lib/guard_against_physical_delete/support_counter_cache/associations/builder/belongs_to.rb', line 6

def self.included(obj)
  class << obj
    private
    def add_counter_cache_methods_with_logical_delete(mixin)
      add_counter_cache_methods_without_logical_delete mixin
      add_logical_delete_counter_cache_methods(mixin)
    end
    alias_method_chain :add_counter_cache_methods, :logical_delete

    def add_logical_delete_counter_cache_methods(mixin)
      mixin.class_eval do
        def belongs_to_counter_cache_after_logical_delete(reflection)
          cache_column = reflection.counter_cache_column
          delete_column = self.logical_delete_column
          return unless send("#{delete_column}_changed?")
          return unless send("#{delete_column}_was").nil?

          record = send(reflection.name)
          record.class.decrement_counter(cache_column, record.id) unless record.nil?
        end

        def belongs_to_counter_cache_after_revive(reflection)
          cache_column = reflection.counter_cache_column
          delete_column = self.logical_delete_column
          return unless send("#{delete_column}_changed?")
          return if send("#{delete_column}_was").nil?
          return unless send(delete_column).nil?
          record = send(reflection.name)
          record.class.increment_counter(cache_column, record.id) unless record.nil?
        end
      end
      mixin.redefine_method("belongs_to_counter_cache_after_update_for_#{name}") do
        # do nothing
      end if mixin.method_defined?("belongs_to_counter_cache_after_update_for_#{name}")
    end

    def add_counter_cache_callbacks_with_logical_delete(model, reflection)
      add_counter_cache_callbacks_without_logical_delete model, reflection

      return unless model.logical_delete?

      model.after_update lambda { |record|
        record.belongs_to_counter_cache_after_logical_delete(reflection)
        record.belongs_to_counter_cache_after_revive(reflection)
      }
    end
    alias_method_chain :add_counter_cache_callbacks, :logical_delete
  end
end