gitlab-org--gitlab-foss/lib/gitlab/sql/pattern.rb

47 lines
1.1 KiB
Ruby
Raw Normal View History

module Gitlab
module SQL
2017-08-28 22:14:41 +00:00
module Pattern
extend ActiveSupport::Concern
2017-08-28 22:14:41 +00:00
MIN_CHARS_FOR_PARTIAL_MATCHING = 3
REGEX_QUOTED_WORD = /(?<=^| )"[^"]+"(?= |$)/
2017-08-28 22:14:41 +00:00
class_methods do
def to_pattern(query)
2017-08-29 09:00:03 +00:00
if partial_matching?(query)
2017-08-28 22:14:41 +00:00
"%#{sanitize_sql_like(query)}%"
2017-08-29 09:00:03 +00:00
else
sanitize_sql_like(query)
2017-08-28 22:14:41 +00:00
end
end
2017-08-28 22:14:41 +00:00
def partial_matching?(query)
query.length >= MIN_CHARS_FOR_PARTIAL_MATCHING
end
def fuzzy_arel_match(column, query)
words = select_fuzzy_words(query)
matches = words.map { |word| arel_table[column].matches(to_pattern(word)) }
matches.reduce { |result, match| result.and(match) }
end
def select_fuzzy_words(query)
quoted_words = query.scan(REGEX_QUOTED_WORD)
query = quoted_words.reduce(query) { |q, quoted_word| q.sub(quoted_word, '') }
words = query.split(/\s+/)
quoted_words.map! { |quoted_word| quoted_word[1..-2] }
words.concat(quoted_words)
words.select { |word| partial_matching?(word) }
end
end
end
end
end