19th Sat
Hash#only and Hash#only!
以前紹介した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 on Sat 19 Jan 2008 at 13:30 with 0 comments