elasticsearch_plugin_controller.rb
1.78 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
class ElasticsearchPluginController < ApplicationController
no_design_blocks
SEARCHABLE_MODELS = {communities: true, articles: true, people: true}
def index
search()
render :action => 'search'
end
def search
@results = []
@query = params[:q]
@checkbox = {}
if params[:model].present?
params[:model].keys.each do |model|
@checkbox[model.to_sym] = true
results model
end
else
unless params[:q].blank?
SEARCHABLE_MODELS.keys.each do |model|
results model
end
end
end
end
private
def get_query text, klass
query = {}
unless text.blank?
fields = klass::SEARCHABLE_FIELDS.map do |key, value|
if value[:weight]
"#{key}^#{value[:weight]}"
else
"#{key}"
end
end
query = {
query: {
match_all: {
}
},
filter: {
regexp: {
name: {
value: ".*" + text + ".*" }
}
},
suggest: {
autocomplete: {
text: text,
term: {
field: "name",
suggest_mode: "always"
}
}
}
}
end
query
end
def get_terms params
terms = {}
return terms unless params[:filter].present?
params[:filter].keys.each do |model|
terms[model] = {}
params[:filter][model].keys.each do |filter|
@checkbox[filter.to_sym] = true
terms[model][params[:filter][model.to_sym][filter]] = filter
end
end
terms
end
def results model
klass = model.to_s.classify.constantize
query = get_query params[:q], klass
@results |= klass.__elasticsearch__.search(query).records.to_a
end
end