2010-12-06 106 views
0

我有一个YAML file of groups,我想进入一个MongoDB的集合称为组与像{"name" => "golf", "parent" => "sports"}文件(顶级团体,喜欢运动,也只是{"name" => "sports"}没有parent。)递归DFS Ruby的方法

我们试图traverse the nested hash,但我不确定它是否正常工作。我宁愿使用递归方法而不是lambda proc。我们应该改变什么才能使其发挥作用?

谢谢!

马特

回答

2

这里的工作代码:

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 

查看Interests YAML
查看acani source

+0

太棒了。 +1 – 2010-12-24 04:34:24

0

你的问题就是如何把这段代码转换:

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]}) 
} 

...使用方法,而不是一个拉姆达?如果是这样,那么:

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 

如果这不是你要求的,请帮助澄清。

+0

谢谢,但lambda不起作用,所以只是将其转换为方法并不能解决核心问题。我的问题是如何让它工作(最好用一种方法代替lambda)。参见[我的答案](http://stackoverflow.com/questions/4368860/recursive-dfs-ruby-method/4524173#4524173)。 – ma11hew28 2010-12-24 04:27:40