index.rb
2.98 KB
1
2
3
4
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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
module Searchkick
class Index
attr_reader :name
def initialize(name)
@name = name
end
def create(options = {})
client.indices.create index: name, body: options
end
def delete
client.indices.delete index: name
end
def exists?
client.indices.exists index: name
end
def refresh
client.indices.refresh index: name
end
def store(record)
client.index(
index: name,
type: document_type(record),
id: record.id,
body: search_data(record)
)
end
def remove(record)
client.delete(
index: name,
type: document_type(record),
id: record.id
)
end
def import(records)
records.group_by{|r| document_type(r) }.each do |type, batch|
client.bulk(
index: name,
type: type,
body: batch.map{|r| data = search_data(r); {index: {_id: data["_id"] || data["id"] || r.id, data: data}} }
)
end
end
def retrieve(record)
client.get(
index: name,
type: document_type(record),
id: record.id
)["_source"]
end
def klass_document_type(klass)
klass.model_name.to_s.underscore
end
protected
def client
Searchkick.client
end
def document_type(record)
klass_document_type(record.class)
end
def search_data(record)
source = record.search_data
# stringify fields
source = source.inject({}){|memo,(k,v)| memo[k.to_s] = v; memo}
# Mongoid 4 hack
if defined?(BSON::ObjectId) and source["_id"].is_a?(BSON::ObjectId)
source["_id"] = source["_id"].to_s
end
options = record.class.searchkick_options
# conversions
conversions_field = options[:conversions]
if conversions_field and source[conversions_field]
source[conversions_field] = source[conversions_field].map{|k, v| {query: k, count: v} }
end
# hack to prevent generator field doesn't exist error
(options[:suggest] || []).map(&:to_s).each do |field|
source[field] = nil if !source[field]
end
# locations
(options[:locations] || []).map(&:to_s).each do |field|
if source[field]
if source[field].first.is_a?(Array) # array of arrays
source[field] = source[field].map{|a| a.map(&:to_f).reverse }
else
source[field] = source[field].map(&:to_f).reverse
end
end
end
cast_big_decimal(source)
# p search_data
source.as_json
end
# change all BigDecimal values to floats due to
# https://github.com/rails/rails/issues/6033
# possible loss of precision :/
def cast_big_decimal(obj)
case obj
when BigDecimal
obj.to_f
when Hash
obj.each do |k, v|
obj[k] = cast_big_decimal(v)
end
when Enumerable
obj.map do |v|
cast_big_decimal(v)
end
else
obj
end
end
end
end