Maintenance: Search works if multiple KBs are present in database.

This commit is contained in:
Mantas Masalskis 2020-09-23 16:10:50 +02:00 committed by Thorsten Eckel
parent c6e11f3f4e
commit 5bb948d87f
4 changed files with 27 additions and 11 deletions

View file

@ -96,7 +96,7 @@ class KnowledgeBase::Answer::Translation < ApplicationModel
} }
end end
def search_es_filter(es_response, _query, kb_locale, options) def search_es_filter(es_response, _query, kb_locales, options)
return es_response if options[:user]&.permissions?('knowledge_base.editor') return es_response if options[:user]&.permissions?('knowledge_base.editor')
answer_translations_id = es_response.pluck(:id) answer_translations_id = es_response.pluck(:id)
@ -104,7 +104,7 @@ class KnowledgeBase::Answer::Translation < ApplicationModel
allowed_answer_translation_ids = KnowledgeBase::Answer allowed_answer_translation_ids = KnowledgeBase::Answer
.internal .internal
.joins(:translations) .joins(:translations)
.where(knowledge_base_answer_translations: { id: answer_translations_id, kb_locale_id: kb_locale.id }) .where(knowledge_base_answer_translations: { id: answer_translations_id, kb_locale_id: kb_locales.map(&:id) })
.pluck('knowledge_base_answer_translations.id') .pluck('knowledge_base_answer_translations.id')
es_response.filter { |elem| allowed_answer_translation_ids.include? elem[:id].to_i } es_response.filter { |elem| allowed_answer_translation_ids.include? elem[:id].to_i }

View file

@ -20,27 +20,27 @@ class KnowledgeBase
user: current_user user: current_user
} }
kb_locale = KnowledgeBase::Locale.preferred(current_user, KnowledgeBase.first) kb_locales = KnowledgeBase.active.map { |elem| KnowledgeBase::Locale.preferred(current_user, elem) }
# try search index backend # try search index backend
if SearchIndexBackend.enabled? if SearchIndexBackend.enabled?
search_es(params[:query], kb_locale, options) search_es(params[:query], kb_locales, options)
else else
# fallback do sql query # fallback do sql query
search_sql(params[:query], kb_locale, options) search_sql(params[:query], kb_locales, options)
end end
end end
def search_es(query, kb_locale, options) def search_es(query, kb_locales, options)
options[:query_extension] = { bool: { filter: { term: { kb_locale_id: kb_locale.id } } } } options[:query_extension] = { bool: { filter: { terms: { kb_locale_id: kb_locales.map(&:id) } } } }
es_response = SearchIndexBackend.search(query, name, options) es_response = SearchIndexBackend.search(query, name, options)
es_response = search_es_filter(es_response, query, kb_locale, options) if defined? :search_es_filter es_response = search_es_filter(es_response, query, kb_locales, options) if defined? :search_es_filter
es_response.map { |item| lookup(id: item[:id]) }.compact es_response.map { |item| lookup(id: item[:id]) }.compact
end end
def search_sql(query, kb_locale, options) def search_sql(query, kb_locales, options)
table_name = arel_table.name table_name = arel_table.name
order_sql = search_get_order_sql(options[:sort_by], options[:order_by], "#{table_name}.updated_at ASC") order_sql = search_get_order_sql(options[:sort_by], options[:order_by], "#{table_name}.updated_at ASC")
@ -48,7 +48,7 @@ class KnowledgeBase
query.delete! '*' query.delete! '*'
search_fallback("%#{query}%", options: options) search_fallback("%#{query}%", options: options)
.where(kb_locale: kb_locale) .where(kb_locale: kb_locales)
.order(Arel.sql(order_sql)) .order(Arel.sql(order_sql))
.offset(options[:from]) .offset(options[:from])
.limit(options[:limit]) .limit(options[:limit])

View file

@ -3,6 +3,7 @@ FactoryBot.define do
transient do transient do
add_translation { true } add_translation { true }
translation_traits { [] } translation_traits { [] }
translation_attributes { {} }
knowledge_base { nil } knowledge_base { nil }
end end
@ -11,7 +12,7 @@ FactoryBot.define do
before(:create) do |answer, context| before(:create) do |answer, context|
next if answer.translations.present? next if answer.translations.present?
answer.translations << build('knowledge_base/answer/translation', *context.translation_traits, answer: answer) answer.translations << build('knowledge_base/answer/translation', *context.translation_traits, answer: answer, **context.translation_attributes)
end end
trait :draft # empty placeholder for better readability trait :draft # empty placeholder for better readability

View file

@ -43,5 +43,20 @@ RSpec.describe KnowledgeBase::Answer::Translation, type: :model, current_user_id
include_examples 'verify given permissions', trait: :internal, admin: true, agent: true, customer: false include_examples 'verify given permissions', trait: :internal, admin: true, agent: true, customer: false
include_examples 'verify given permissions', trait: :draft, admin: true, agent: false, customer: false include_examples 'verify given permissions', trait: :draft, admin: true, agent: false, customer: false
include_examples 'verify given permissions', trait: :archived, admin: true, agent: false, customer: false include_examples 'verify given permissions', trait: :archived, admin: true, agent: false, customer: false
shared_examples 'verify multiple KBs support' do |elasticsearch:|
it 'searches in multiple KBs', searchindex: elasticsearch do
title = Faker::Appliance.equipment
user = create(:admin)
create_list(:knowledge_base_answer, 2, :published, translation_attributes: { title: title })
configure_elasticsearch(required: true, rebuild: true) if elasticsearch
expect(described_class.search({ query: title, current_user: user }).count).to be 2
end
end
include_examples 'verify multiple KBs support', elasticsearch: true
include_examples 'verify multiple KBs support', elasticsearch: false
end end
end end