Module: FullSearch::Callbacks

Extended by:
Quoting
Defined in:
lib/full_search/callbacks.rb

Class Method Summary collapse

Class Method Details

.associated_class(model, association_name) ⇒ Object 



94
95
96
97
 
# File 'lib/full_search/callbacks.rb', line 94

def self.associated_class(model, association_name)
  reflection = model.reflect_on_association(association_name.to_sym)
  reflection&.klass
end

.association_key(model, association_name) ⇒ Object 



99
100
101
102
 
# File 'lib/full_search/callbacks.rb', line 99

def self.association_key(model, association_name)
  reflection = model.reflect_on_association(association_name.to_sym)
  reflection&.foreign_key&.to_s || "#{association_name}_id"
end

.connection ⇒ Object 



90
91
92
 
# File 'lib/full_search/callbacks.rb', line 90

def self.connection
  ActiveRecord::Base.connection
end

.install!(model) ⇒ Object 



5
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
 
# File 'lib/full_search/callbacks.rb', line 5

def self.install!(model)
  return if model.instance_variable_get(:@__full_search_callbacks_installed)

  dsl = model.full_search_dsl

  source_fields = dsl.fields.select(&:source)
  return if source_fields.empty?

  model.after_save do
    FullSearch::Callbacks.reindex_record!(self)
  end

  model.after_destroy do
    FullSearch::Callbacks.remove_record!(self)
  end

  dsl.fields.each do |field|
    next unless field.reindex_on

    assoc_class = associated_class(model, field.reindex_on)
    assoc_class&.after_save do |record|
      FullSearch::Callbacks.reindex_dependents!(record, model, field)
    end
    assoc_class&.after_destroy do |record|
      FullSearch::Callbacks.reindex_dependents!(record, model, field)
    end
  end

  model.instance_variable_set(:@__full_search_callbacks_installed, true)
end

.reindex_dependents!(parent_record, dependent_model, field) ⇒ Object 



70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
 
# File 'lib/full_search/callbacks.rb', line 70

def self.reindex_dependents!(parent_record, dependent_model, field)
  fk = association_key(dependent_model, field.reindex_on)
  conn = ActiveRecord::Base.connection
  sql = "SELECT id FROM #{qt(dependent_model.table_name)} WHERE #{qc(fk)} = #{q(parent_record.id)}"
  dependent_ids = conn.execute(sql).map { |r| r["id"] }

  dependent_ids.each do |dep_id|
    if field.async
      FullSearch::ReindexJob.perform_later(dependent_model.name, dep_id, field.name)
    else
      dependent = dependent_model.find_by(id: dep_id)
      reindex_field!(dependent, field.name) if dependent
    end
  end
end

.reindex_field!(record, field_name) ⇒ Object 



51
52
53
54
55
56
57
58
59
60
61
62
 
# File 'lib/full_search/callbacks.rb', line 51

def self.reindex_field!(record, field_name)
  dsl = record.class.full_search_dsl
  field = dsl.fields.find { |f| f.name == field_name }
  return unless field&.source

  value = record.instance_exec(&field.source)
  table = qt(FullSearch::Index.fts_table_name(record.class))
  conn = ActiveRecord::Base.connection
  conn.execute(
    "UPDATE #{table} SET #{qc(field.name)} = #{q(value.to_s)} WHERE rowid = #{q(record.id)}"
  )
end

.reindex_record!(record) ⇒ Object 



40
41
42
43
44
45
46
47
48
49
 
# File 'lib/full_search/callbacks.rb', line 40

def self.reindex_record!(record)
  dsl = record.class.full_search_dsl
  return unless dsl

  dsl.fields.each do |field|
    next unless field.source

    reindex_field!(record, field.name)
  end
end

.remove_record!(record) ⇒ Object 



64
65
66
67
68
 
# File 'lib/full_search/callbacks.rb', line 64

def self.remove_record!(record)
  table = qt(FullSearch::Index.fts_table_name(record.class))
  conn = ActiveRecord::Base.connection
  conn.execute("DELETE FROM #{table} WHERE rowid = #{q(record.id)}")
end

.reset_installed_flag!(model) ⇒ Object 



36
37
38
 
# File 'lib/full_search/callbacks.rb', line 36

def self.reset_installed_flag!(model)
  model.instance_variable_set(:@__full_search_callbacks_installed, false)
end