确定代码是否在某个块内运行

Question

是否可以确定代码是否在某个块内运行？下面描述了我的用例，其中混入了一些伪代码，这证明了我的想法。欢迎任何建议！当从 WatchTower.bulk_operation do; end 中调用模型上的任何操作时，我希望模型上的块不运行。

# watch_tower.rb

module WatchTower
  extend ActiveSupport::Concern

  class_methods do
    def not_safe_for_bulk
      yield if block_given? && # not called from within bulk_operation block
    end
  end

  def self.bulk_operation
    yield if block_given?
  end
end


# Inside some other file, callbacks on number shouldn't run

WatchTower.bulk_operation do
    5.times { Number.create }
end


# Inside some other file, callbacks on number SHOULD run

Number.create


# number.rb

class Number < ActiveRecord::Base
  include WatchTower

  not_safe_for_bulk do
    after_commit :something_destructive
  end
end

Answer 1

只要你不是从多个线程调用，这样的工作是否可行：

def self.bulk_operation
  @block_op = true
  yield if block_given?
  @block_op = false
end

然后在您的方法调用中，只需检查 @block_op 标志。