如何在 Ruby 哈希中设置嵌套键的动态值



这应该很容易,但我找不到合适的解决方案。对于第一级键:

resource.public_send("#{key}=", value)

但对于foo.bar.lolo

我知道我可以像下面这样得到它:

'foo.bar.lolo'.split('.').inject(resource, :send)

resource.instance_eval("foo.bar.lolo")

但是如何将值设置为最后一个变量,假设我不知道嵌套级别,它可能是第二个或第三个。

有没有适用于所有级别的通用方法?对于我的例子,我可以像下面这样做:

resource.public_send("fofo").public_send("bar").public_send("lolo=", value)

哈希的答案,只是出于好奇:

hash = { a: { b: { c: 1 } } }
def deep_set(hash, value, *keys)
  keys[0...-1].inject(hash) do |acc, h|
    acc.public_send(:[], h)
  end.public_send(:[]=, keys.last, value)
end
deep_set(hash, 42, :a, :b, :c)
#⇒ 42
hash
#⇒ { a: { b: { c: 42 } } }

Ruby 中的哈希默认不会给你这些点方法。

你可以链接发送调用(这适用于任何对象,但你不能以这种方式正常访问哈希键):

  "foo".send(:downcase).send(:upcase)

当使用嵌套哈希时,可变性的棘手概念是相关的。例如:

  hash = { a: { b: { c: 1 } } }
  nested = hash[:a][:b]
  nested[:b] = 2
  hash
  # => { a: { b: { c: 2 } }

这里的"可变性"意味着当您将嵌套哈希存储到一个单独的变量中时,它实际上仍然是指向原始哈希的指针。可变性对于这种情况很有用,但如果您不了解它,它也会产生错误。

您可以将:a:b分配给变量,使其在某种意义上"动态"。

有更高级的方法可以做到这一点,例如在较新的Ruby中dig

versions.
  hash = { a: { b: { c: 1 } } }
  keys_to_get_nested_hash = [:a, :b]
  nested_hash = hash.dig *keys_to_get_nested_hash
  nested_hash[:c] = 2
  hash
  # => { a: { b: { c: 2 } } }

如果您使用 OpenStruct那么您可以为您的哈希提供点方法访问器。老实说,链接send调用并不是我经常使用的东西。如果它能帮助你编写代码,那就太好了。但是您不应该发送用户生成的输入,因为它不安全。

尽管您可以实现一些方法以现在设置的方式执行操作,但我强烈建议您重新考虑数据结构。

为了澄清您的一些术语,示例中的key不是键,而是方法调用。在 Ruby 中,当你有像 my_thing.my_other_thing 这样的代码时,my_other_thing总是一个方法,而不是一个键,至少不是正确意义上的术语。

的确,您可以通过以这种方式链接对象来创建类似哈希的结构,但这确实有代码气味。如果您认为foo.bar.lolo是在哈希中查找嵌套lolo键的一种方式,那么您可能应该使用常规哈希。

x = {foo: {bar: 'lolo'}}
x[:foo][:bar] # => 'lolo'
x[:foo][:bar] = 'new_value' # => 'new_value'

此外,尽管可以通过这种方式使用发送/instance_eval方法,但这不是最佳实践,甚至会产生安全问题。

如果你想允许初始化缺少的嵌套键,我建议对阿列克谢·马蒂乌什金的解决方案进行以下重构

我不想对实际答案进行更改,因为它完全有效,并且引入了一些额外的内容。

hash = { a: {} } # missing some nested keys
def deep_set(hash, value, *keys)
  keys[0...-1].inject(hash) do |acc, h|
    acc[h] ||= {} # initialize the missing keys (ex: b in this case) 
    acc.public_send(:[], h)
  end.public_send(:[]=, keys.last, value)
end
deep_set(hash, 42, :a, :b, :c)
#⇒ 42
hash
#⇒ { a: { b: { c: 42 } } }

假设已知密钥存在,那么Hash#dig给出了一个更干净的解决方案:

hash = { a: { b: { c: 1 } } }
def deep_set(hash, value, *keys)
  hash.dig(*keys[0..-2])[keys[-1]] = value
end
deep_set(hash, 42, :a, :b, :c)
#⇒ 42
hash
#⇒ { a: { b: { c: 42 } } }

这只是示例代码。如果密钥未知或deep_set收到的密钥少于两个,它将不起作用。这两个问题都是可以解决的,但超出了OP的问题。

最新更新