我有一个Hash
,我想在深层插入一些数据,但任何级别都可能缺少密钥。因此,在更新其每个级别的值之前,我有条件地对其进行初始化。
有什么更好的方法来编写这个或一种方法可以使代码不那么丑陋?
data[:foo] ||= {}
data[:foo][:bar] ||= {}
data[:foo][:bar][:baz] ||= []
data[:foo][:bar][:baz] << 99
使用哈希自动vivization:
data = Hash.new { |h, k| h[k] = h.dup.clear }
#⇒ {}
# or, credits to @Amadan:
data = Hash.new { |h, k| h[k] = Hash.new(&h.default_proc) }
#⇒ {}
data[:foo][:bar][:baz] = 42
data
#⇒ {:foo=>{:bar=>{:baz=>42}}}
这里使用的技巧是我们使用Hash#default_proc
来创建嵌套键。
对于您的情况:
(data[:foo][:bar][:baz] = []) << 99
可以使用递归。
def stuff_it(h, first_key, *rest_keys, val)
if rest_keys.empty?
(h[first_key] ||= []) << val
else
h[first_key] = stuff_it(h[first_key] ||= {}, *rest_keys, val)
end
h
end
stuff_it({ a: 1 }, :foo, :bar, :baz, 99)
#=> {:a=>1, :foo=>{:bar=>{:baz=>[99]}}}
stuff_it({ a: 1, foo: { b: 2 } }, :foo, :bar, :baz, 99)
#=> {:a=>1, :foo=>{:b=>2, :bar=>{:baz=>[99]}}}
stuff_it({ a: 1, foo: { b: 2, bar: { c: 3 } } }, :foo, :bar, :baz, 99)
#=> {:a=>1, :foo=>{:b=>2, :bar=>{:c=>3, :baz=>[99]}}}
h = { a: 1, foo: { b: 2, bar: { c: 3, baz: [88] } } }
stuff_it(h, :foo, :bar, :baz, 99)
#=> {:a=>1, :foo=>{:b=>2, :bar=>{:c=>3, :baz=>[88, 99]}}}
h # => {:a=>1, :foo=>{:b=>2, :bar=>{:c=>3, :baz=>[88, 99]}}}
从上一个例子中可以看出,该方法具有破坏性。可以通过进行小的更改来使其具有非破坏性。
def stuff_it(g, first_key, *rest_keys, val)
h = g.merge(g)
if rest_keys.empty?
h[first_key] = h[first_key] ? h[first_key].dup << val : [val]
else
h[first_key] = stuff_it(h[first_key] ||= {}, *rest_keys, val)
end
h
end
h = { a: 1, foo: { b: 2, bar: { c: 3 } } }
stuff_it(h, :foo, :bar, :baz, 99)
#=> {:a=>1, :foo=>{:b=>2, :bar=>{:c=>3, :baz=>[99]}}}
h #=> { a: 1, foo: { b: 2, bar: { c: 3 } } }
h = { a: 1, foo: { b: 2, bar: { c: 3, baz: [88] } } }
stuff_it(h, :foo, :bar, :baz, 99)
#=> {:a=>1, :foo=>{:b=>2, :bar=>{:c=>3, :baz=>[88, 99]}}}
h #=> {:a=>1, :foo=>{:b=>2, :bar=>{:c=>3, :baz=>[88]}}}
你也可以做这样的事情:
class SpecialHash < Hash
def [](key)
if has_key?(key)
super(key)
else
self[key] = self.class.new
end
end
end
h = SpecialHash.new
h[:foo][:bar][:baz] = "Baz"
h # => {:foo=>{:bar=>{:baz=>"Baz"}}}
它的鸭子类型与Hash
完全相同。
您可以重新格式化与以下代码相同的代码:
class SpecialHash < Hash
def [](key)
return super if has_key?(key)
self[key] = self.class.new
end
end
甚至
class SpecialHash < Hash
def [](key)
has_key?(key) ? super : self[key] = self.class.new
end
end