66 lines
2 KiB
Ruby
66 lines
2 KiB
Ruby
# rubocop:disable all
|
|
class AddTrigramIndexesForSearching < ActiveRecord::Migration[4.2]
|
|
disable_ddl_transaction!
|
|
|
|
def up
|
|
return unless Gitlab::Database.postgresql?
|
|
|
|
create_trigrams_extension
|
|
|
|
unless trigrams_enabled?
|
|
raise 'You must enable the pg_trgm extension. You can do so by running ' \
|
|
'"CREATE EXTENSION pg_trgm;" as a PostgreSQL super user, this must be ' \
|
|
'done for every GitLab database. For more information see ' \
|
|
'http://www.postgresql.org/docs/current/static/sql-createextension.html'
|
|
end
|
|
|
|
# trigram indexes are case-insensitive so we can just index the column
|
|
# instead of indexing lower(column)
|
|
to_index.each do |table, columns|
|
|
columns.each do |column|
|
|
execute "CREATE INDEX CONCURRENTLY index_#{table}_on_#{column}_trigram ON #{table} USING gin(#{column} gin_trgm_ops);"
|
|
end
|
|
end
|
|
end
|
|
|
|
def down
|
|
return unless Gitlab::Database.postgresql?
|
|
|
|
to_index.each do |table, columns|
|
|
columns.each do |column|
|
|
remove_index table, name: "index_#{table}_on_#{column}_trigram"
|
|
end
|
|
end
|
|
end
|
|
|
|
def trigrams_enabled?
|
|
res = execute("SELECT true AS enabled FROM pg_available_extensions WHERE name = 'pg_trgm' AND installed_version IS NOT NULL;")
|
|
row = res.first
|
|
|
|
row && row['enabled'] == true
|
|
end
|
|
|
|
def create_trigrams_extension
|
|
# This may not work if the user doesn't have permission. We attempt in
|
|
# case we do have permission, particularly for test/dev environments.
|
|
begin
|
|
enable_extension 'pg_trgm'
|
|
rescue
|
|
end
|
|
end
|
|
|
|
def to_index
|
|
{
|
|
ci_runners: [:token, :description],
|
|
issues: [:title, :description],
|
|
merge_requests: [:title, :description],
|
|
milestones: [:title, :description],
|
|
namespaces: [:name, :path],
|
|
notes: [:note],
|
|
projects: [:name, :path, :description],
|
|
snippets: [:title, :file_name],
|
|
users: [:username, :name, :email]
|
|
}
|
|
end
|
|
end
|