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 



107
108
109
110
 
# File 'lib/full_search/callbacks.rb', line 107

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 



112
113
114
115
 
# File 'lib/full_search/callbacks.rb', line 112

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 



103
104
105
 
# File 'lib/full_search/callbacks.rb', line 103

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
35
36
37
38
 
# 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_commit do
    next if FullSearch.bulk_importing?(self.class)
    FullSearch::Callbacks.reindex_record!(self)
  end

  model.after_destroy_commit do
    next if FullSearch.bulk_importing?(self.class)
    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_commit do |record|
      next if FullSearch.bulk_importing?(record.class)
      FullSearch::Callbacks.reindex_dependents!(record, model, field)
    end
    assoc_class&.after_destroy_commit do |record|
      next if FullSearch.bulk_importing?(record.class)
      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 



83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
 
# File 'lib/full_search/callbacks.rb', line 83

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 



60
61
62
63
64
65
66
67
68
69
70
71
72
73
 
# File 'lib/full_search/callbacks.rb', line 60

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 = FullSearch::Model.evaluate_source(record, field)
  table = qt(FullSearch::Index.fts_table_name(record.class))
  conn = ActiveRecord::Base.connection
  conn.execute(
    "UPDATE #{table} SET #{qc(field.as || field.name)} = #{q(value.to_s)} WHERE rowid = #{q(record.id)}"
  )
rescue => e
  raise unless e.message.include?("no such table")
end

.reindex_record!(record) ⇒ Object 



44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
 
# File 'lib/full_search/callbacks.rb', line 44

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

  model_name = record.class.name
  dsl.fields.each do |field|
    next unless field.source

    if model_name && field.async_source
      FullSearch::ReindexJob.perform_later(model_name, record.id, field.name)
    else
      reindex_field!(record, field.name)
    end
  end
end

.remove_record!(record) ⇒ Object 



75
76
77
78
79
80
81
 
# File 'lib/full_search/callbacks.rb', line 75

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)}")
rescue => e
  raise unless e.message.include?("no such table")
end

.reset_installed_flag!(model) ⇒ Object 



40
41
42
 
# File 'lib/full_search/callbacks.rb', line 40

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