Module: AwesomeCounterCache::InstanceMethods

Defined in:
lib/awesome_counter_cache/instance_methods.rb

Instance Method Summary collapse

Instance Method Details

#awesome_counter_cache_after_commit_on_create(counter_cache, model) ⇒ Object 



2
3
4
5
6
7
8
9
 
# File 'lib/awesome_counter_cache/instance_methods.rb', line 2

def awesome_counter_cache_after_commit_on_create(counter_cache, model)
  id = counter_cache.id
  state = @awesome_counter_cache_data.fetch(id)
  state.delta_current = counter_cache.delta_magnitude.call(model)
  model.create_awesome_counter_cache_for(counter_cache.relation_name, @awesome_counter_cache_data.fetch(id), counter_cache)
  state.delta_original = state.delta_current
  state.delta_current = nil
end

#awesome_counter_cache_after_commit_on_destroy(counter_cache, model) ⇒ Object 



11
12
13
14
15
16
17
18
 
# File 'lib/awesome_counter_cache/instance_methods.rb', line 11

def awesome_counter_cache_after_commit_on_destroy(counter_cache, model)
  id = counter_cache.id
  state = @awesome_counter_cache_data.fetch(id)
  state.delta_current = counter_cache.delta_magnitude.call(model)
  model.destroy_awesome_counter_cache_for(counter_cache.relation_name, @awesome_counter_cache_data.fetch(id), counter_cache)
  state.delta_original = state.delta_current
  state.delta_current = nil
end

#awesome_counter_cache_after_commit_on_update(counter_cache, model) ⇒ Object 



20
21
22
23
24
25
26
27
 
# File 'lib/awesome_counter_cache/instance_methods.rb', line 20

def awesome_counter_cache_after_commit_on_update(counter_cache, model)
  id = counter_cache.id
  state = @awesome_counter_cache_data.fetch(id)
  state.delta_current = counter_cache.delta_magnitude.call(model)
  model.update_awesome_counter_cache_for(counter_cache.relation_name, @awesome_counter_cache_data.fetch(id), counter_cache)
  state.delta_original = state.delta_current
  state.delta_current = nil
end

#awesome_counter_cache_after_initialize(counter_cache, model) ⇒ Object 



29
30
31
32
33
34
35
36
37
38
39
40
41
42
 
# File 'lib/awesome_counter_cache/instance_methods.rb', line 29

def awesome_counter_cache_after_initialize(counter_cache, model)
  id = counter_cache.id

  @awesome_counter_cache_data ||= {}
  @awesome_counter_cache_data[id] ||= AwesomeCounterCache::State.new

  primary_key_column = self.class.primary_key

  if read_attribute(primary_key_column).nil? # `new_record?` always returns false so check if primary key is nil instead - kaspernj
    @awesome_counter_cache_data.fetch(id).delta_original ||= 0
  else
    @awesome_counter_cache_data.fetch(id).delta_original ||= counter_cache.delta_magnitude.call(model)
  end
end

#awesome_counter_cache_reset_original_values ⇒ Object 



62
63
64
65
66
67
 
# File 'lib/awesome_counter_cache/instance_methods.rb', line 62

def awesome_counter_cache_reset_original_values
  self.class.awesome_counter_caches.each do |id, counter_cache|
    state = @awesome_counter_cache_data.fetch(id)
    state.delta_original = counter_cache.delta_magnitude.call(self)
  end
end

#create_awesome_counter_cache_for(relation_name, state, counter_cache) ⇒ Object 



44
45
46
47
48
49
50
51
52
53
54
 
# File 'lib/awesome_counter_cache/instance_methods.rb', line 44

def create_awesome_counter_cache_for(relation_name, state, counter_cache)
  relation_model = __send__(relation_name)
  return if relation_model.blank?

  addition = state.delta_current

  if addition != 0
    primary_key_value = relation_model.read_attribute(relation_model.class.primary_key)
    relation_model.class.update_counters(primary_key_value, counter_cache.column_name => addition) # rubocop:disable Rails/SkipsModelValidations
  end
end

#destroy_awesome_counter_cache_for(relation_name, state, counter_cache) ⇒ Object 



69
70
71
72
73
74
75
76
77
78
79
 
# File 'lib/awesome_counter_cache/instance_methods.rb', line 69

def destroy_awesome_counter_cache_for(relation_name, state, counter_cache)
  relation_model = __send__(relation_name)
  return if relation_model.blank?

  addition = -state.delta_original

  if addition != 0
    primary_key_value = relation_model.read_attribute(relation_model.class.primary_key)
    relation_model.class.update_counters(primary_key_value, counter_cache.column_name => addition) # rubocop:disable Rails/SkipsModelValidations
  end
end

#reload(*args, &blk) ⇒ Object 



56
57
58
59
60
 
# File 'lib/awesome_counter_cache/instance_methods.rb', line 56

def reload(*args, &blk)
  result = super
  awesome_counter_cache_reset_original_values
  result
end

#update_awesome_counter_cache_for(relation_name, state, counter_cache) ⇒ Object

rubocop:disable Metrics/AbcSize



81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
 
# File 'lib/awesome_counter_cache/instance_methods.rb', line 81

def update_awesome_counter_cache_for(relation_name, state, counter_cache) # rubocop:disable Metrics/AbcSize
  addition = state.delta_current - state.delta_original
  relation_model_class = self.class.reflections.fetch(relation_name.to_s).klass
  primary_key_value = read_attribute(counter_cache.relation_foreign_key)

  if saved_changes.key?(counter_cache.relation_foreign_key)
    # Record change from one to another - reduce and increase based on previously recorded values
    old_record_id = saved_changes.fetch(counter_cache.relation_foreign_key).first

    if old_record_id && !state.delta_original.zero?
      # Reduce the count of the previous record
      relation_model_class.update_counters(old_record_id, counter_cache.column_name => -state.delta_original) # rubocop:disable Rails/SkipsModelValidations
    end

    unless state.delta_current.zero?
      # Increase the count of the new record
      relation_model_class.update_counters(primary_key_value, counter_cache.column_name => state.delta_current) # rubocop:disable Rails/SkipsModelValidations
    end
  elsif !addition.zero?
    # Add to the current record
    relation_model_class.update_counters(primary_key_value, counter_cache.column_name => addition) # rubocop:disable Rails/SkipsModelValidations
  end
end