rails--rails/activejob/lib/active_job/railtie.rb

88 lines
2.5 KiB
Ruby

# frozen_string_literal: true
require "global_id/railtie"
require "active_job"
module ActiveJob
# = Active Job Railtie
class Railtie < Rails::Railtie # :nodoc:
config.active_job = ActiveSupport::OrderedOptions.new
config.active_job.custom_serializers = []
config.active_job.log_query_tags_around_perform = true
initializer "active_job.deprecator" do |app|
app.deprecators[:active_job] = ActiveJob.deprecator
end
initializer "active_job.logger" do
ActiveSupport.on_load(:active_job) { self.logger = ::Rails.logger }
end
initializer "active_job.custom_serializers" do |app|
config.after_initialize do
custom_serializers = app.config.active_job.custom_serializers
ActiveJob::Serializers.add_serializers custom_serializers
end
end
initializer "active_job.set_configs" do |app|
options = app.config.active_job
options.queue_adapter ||= :async
config.after_initialize do
options.each do |k, v|
k = "#{k}="
if ActiveJob.respond_to?(k)
ActiveJob.send(k, v)
end
end
end
ActiveSupport.on_load(:active_job) do
# Configs used in other initializers
options = options.except(
:log_query_tags_around_perform,
:custom_serializers
)
options.each do |k, v|
k = "#{k}="
if ActiveJob.respond_to?(k)
ActiveJob.send(k, v)
elsif respond_to? k
send(k, v)
end
end
end
ActiveSupport.on_load(:action_dispatch_integration_test) do
include ActiveJob::TestHelper
end
end
initializer "active_job.set_reloader_hook" do |app|
ActiveSupport.on_load(:active_job) do
ActiveJob::Callbacks.singleton_class.set_callback(:execute, :around, prepend: true) do |_, inner|
app.reloader.wrap do
inner.call
end
end
end
end
initializer "active_job.query_log_tags" do |app|
query_logs_tags_enabled = app.config.respond_to?(:active_record) &&
app.config.active_record.query_log_tags_enabled &&
app.config.active_job.log_query_tags_around_perform
if query_logs_tags_enabled
app.config.active_record.query_log_tags |= [:job]
ActiveSupport.on_load(:active_record) do
ActiveRecord::QueryLogs.taggings[:job] = ->(context) { context[:job].class.name if context[:job] }
end
end
end
end
end