• 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
  • 28
  • 29
  • 30
  • 31

以前紹介した
Hash#only
にちょっとした修正。

lib/hash_ext.rb

ruby>>
class Hash
def only(*args)
args = *args if args[0].is_a? Array
args.inject({}){|hash, key| hash[key] = self[key] if include? key; hash}
end

def only!(*args)
args = *args if args[0].is_a? Array
args.inject({}){|hash, key| hash[key] = self[key]; hash}
end
<<--

以前のHash#onlyは、指定したキーが存在しない場合でも

ruby>>
{:foo => 1}.only(:bar) #=> {:bar => nil}
<<--

というように、nilを指すHashを返していましたが、
キーが存在しない場合に空のHashを返すようにしてみました。
従来の挙動をするメソッドは、Hash#only!と感嘆符付きの
メソッドにしています。
修正後の挙動は以下の通り。

ruby>>
{:foo => 1}.only(:bar) #=> {}
{:foo => 1}.only!(:bar) #=> {:bar => nil}
<<--

posted by genki genki on Sat 19 Jan 2008 at 13:30 with 0 comments