使用Hash #dig或Lonely运算符(&.)安全地为嵌套哈希值赋值

Question

h = {
  data: {
    user: {
      value: "John Doe" 
    }
  }
}

要为嵌套哈希值赋值,我们可以使用

h[:data][:user][:value] = "Bob"

但是,如果缺少中间的任何部分,则会导致错误.

就像是

h.dig(:data, :user, :value) = "Bob"

不起作用,因为还没有Hash#dig=.

为了安全地分配价值,我们可以做到

h.dig(:data, :user)&.[]=(:value, "Bob")    # or equivalently
h.dig(:data, :user)&.store(:value, "Bob")

但有更好的方法吗？

Answer 1

它并非没有它的警告(如果你从其他地方收到哈希值,它就不起作用),但一个常见的解决方案是:

hash = Hash.new {|h,k| h[k] = h.class.new(&h.default_proc) }

hash[:data][:user][:value] = "Bob"
p hash
# => { :data => { :user => { :value => "Bob" } } }

Answer 2

并以@rellampec 的回答为基础，这些回答不会引发错误：

def dig_set(obj, keys, value)
  key = keys.first
  if keys.length == 1
    obj[key] = value
  else
    obj[key] = {} unless obj[key]
    dig_set(obj[key], keys.slice(1..-1), value)
  end
end

obj = {d: 'hey'}
dig_set(obj, [:a, :b, :c], 'val')
obj #=> {d: 'hey', a: {b: {c: 'val'}}}