65 lines
1.7 KiB
Ruby
65 lines
1.7 KiB
Ruby
# TODO: make this private, not-public facing.
|
|
# for now we keep this here for simplicity reasons
|
|
get '/search/index' do
|
|
es_client = Elasticsearch::Client.new log: true
|
|
|
|
# first delete the index
|
|
begin
|
|
es_client.indices.delete index: 'ribica'
|
|
rescue
|
|
logger.warn "Ribica index could not be deleted. Continuing with indexing operation..."
|
|
end
|
|
|
|
# now index items
|
|
all_items = Item.includes(sub_category: { category: :section }).all.to_a
|
|
all_items.each do |item|
|
|
es_client.index index: 'ribica', type: 'items', id: item.id, body: {
|
|
title: 'Test',
|
|
name: item.name,
|
|
code: item.code,
|
|
description: item.description,
|
|
sub_category: item.sub_category.name,
|
|
category: item.sub_category.category.name,
|
|
section: item.sub_category.category.section.name
|
|
}
|
|
end
|
|
|
|
"ok".to_json
|
|
end
|
|
|
|
get '/search' do
|
|
es_client = Elasticsearch::Client.new log: true
|
|
q = params[:q]
|
|
|
|
# for now we do the basic query
|
|
results = es_client.search index: 'ribica', type: 'items', body: { query: { match: { _all: q } } }
|
|
ids = results["hits"]["hits"].map do |r|
|
|
r["_id"]
|
|
end
|
|
|
|
ids_with_score = {}
|
|
results["hits"]["hits"].each do |r|
|
|
ids_with_score[r["_id"].to_i] = {:score => r["_score"], :item => nil}
|
|
end
|
|
|
|
if ids.length > 0
|
|
res = Item.where(:id => ids).to_a
|
|
# make sure we have correct relevance order, since `where in` does not guarantee order
|
|
res.each do |ii|
|
|
ids_with_score[ii.id][:item] = ii
|
|
end
|
|
final = []
|
|
ids_with_score.each do |k,v|
|
|
final << v
|
|
end
|
|
final.sort_by! {|v| -v[:score]}
|
|
|
|
final = final.map do |f|
|
|
f[:item]
|
|
end
|
|
prepare_items_for_mass_display(final)
|
|
else
|
|
[].to_json
|
|
end
|
|
end
|