ブロックを受け取るメソッドを再帰的に呼び出したい

Question

Rubyで、ブロックを引数にもつメソッドを再帰的に呼び出す場合、どのようにすれば良いでしょうか。

例えば以下のコードを考えています

class Node
  attr_reader :id, :name
  attr_accessor :children
  
  def initialize(id, name)
    @id = id
    @name = name
    @children = []
  end
  
  # selfを根とした木の配列を再帰的に作る
  def subtree(prc)
    children.flat_map {|ch| ch.subtree(prc) } + [prc.call(self)]
  end
  
end

root = Node.new(0, 'root')
ch_1 = Node.new(1, 'child_1')
ch_2 = Node.new(2, 'child_2')

# ノードを結合
root.children << ch_1
root.children << ch_2

# procオブジェクトを渡す場合
p root.subtree(proc {|n| n.id})
p root.subtree(proc {|n| n.name})
# [1, 2, 0]
# ["child_1", "child_2", "root"]

# ブロックを渡す場合, エラーになる
# このように呼び出したい
p root.subtree(&:id)
p root.subtree(&:name)

まずは、root.subtree(proc {|n| n.id})のようにprocオブジェクトを引数に渡して呼ぶことはできましたが、
目標はroot.subtree(&:id)のように省略記法のブロックを渡して呼び出したいです。
よろしくお願いします。

Answer 1

  def subtree(&prc)
    children.flat_map {|ch| ch.subtree(&prc) } + [prc.call(self)]
  end

です。

参考：
メソッド呼び出し(super・ブロック付き・yield) (Ruby 2.6.0)

Comments

[y-doi]

`&' で修飾したProcオブジェクトを渡せばいいんですね。
ありがとうございました、助かりました。