granblue.team/hensei-api
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Implement JP search and trigram for EN

Browse source
This commit is contained in:
Justin Edmund 2022-03-08 17:43:40 -08:00
parent 0661b5a50d
commit 88f316d6a7
4 changed files with 58 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
app/controllers/api/v1/search_controller.rb
View file

@ -1,28 +1,42 @@
class Api::V1::SearchController < Api::V1::ApiController
def characters
locale = params[:locale] || 'en'
if params[:query].present?
excludes = params[:excludes] ?
params[:excludes].split(',').map { |e| "%#{e.gsub(/\([^()]*\)/, '').strip}%" } : ''
@characters = Character.where("name_en ILIKE ? AND name_en NOT ILIKE ALL(ARRAY[?])", "%#{params[:query]}%", excludes).limit(10)
# @characters = Character.search(query).limit(10)
if locale == 'ja'
@characters = Character.jp_search(params[:query]).limit(10)
else
# @characters = Character.where("name_en ILIKE ? AND name_en NOT ILIKE ALL(ARRAY[?])", "%#{params[:query]}%", excludes).limit(10)
@characters = Character.en_search(params[:query]).limit(10)
end
else
@characters = Character.all
end
end
def weapons
locale = params[:locale] || 'en'
if params[:query].present?
@weapons = Weapon.search(params[:query]).limit(10)
if locale == 'ja'
@weapons = Weapon.jp_search(params[:query]).limit(10)
else
@weapons = Weapon.en_search(params[:query]).limit(10)
end
else
@weapons = Weapon.all
end
end
def summons
locale = params[:locale] || 'en'
if params[:query].present?
excludes = params[:excludes] ? params[:excludes].split(',').each { |e| "!#{e}" }.join(' ') : ''
@summons = Summon.search(params[:query]).limit(10)
if locale == 'ja'
@summons = Summon.jp_search(params[:query]).limit(10)
else
@summons = Summon.en_search(params[:query]).limit(10)
end
else
@summons = Summon.all
end

16
app/models/character.rb
View file

@ -1,12 +1,20 @@
class Character < ApplicationRecord
include PgSearch::Model
pg_search_scope :search,
against: [:name_en, :name_jp],
pg_search_scope :en_search,
against: :name_en,
using: {
trigram: {
threshold: 0.18
}
}
pg_search_scope :jp_search,
against: :name_jp,
using: {
tsearch: {
negation: true,
prefix: true
prefix: true,
dictionary: "simple"
}
}

15
app/models/summon.rb
View file

@ -1,11 +1,20 @@
class Summon < ApplicationRecord
include PgSearch::Model
pg_search_scope :search,
against: [:name_en, :name_jp],
pg_search_scope :en_search,
against: :name_en,
using: {
trigram: {
threshold: 0.18
}
}
pg_search_scope :jp_search,
against: :name_jp,
using: {
tsearch: {
prefix: true
prefix: true,
dictionary: "simple"
}
}

15
app/models/weapon.rb
View file

@ -1,11 +1,20 @@
class Weapon < ApplicationRecord
include PgSearch::Model
pg_search_scope :search,
against: [:name_en, :name_jp],
pg_search_scope :en_search,
against: :name_en,
using: {
trigram: {
threshold: 0.18
}
}
pg_search_scope :jp_search,
against: :name_jp,
using: {
tsearch: {
prefix: true
prefix: true,
dictionary: "simple"
}
}