60 lines
1.7 KiB
Ruby
60 lines
1.7 KiB
Ruby
enabled = ENV['ES_ENABLED'] == 'true'
|
|
host = ENV.fetch('ES_HOST') { 'localhost' }
|
|
port = ENV.fetch('ES_PORT') { 9200 }
|
|
user = ENV.fetch('ES_USER') { nil }
|
|
password = ENV.fetch('ES_PASS') { nil }
|
|
fallback_prefix = ENV.fetch('REDIS_NAMESPACE') { nil }
|
|
prefix = ENV.fetch('ES_PREFIX') { fallback_prefix }
|
|
|
|
Chewy.settings = {
|
|
host: "#{host}:#{port}",
|
|
prefix: prefix,
|
|
enabled: enabled,
|
|
journal: false,
|
|
user: user,
|
|
password: password,
|
|
sidekiq: { queue: 'pull' },
|
|
}
|
|
|
|
# We use our own async strategy even outside the request-response
|
|
# cycle, which takes care of checking if ElasticSearch is enabled
|
|
# or not. However, mind that for the Rails console, the :urgent
|
|
# strategy is set automatically with no way to override it.
|
|
Chewy.root_strategy = :custom_sidekiq
|
|
Chewy.request_strategy = :custom_sidekiq
|
|
Chewy.use_after_commit_callbacks = false
|
|
|
|
module Chewy
|
|
class << self
|
|
def enabled?
|
|
settings[:enabled]
|
|
end
|
|
end
|
|
end
|
|
|
|
# ElasticSearch uses Faraday internally. Faraday interprets the
|
|
# http_proxy env variable by default which leads to issues when
|
|
# Mastodon is run with hidden services enabled, because
|
|
# ElasticSearch is *not* supposed to be accessed through a proxy
|
|
Faraday.ignore_env_proxy = true
|
|
|
|
# Elasticsearch 7.x workaround
|
|
Elasticsearch::Transport::Client.prepend Module.new {
|
|
def search(arguments = {})
|
|
arguments[:rest_total_hits_as_int] = true
|
|
super arguments
|
|
end
|
|
}
|
|
|
|
Elasticsearch::API::Indices::IndicesClient.prepend Module.new {
|
|
def create(arguments = {})
|
|
arguments[:include_type_name] = true
|
|
super arguments
|
|
end
|
|
|
|
def put_mapping(arguments = {})
|
|
arguments[:include_type_name] = true
|
|
super arguments
|
|
end
|
|
}
|