debian-mirror-gitlab/config/initializers/forbid_sidekiq_in_transactions.rb

57 lines
1.8 KiB
Ruby
Raw Normal View History

2017-09-10 17:25:29 +05:30
module Sidekiq
module Worker
2018-03-17 18:26:18 +05:30
EnqueueFromTransactionError = Class.new(StandardError)
2017-09-10 17:25:29 +05:30
def self.skipping_transaction_check(&block)
2019-09-30 21:07:59 +05:30
previous_skip_transaction_check = self.skip_transaction_check
Thread.current[:sidekiq_worker_skip_transaction_check] = true
2017-09-10 17:25:29 +05:30
yield
ensure
2019-09-30 21:07:59 +05:30
Thread.current[:sidekiq_worker_skip_transaction_check] = previous_skip_transaction_check
end
def self.skip_transaction_check
Thread.current[:sidekiq_worker_skip_transaction_check]
2017-09-10 17:25:29 +05:30
end
module ClassMethods
2018-03-17 18:26:18 +05:30
module NoEnqueueingFromTransactions
2017-09-10 17:25:29 +05:30
%i(perform_async perform_at perform_in).each do |name|
define_method(name) do |*args|
2019-09-30 21:07:59 +05:30
if !Sidekiq::Worker.skip_transaction_check && Gitlab::Database.inside_transaction?
2018-03-17 18:26:18 +05:30
begin
raise Sidekiq::Worker::EnqueueFromTransactionError, <<~MSG
`#{self}.#{name}` cannot be called inside a transaction as this can lead to
race conditions when the worker runs before the transaction is committed and
tries to access a model that has not been saved yet.
Use an `after_commit` hook, or include `AfterCommitQueue` and use a `run_after_commit` block instead.
MSG
rescue Sidekiq::Worker::EnqueueFromTransactionError => e
2020-11-24 15:15:51 +05:30
Gitlab::AppLogger.error(e.message) if ::Rails.env.production?
2020-01-01 13:55:28 +05:30
Gitlab::ErrorTracking.track_and_raise_for_dev_exception(e)
2018-03-17 18:26:18 +05:30
end
end
super(*args)
2017-09-10 17:25:29 +05:30
end
end
end
2018-03-17 18:26:18 +05:30
prepend NoEnqueueingFromTransactions
2017-09-10 17:25:29 +05:30
end
end
end
module ActiveRecord
class Base
module SkipTransactionCheckAfterCommit
2021-01-03 14:25:43 +05:30
def committed!(*args, **kwargs)
2017-09-10 17:25:29 +05:30
Sidekiq::Worker.skipping_transaction_check { super }
end
end
prepend SkipTransactionCheckAfterCommit
end
end