Ruby 複雑なhashをmergeする方法を知りたい

Question

前提・実現したいこと

配列を含むhashをmergeし、出力結果のようになるにはどのようなコードになるでしょうか。
解答お願いいたします。

コード

obj1 = { a: 3,  b: { c: 1, d: [ 6, 3 ], e: 2 } }
obj2 = { a: 1, b: { c: 2, d: [ 2, 0, 4 ], f: 1 } }
obj3 = { b: { d: [ 1 ] }　}

def merge(*object)
 #ここコードを知りたい
end

望ましい出力結果

{  a: 1,
  b: {
    c: 2,
    d: [6, 2, 1, 3, 0, 4],
    e: 2,
    f: 1  }
}

補足情報

＊ライブラリは無使用でお願いします

Comments

[asm]

配列を結合したときの順番が
d: [6, 3, 2, 0, 4, 1]
ならば割と簡単なのですが、並び替えは可能でしょうか？

[onoderamasuo]

結合後の並び替えは可能です。

[otn]

えー、Arrayは順不同だったの？
順番が肝かと思ったのに。

Answer 1

Hash#mergeにはブロックを渡せますので

def merge(*object)
  x = object.shift
  func = proc do |key, old_val, new_val|
    case old_val
    when Hash
      old_val.merge(new_val, &func)
    when Array
      old_val + new_val
    else
      new_val
    end
  end
  x.merge(*object, &func)
end

p merge(obj1, obj2, obj3)
# => {:a=>1, :b=>{:c=>2, :d=>[6, 3, 2, 0, 4, 1], :e=>2, :f=>1}}

が可能です。

Comments

[onoderamasuo]

ありがとうございます。
とても勉強になりました。

Answer 2

obj1 = { a: 3,  b: { c: 1, d: [ 6, 3 ], e: 2 } }
obj2 = { a: 1, b: { c: 2, d: [ 2, 0, 4 ], f: 1 } }
obj3 = { b: { d: [ 1 ] } }

def merge(*object)
    {}.merge(*object) do |k, v1, v2|
        if Array===v1 and Array===v2
            if v1.size < v2.size
                v1 << [nil] * (v2.size-v1.size)
            end
            v1.zip(v2).flatten.compact
        elsif Hash===v1 and Hash===v2
            merge(v1,v2)
        else
            v2
        end
    end
end

p merge(obj1, obj2, obj3)
#=> {:a=>1, :b=>{:c=>2, :d=>[6, 1, 2, 3, 0, 4], :e=>2, :f=>1}}

で良いかと思ったのですが、結果が、

{:a=>1, :b=>{:c=>2, :d=>[6, 1, 2, 3, 0, 4], :e=>2, :f=>1}}

になります。
:d=>[6, 2, 1, 3, 0, 4]にするためには、Arrayのマージをハッシュのマージが全部終わってから行う必要があるので、かなり面倒ですが、興味がわいたのでやってみました。

obj1 = { a: 3,  b: { c: 1, d: [ 6, 3 ], e: 2 } }
obj2 = { a: 1, b: { c: 2, d: [ 2, 0, 4 ], f: 1 } }
obj3 = { b: { d: [ 1 ] } }

class Foo
    def initialize(x)
        @x = [x]
    end
    def add(x)
        @x << x
        self
    end
    def merge
        l = @x.map(&:size).max
        ([nil]*l).zip(*@x).flatten.compact
    end
end

def merge_sub(*object)
    {}.merge(*object) do |k, v1, v2|
        if Array===v1 and Array===v2
            Foo.new(v1).add(v2)
        elsif Foo===v1 and Array===v2
            v1.add(v2)
        elsif Hash===v1 and Hash===v2
            merge_sub(v1,v2)
        else
            v2
        end
    end
end

def transform_sub(hash)
    hash.transform_values do |v|
        case v
        when Foo
            v.merge
        when Hash
            transform_sub(v)
        else
            v
        end
    end
end

def merge(*object)
    transform_sub(merge_sub(*object))
end

p merge(obj1, obj2, obj3)
#=> {:a=>1, :b=>{:c=>2, :d=>[6, 2, 1, 3, 0, 4], :e=>2, :f=>1}}

Comments

[onoderamasuo]

興味を持っていただきありがとうございます。
とても勉強になりました。