elasticsearch_helper.rb
2.4 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
module ElasticsearchHelper
def self.searchable_types
{
:all => { label: _("All Results")},
:text_article => { label: _("Articles")},
:uploaded_file => { label: _("Files")},
:community => { label: _("Communities")},
:event => { label: _("Events")},
:person => { label: _("People")}
}
end
def self.search_filters
{
:lexical => { label: _("Alphabetical Order")},
:recent => { label: _("More Recent Order")},
:access => { label: _("More accessed")}
}
end
def fields_from_models klasses
fields = Set.new
klasses.each do |klass|
klass::SEARCHABLE_FIELDS.map do |key, value|
if value and value[:weight]
fields.add "#{key}^#{value[:weight]}"
else
fields.add "#{key}"
end
end
end
fields.to_a
end
def process_results
selected_type = (params[:selected_type]|| :all).to_sym
if selected_type == :all
search_from_all_models
else
search_from_model selected_type
end
end
def search_from_all_models
query = get_query params[:query]
models = searchable_models
Elasticsearch::Model.search(query, models, size: default_per_page(params[:per_page])).page(params[:page]).records
end
def search_from_model model
begin
klass = model.to_s.classify.constantize
query = get_query params[:query], klass
klass.search(query, size: default_per_page(params[:per_page])).page(params[:page]).records
rescue
[]
end
end
def default_per_page per_page
per_page ||= 10
end
private
def get_query text, klass=nil
fields = klass.nil? ? (fields_from_models searchable_models) : (fields_from_models [klass])
query = {}
unless text.blank?
text = text.downcase
query = {
query: {
multi_match: {
query: text,
type: "phrase",
fields: fields,
zero_terms_query: "none"
},
},
sort: [
{"name.raw" => {"order" => "asc"}}
],
suggest: {
autocomplete: {
text: text,
term: {
field: "name",
suggest_mode: "always"
}
}
}
}
end
query
end
def searchable_models
SEARCHABLE_TYPES.except(:all).keys.map { | model | model.to_s.classify.constantize }
end
end