methodの途中の作業

• November 9, 2009

yieldを使う

rubyであるメソッドの途中で別の作業をしたいときに、その内容をブロックとして与えて、yieldで実行することができる。ブロックを与えないときには何もしないのかと思いきや、エラーが起こるので、yield if block_given?としておけば良い。

問題なのは、いくつかの操作をメソッドの途中でしたいときだ。ブロックは二つ与えられないしと思っていたら、こんな感じで実現できることに気がついた。

def test
  puts 0
  yield(0) if block_given?
  puts 1
  yield(1) if block_given?
  puts 2
end
test{|i|
case i
when 0; puts 'a'
when 1; puts 'b'
end
}
test{ puts 'a' }
test

しかし、手続オブジェクトを引数に使うと、複数の手続を指定できるので、素直に書ける。

def test(a=nil,b=nil)
  puts 0
  a.call if a
  puts 1
  b.call if b
  puts 2
end
a=Proc.new{ puts 'a' }
b=Proc.new{ puts 'b' }
test(a,b)
test(a)
test()

def test(a=nil)
  puts 0
  a.call if a.class==Proc
  puts 1
  yield if block_given?
  puts 2
end
test(Proc.new{ puts 'a' }){ puts 'b' }
test{puts 'b'}
test()