Ruby рекурсивная карта хэша объектов

Question

У меня есть хэш, который имеет неизвестную коллекцию и смесь вложенных массивов, хэшей, массивов хешей и строк. Это результат JSON.parse. Структура данных должна быть такой же, как и в начале. Конечной целью является преобразование строк в Fixnums, которые могут быть Fixnums.

Следующие работы прекрасно, но мне было интересно, можно ли сократить его. Обратите внимание, как мне нужен ключ и значение в методе clean, поскольку не все строки, которые могут быть Fixnums, должны быть. Есть идеи?

def clean_node(node) 
    if node.class == String 
     clean(node) 
    elsif node.class == Array 
     node.each_with_index do |obj, i| 
     if obj.class == String 
      node[i] = clean(node[i], obj) 
     else 
      clean_node(obj) 
     end 
     end 
    elsif node.class == Hash 
     node.each_pair do |key, value| 
     if value.class == String 
      node[key] = clean(key, value) 
     else 
      clean_node(value) 
     end 
     end 
    end 
end 

def clean(key, value) 
    FIXNUM_KEYS.include?(key)? value.to_i : value 
end 

Answer 1

Хотя я не смотрел в рекурсии, я должен прокомментировать тот факт, что вы пишете, если заявления, которые были бы легче читать в сазе:

def clean_node(node) 
    case node 
    when String then clean(node) 
    when Array 
     node.each_with_index do |obj, i| 
     case obj 
      when String 
      node[i] = clean(node[i], obj) 
      else 
      clean_node(obj) 
     end 
     end 
    when Hash.... 

Answer 2

Там нет необходимости чтобы разделить обработку строк, вы можете сделать все это в одной рекурсивной подпрограмме, если вы вернете значение node в конце процедуры. Это имеет побочный эффект удаления некоторых параметров, и вы можете использовать на месте .map! для обработки массивов.

Использование case делает выбор по типу немного более легким для чтения.

Добавление способности справляться с массивами строк может быть осуществлено множеством способов. Я решил добавить переменную состояния (общий шаблон в рекурсии, например, для подсчета глубины). Затем в случае массивов рекурсия наследует состояние с текущего уровня, тогда как хэш-регистр определяет новое состояние (для преобразования или нет) на основе поиска в списке ключей для применения преобразований.

def clean_node(node, convert_item = false) 

    case node 

    when String 
     node = node.to_i if convert_item 

    when Array 
     node.map! do |obj| 
     clean_node(obj, convert_item) 
     end 

    when Hash 
     node.each_pair do |key, value| 
     node[key] = clean_node(value, FIXNUM_KEYS.include?(key)) 
     end 
    end 

    node 
end