Commit 656454bd authored by Dmitry Gruzd's avatar Dmitry Gruzd Committed by Terri Chu

Remove type filter from issues/mr/notes ES queries

parent 4a4b80a6
......@@ -65,10 +65,10 @@ module Elastic
}
end
def basic_query_hash(fields, query, count_only: false)
def basic_query_hash(fields, query, options = {})
fields = CustomLanguageAnalyzers.add_custom_analyzers_fields(fields)
fields = remove_fields_boost(fields) if count_only
fields = remove_fields_boost(fields) if options[:count_only]
query_hash =
if query.present?
......@@ -82,18 +82,46 @@ module Elastic
}
}
build_query_filter(simple_query_string, options)
else
{
query: {
bool: {
must: { match_all: {} }
}
},
track_scores: true
}
end
if options[:count_only]
query_hash[:size] = 0
else
query_hash[:highlight] = highlight_options(fields)
end
query_hash
end
def build_query_filter(simple_query_string, options)
must = []
filter = [{
filter = if options[:no_join_project]
[]
else
[
{
term: {
type: {
_name: context.name(:doc, :is_a, self.es_type),
value: self.es_type
}
}
}]
}
]
end
if count_only
if options[:count_only]
filter << simple_query_string
else
must << simple_query_string
......@@ -107,24 +135,6 @@ module Elastic
}
}
}
else
{
query: {
bool: {
must: { match_all: {} }
}
},
track_scores: true
}
end
if count_only
query_hash[:size] = 0
else
query_hash[:highlight] = highlight_options(fields)
end
query_hash
end
def iid_query_hash(iid)
......
......@@ -6,6 +6,9 @@ module Elastic
include StateFilter
def elastic_search(query, options: {})
options[:features] = 'issues'
options[:no_join_project] = true
query_hash =
if query =~ /#(\d+)\z/
iid_query_hash(Regexp.last_match(1))
......@@ -13,11 +16,9 @@ module Elastic
# iid field can be added here as lenient option will
# pardon format errors, like integer out of range.
fields = %w[iid^3 title^2 description]
basic_query_hash(fields, query, count_only: options[:count_only])
basic_query_hash(fields, query, options)
end
options[:features] = 'issues'
options[:no_join_project] = true
context.name(:issue) do
query_hash = context.name(:authorized) { authorization_filter(query_hash, options) }
query_hash = context.name(:confidentiality) { confidentiality_filter(query_hash, options) }
......
......@@ -6,6 +6,9 @@ module Elastic
include StateFilter
def elastic_search(query, options: {})
options[:features] = 'merge_requests'
options[:no_join_project] = Elastic::DataMigrationService.migration_has_finished?(:add_new_data_to_merge_requests_documents)
query_hash =
if query =~ /\!(\d+)\z/
iid_query_hash(Regexp.last_match(1))
......@@ -14,11 +17,9 @@ module Elastic
# pardon format errors, like integer out of range.
fields = %w[iid^3 title^2 description]
basic_query_hash(fields, query, count_only: options[:count_only])
basic_query_hash(fields, query, options)
end
options[:features] = 'merge_requests'
options[:no_join_project] = Elastic::DataMigrationService.migration_has_finished?(:add_new_data_to_merge_requests_documents)
context.name(:merge_request) do
query_hash = context.name(:authorized) { project_ids_filter(query_hash, options) }
query_hash = context.name(:match) { state_filter(query_hash, options) }
......
......@@ -6,7 +6,7 @@ module Elastic
def elastic_search(query, options: {})
options[:in] = %w[title^2 description]
query_hash = basic_query_hash(options[:in], query, count_only: options[:count_only])
query_hash = basic_query_hash(options[:in], query, options)
query_hash = context.name(:milestone, :related) { project_ids_filter(query_hash, options) }
search(query_hash, options)
......
......@@ -11,9 +11,10 @@ module Elastic
def elastic_search(query, options: {})
options[:in] = ['note']
query_hash = basic_query_hash(%w[note], query, count_only: options[:count_only])
options[:no_join_project] = true
query_hash = basic_query_hash(%w[note], query, options)
context.name(:note) do
query_hash = context.name(:authorized) { project_ids_filter(query_hash, options) }
query_hash = context.name(:confidentiality) { confidentiality_filter(query_hash, options) }
......
......@@ -6,7 +6,7 @@ module Elastic
def elastic_search(query, options: {})
options[:in] = %w[name^10 name_with_namespace^2 path_with_namespace path^9 description]
query_hash = basic_query_hash(options[:in], query, count_only: options[:count_only])
query_hash = basic_query_hash(options[:in], query, options)
filters = [{ terms: { _name: context.name(:doc, :is_a, es_type), type: [es_type] } }]
......
......@@ -78,8 +78,7 @@ RSpec.describe Issue, :elastic do
it "names elasticsearch queries" do
described_class.elastic_search('*').total_count
assert_named_queries('doc:is_a:issue',
'issue:match:search_terms',
assert_named_queries('issue:match:search_terms',
'issue:authorized:project')
end
......
......@@ -43,8 +43,7 @@ RSpec.describe MergeRequest, :elastic do
it "names elasticsearch queries" do
described_class.elastic_search('*').total_count
assert_named_queries('doc:is_a:merge_request',
'merge_request:match:search_terms',
assert_named_queries('merge_request:match:search_terms',
'merge_request:authorized:project')
end
......
......@@ -58,8 +58,7 @@ RSpec.describe Note, :elastic, :clean_gitlab_redis_shared_state do
it "names elasticsearch queries" do
described_class.elastic_search('*').total_count
assert_named_queries("doc:is_a:note",
"note:match:search_terms",
assert_named_queries("note:match:search_terms",
"note:authorized")
end
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment