ruby - 一个 block 如何检测到它在另一个 block 内？

Question

这是我的代码:

def block
  puts "from block"
  yield
end

block do
  puts "from command line"
  block do

  end
end

这是输出:

from block
from command line
from block

我想知道第二个 block 如何检测到它在另一个 block 内(使用相同的方法)。

所以输出将是这样的:

from block 1
from command line
from block 2

这可能吗？因为我希望嵌套 block 知道这一点并运行一些额外的代码。

谢谢!

Answer 1

您可以使用实例变量跟踪 block 级别，每当您进入 block 时递增它，并在您离开 block 时递减它:

def block
  @block_level ||= 0
  @block_level += 1

  puts "from block #@block_level"

  yield

  @block_level -= 1
end