ruby - 递归 DFS Ruby 方法

Question

我有一个YAML 组文件，我想将它放入一个名为 groups 的 MongoDB 集合中，其中包含类似文档{"name" => "golf", "parent" => "sports"}（顶级组，如运动，只是{"name" => "sports"}没有parent.）

我们正在尝试遍历嵌套的 hash，但我不确定它是否工作正常。我更喜欢使用递归方法而不是 lambda proc。我们应该改变什么才能让它发挥作用？

谢谢！

马特

Answer 1

这是工作代码：

require 'mongo'
require 'yaml'

conn = Mongo::Connection.new
db = conn.db("acani")
interests = db.collection("interests")
@@interest_id = 0
interests_hash = YAML::load_file('interests.yml')

def interests.insert_interest(interest, parent=nil)
  interest_id = @@interest_id.to_s(36)
  if interest.is_a? String # base case
    insert({:_id => interest_id, :n => interest, :p => parent})
    @@interest_id += 1
  else # it's a hash
    interest = interest.first # get key-value pair in hash
    interest_name = interest[0]
    insert({:_id => interest_id, :n => interest_name, :p => parent})
    @@interest_id += 1
    interest[1].each do |i|
      insert_interest(i, interest_name)
    end
  end
end

interests.insert_interest interests_hash

查看兴趣 YAML。
查看acani 源代码。

Answer 2

您的问题只是如何转换此代码：

insert_enumerable = lambda {|obj, collection|
   # obj = {:value => obj} if !obj.kind_of? Enumerable
   if(obj.kind_of? Array or obj.kind_of? Hash)
      obj.each do |k, v|
        v = (v.nil?) ? k : v
        insert_enumerable.call({:value => v, :parent => obj}, collection)
      end
   else
      obj = {:value => obj}
   end
   # collection.insert({name => obj[:value], :parent => obj[:parent]})
   pp({name => obj[:value], :parent => obj[:parent]})
}

...使用方法而不是 lambda？如果是这样，那么：

def insert_enumerable( obj, collection )
   # obj = {:value => obj} if !obj.kind_of? Enumerable
   if(obj.kind_of? Array or obj.kind_of? Hash)
      obj.each do |k, v|
        v = (v.nil?) ? k : v
        insert_enumerable({:value => v, :parent => obj}, collection)
      end
   else
      obj = {:value => obj}
   end
   # collection.insert({name => obj[:value], :parent => obj[:parent]})
   pp({name => obj[:value], :parent => obj[:parent]})
end

如果这不是您要问的，请帮助澄清。