elasticsearch_helper.rb
2.71 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
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.filters
{
:relevance => { label: _("Relevance")},
:lexical => { label: _("Alphabetical")},
:more_recent => { label: _("More Recent")},
}
end
def process_results
selected_type = (params[:selected_type].presence|| :all).to_sym
selected_type == :all ? search_from_all_models : search_from_model(selected_type)
end
private
def search_from_all_models
query = get_query params[:query], sort_by: get_sort_by(params[:selected_filter])
Elasticsearch::Model.search(query,searchable_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, sort_by: get_sort_by(params[:selected_filter])
klass.search(query, size: default_per_page(params[:per_page])).page(params[:page]).records
rescue
[]
end
end
def default_per_page per_page=nil
per_page ||= 10
end
def get_sort_by sort_by
case sort_by
when "lexical"
{ "name.raw" => {"order" => "asc" }}
when "recent"
{ "created_at" => {"order" => "desc"}}
end
end
def searchable_models
begin
ElasticsearchHelper::searchable_types.except(:all).keys.map { | model | model.to_s.classify.constantize }
rescue
[]
end
end
def query_method expression="", fields=[]
query_exp = {}
if expression.blank?
query_exp = {
match_all: {}
}
else
query_exp = {
multi_match: {
query: expression.downcase,
fields: fields,
tie_breaker: 0.4,
minimum_should_match: "40%"
}
}
end
query_exp
end
def get_query text="", options={}
klass = options[:klass]
sort_by = options[:sort_by]
fields = klass.nil? ? (fields_from_models searchable_models) : (fields_from_models [klass])
query = {
query: query_method(text, fields)
}
query[:sort] = [sort_by,"_score"] if sort_by
query
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
end