Module: Elasticsearch::Model::Adapter::Multiple::Records

Defined in:
lib/elasticsearch/model/adapters/multiple.rb

Instance Method Summary collapse

Instance Method Details

#__adapter_for_klass(klass) ⇒ Object

This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.

Returns the adapter registered for a particular ‘klass` or `nil` if not available



105
106
107
 
# File 'lib/elasticsearch/model/adapters/multiple.rb', line 105

def __adapter_for_klass(klass)
  Adapter.adapters.select { |name, checker| checker.call(klass) }.keys.first
end

#__ids_by_type ⇒ Object

This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.

Returns the record IDs grouped by class based on type ‘_type`

Example:

{ Foo => ["1"], Bar => ["1", "5"] }



74
75
76
77
78
79
80
81
82
83
 
# File 'lib/elasticsearch/model/adapters/multiple.rb', line 74

def __ids_by_type
  ids_by_type = {}

  response.response["hits"]["hits"].each do |hit|
    type = __type_for_hit(hit)
    ids_by_type[type] ||= []
    ids_by_type[type] << hit[:_id]
  end
  ids_by_type
end

#__records_by_type ⇒ Object

This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.

 Returns the collection of records grouped by class based on ‘_type`

Example:

Foo  => {"1"=> #<Foo id: 1, title: "ABC", ...},
Bar  => #<Bar id: 1, name: "XYZ", ...}

}



39
40
41
42
43
44
45
46
47
 
# File 'lib/elasticsearch/model/adapters/multiple.rb', line 39

def __records_by_type
  result = __ids_by_type.map do |klass, ids|
    records = __records_for_klass(klass, ids)
    ids     = records.map(&:id).map(&:to_s)
    [ klass, Hash[ids.zip(records)] ]
  end

  Hash[result]
end

#__records_for_klass(klass, ids) ⇒ Object

This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.

Returns the collection of records for a specific type based on passed ‘klass`



53
54
55
56
57
58
59
60
61
62
63
64
 
# File 'lib/elasticsearch/model/adapters/multiple.rb', line 53

def __records_for_klass(klass, ids)
  adapter = __adapter_for_klass(klass)

  case
    when Elasticsearch::Model::Adapter::ActiveRecord.equal?(adapter)
      klass.where(klass.primary_key => ids)
    when Elasticsearch::Model::Adapter::Mongoid.equal?(adapter)
      klass.where(:id.in => ids)
    else
      klass.find(ids)
  end
end

#__type_for_hit(hit) ⇒ Object

This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.

Returns the class of the model corresponding to a specific ‘hit` in Elasticsearch results

See Also:



91
92
93
94
95
96
97
98
99
 
# File 'lib/elasticsearch/model/adapters/multiple.rb', line 91

def __type_for_hit(hit)
  @@__types ||= {}

  @@__types[ "#{hit[:_index]}::#{hit[:_type]}" ] ||= begin
    Registry.all.detect do |model|
      model.index_name == hit[:_index] && model.document_type == hit[:_type]
    end
  end
end

#records ⇒ Object

Note:

The order of results in the Elasticsearch response is preserved

Returns a collection of model instances, possibly of different classes (ActiveRecord, Mongoid, …)



18
19
20
21
22
23
24
25
26
 
# File 'lib/elasticsearch/model/adapters/multiple.rb', line 18

def records
  records_by_type = __records_by_type

  records = response.response["hits"]["hits"].map do |hit|
    records_by_type[ __type_for_hit(hit) ][ hit[:_id] ]
  end

  records.compact
end