diff --git a/ChangeLog b/ChangeLog index 48e4ed11f6..e6b5b94a42 100644 --- a/ChangeLog +++ b/ChangeLog @@ -1,3 +1,8 @@ +Tue Nov 17 14:36:00 2015 Kenichi Kamiya + + * lib/ostruct.rb (dig): Implement OpenStruct#dig + [Feature #11688] + Tue Nov 17 14:04:14 2015 NAKAMURA Usaku * ext/socket/lib/socket.rb (Socket#recvmsg{,_nonblock}): default values diff --git a/lib/ostruct.rb b/lib/ostruct.rb index 63146075be..0e13f40692 100644 --- a/lib/ostruct.rb +++ b/lib/ostruct.rb @@ -214,6 +214,24 @@ class OpenStruct modifiable[new_ostruct_member(name)] = value end + # + # Retrieves the value object corresponding to the each +name+ + # objects repeatedly. + # + # address = OpenStruct.new('city' => "Anytown NC", 'zip' => 12345) + # person = OpenStruct.new('name' => 'John Smith', 'address' => address) + # person.dig(:address, 'zip') # => 12345 + # person.dig(:business_address, 'zip') # => nil + # + def dig(name, *names) + begin + name = name.to_sym + rescue NoMethodError + return + end + @table.dig(name, *names) + end + # # Remove the named field from the object. Returns the value that the field # contained if it was defined. diff --git a/test/ostruct/test_ostruct.rb b/test/ostruct/test_ostruct.rb index 2e8fc7bf58..d8d4cd30f0 100644 --- a/test/ostruct/test_ostruct.rb +++ b/test/ostruct/test_ostruct.rb @@ -108,6 +108,17 @@ class TC_OpenStruct < Test::Unit::TestCase assert_equal :bar, os['foo'] end + def test_dig + os1 = OpenStruct.new + os2 = OpenStruct.new + os1.child = os2 + os2.foo = :bar + os2.child = [42] + assert_equal :bar, os1.dig("child", :foo) + assert_nil os1.dig("parent", :foo) + assert_nil os1.dig("child", 0) + end + def test_to_h h = {name: "John Smith", age: 70, pension: 300} os = OpenStruct.new(h)