Рекурсивное объединение многомерных массивов, хэшей и символов - PullRequest
Новая
       24

Рекурсивное объединение многомерных массивов, хэшей и символов

1 голос
/ 17 февраля 2010

Мне нужен кусок кода Ruby для объединения массива содержимого, такого как: 

[{:dim_location=>[{:dim_city=>:dim_state}]},
 :dim_marital_status, 
 {:dim_location=>[:dim_zip, :dim_business]}]

в: 

[{:dim_location => [:dim_business, {:dim_city=>:dim_state}, :dim_zip]},
 :dim_marital_status]

Он должен поддерживать произвольный уровень глубины, хотяглубина редко превышает 8 уровней.

Ответы [ 2 ]

1 голос
/ 17 февраля 2010

Пересмотрено после комментария: 

source = [{:dim_location=>[{:dim_city=>:dim_state}]}, :dim_marital_status, {:dim_location=>[:dim_zip, :dim_business]}]

expected = [{:dim_location => [:dim_business, {:dim_city=>:dim_state}, :dim_zip]}, :dim_marital_status]

source2 = [{:dim_location=>{:dim_city=>:dim_state}}, {:dim_location=>:dim_city}]  

def merge_dim_locations(array)
  return array unless array.is_a?(Array)
  values = array.dup
  dim_locations = values.select {|x| x.is_a?(Hash) && x.has_key?(:dim_location)}
  old_index = values.index(dim_locations[0]) unless dim_locations.empty?
  merged = dim_locations.inject({}) do |memo, obj| 
    values.delete(obj)
    x = merge_dim_locations(obj[:dim_location])
    if x.is_a?(Array) 
      memo[:dim_location] = (memo[:dim_location] || []) + x
    else
      memo[:dim_location] ||= []
      memo[:dim_location] << x
    end
    memo
  end
  unless merged.empty?
    values.insert(old_index, merged)
  end
  values
end


puts "source1:"
puts source.inspect
puts "result1:"
puts merge_dim_locations(source).inspect
puts "expected1:"
puts expected.inspect

puts "\nsource2:"
puts source2.inspect
puts "result2:"
puts merge_dim_locations(source2).inspect
0 голосов
/ 17 февраля 2010

Не думаю, что в вашем вопросе достаточно подробностей, чтобы дать вам полный ответ, но это может помочь вам начать: 

class Hash
  def recursive_merge!(other)
    other.keys.each do |k|
      if self[k].is_a?(Array) && other[k].is_a?(Array)
        self[k] += other[k]
      elsif self[k].is_a?(Hash) && other[k].is_a?(Hash)
        self[k].recursive_merge!(other[k])
      else
        self[k] = other[k]
      end
    end
    self
  end
end
...