jaculabilis/danbooru
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix autocomplete

Browse Source
This commit is contained in:
Albert Yi
2018-09-04 14:22:37 -07:00
parent 68e7de1fc4
commit 3987f45362
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
app/logical/tag_autocomplete.rb
View File

@@ -24,7 +24,7 @@ module TagAutocomplete
def count_sort(query, words) def count_sort(query, words)
words.uniq.slice(0, LIMIT).sort_by do |x| words.uniq.slice(0, LIMIT).sort_by do |x|
x.post_count.to_i * x.weight x.post_count * x.weight
end.reverse end.reverse
end end
@@ -35,7 +35,7 @@ module TagAutocomplete
.order("post_count desc") .order("post_count desc")
.limit(n) .limit(n)
.pluck(:name, :post_count, :category) .pluck(:name, :post_count, :category)
.map {|row| Result.new(*row, 1.0)} .map {|row| Result.new(*row, nil, 1.0)}
end end
def search_fuzzy(query, n=5) def search_fuzzy(query, n=5)
@@ -54,7 +54,7 @@ module TagAutocomplete
.order(Arel.sql("word_similarity(name, #{Tag.connection.quote(query)}) DESC")) .order(Arel.sql("word_similarity(name, #{Tag.connection.quote(query)}) DESC"))
.limit(n) .limit(n)
.pluck(:name, :post_count, :category) .pluck(:name, :post_count, :category)
.map {|row| Result.new(*row, 0.1)} .map {|row| Result.new(*row, nil, 0.1)}
end end
def search_prefix(query, n=3) def search_prefix(query, n=3)
@@ -85,7 +85,7 @@ module TagAutocomplete
.order("post_count desc") .order("post_count desc")
.limit(n) .limit(n)
.pluck(:name, :post_count, :category) .pluck(:name, :post_count, :category)
.map {|row| Result.new(*row, 0.8)} .map {|row| Result.new(*row, nil, 0.8)}
end end
def search_aliases(query, n=10) def search_aliases(query, n=10)