rubyでハッシュのkeyを再帰で取り出したいです

Question

Rubyでネストしたハッシュから全てのkeyを配列を取り出したいです。下記のようなクロージャを使用したコードを書いたのですが、うまくいきません。次のエラーが出てしまいます。

NoMethodError: undefined method 'search_nest' for main:Object

def get_all_key(hash)
  all_keys = []
  search_nest = lambda {
    hash.each do |key, hash_value|
      if hash_value.is_a?(Hash)
        all_keys << search_nest(hash_value)
      elsif hash_value.is_a?(Array)
        hash_value.each do |array_value|
          if array_value.is_a?(Hash)
            all_keys << search_nest(array_value)
          else
            hash_value
          end
        end
      else
        hash_value
      end
    end
  }
  search_nest.call
  all_keys
end

これを利用してminitestで全てのkey値をチェックするつもりです。ハッシュの中に配列があり、その配列の中にハッシュがある場合にも対応しようとしています。
他の書き方もあればぜひ教えて下さい。

よろしくお願い致します。

Answer 1

よく分かりません。
search_nest(hash_value)でどういう動作を期待していますか？

ゼロから書き直すと、

def get_all_keys(arg)
  all_keys = []
  case arg
  when Hash
    arg.each do |k,v|
      all_keys << k
      all_keys.concat get_all_keys(v)
    end
  when Array
    arg.each do |v|
      all_keys.concat get_all_keys(v)
    end
  end
  all_keys
end

Comments

[mitsuru793]

私の説明が足らず申し訳ございません。
ですが、otnさんのソースが私の期待する動作です！こんなに綺麗に書ける方法があることを知り、大変嬉しいです。

ご回答、ありがとうございました。

Answer 2

def get_all_keys(arg)
  all_keys = []
  case arg
  when Hash
    all_keys += arg.keys
    arg.values.each do |v|
      all_keys += get_all_keys(v)
    end
  when Array
    arg.each do |v|
      all_keys += get_all_keys(v)
    end
  end
  all_keys
end

h = {A: "11111", B: "22222"}
p h
p get_all_keys(h)

h2 = {A: "11111", B: {C: "333333"}}
p h2
p get_all_keys(h2)

h3 = {A: "11111", B: {C: [D1: "4444", D2: "55555"]}}
p h3
p get_all_keys(h3)

h4 = {A: "11111", B: nil}
p h4
p get_all_keys(h4)

実行結果：

{:A=>"11111", :B=>"22222"}
[:A, :B]
{:A=>"11111", :B=>{:C=>"333333"}}
[:A, :B, :C]
{:A=>"11111", :B=>{:C=>[{:D1=>"4444", :D2=>"55555"}]}}
[:A, :B, :C, :D1, :D2]
{:A=>"11111", :B=>nil}
[:A, :B]

Comments

[mitsuru793]

ご回答、ありがとうございます。
これが私がしたい動作です。大変助かりました。

rubyでは配列を+で連結できることを初めて知りました。