Как продолжить рефакторинг рубинового хэша

Question

Я пишу функцию, которая принимает строку и возвращает соответствующий класс модели.Старая версия состоит из уродливых операторов case, а реорганизованная версия имеет менее уродливый хеш.Однако хэш все еще кажется мне повторяющимся.Не могли бы вы дать мне какой-нибудь совет?

# original function

def determine_node_label(category)
  label = 
    case category 
    when 'boundary box'
      Slide
    when 'circle', 'rectangle'
      GroupBox
    when 'text'
      Text
    when 'picture', 'pie', 'bar' , 'trend', 'star'
      Content
    else
      Content
    end
  return label 
end

# refactored function

def determine_node_label(category)
  label = {
    "boundary box" => Slide,
    "circle" => GroupBox,
    "rectangle" => GroupBox,
    "text" => Text,
    "picture" => Content,
    "pie" => Content,
    "bar" => Content,
    "trend" => Content,
    "star" => Content
  }
  label.default = Content
  return label["category"]
end

обновление:

Меня больше интересуют решения, предполагающие, что label.default может быть изменен.Приношу свои извинения за то, что не указал это в коде.

Answer 1

Если вам нужно динамическое значение по умолчанию, вы можете использовать Hash.fetch.Также передайте значение по умолчанию в качестве параметра метода.

LABELS = {
  "boundary box" => Slide,
  "circle"       => GroupBox,
  "rectangle"    => GroupBox,
  "text"         => Text
} 

def determine_node_label(category, default = 'Content')
  LABELS.fetch(category, default)
end

Answer 2

Для удобства сопровождения я предлагаю вести данные в следующем хеше:

DATA = {
  %w| boundary\ box |              => 'Slide',
  %w| circle rectangle |           => 'GroupBox',
  %w| text |                       => 'Text',
  %w| picture pie bar trend star | => 'Content'
}
  #=> {["boundary box"]=>"Slide", ["circle", "rectangle"]=>"GroupBox",
  #    ["text"]=>"Text", ["picture", "pie", "bar", "trend", "star"]=>"Content"}

Обратите внимание, что я сделал значения литералами (строками), чтобы продемонстрировать, как будет обрабатываться этот хеш.В реальном приложении значения не обязательно должны быть литералами.

Затем предоставьте метод для создания желаемого хеша h из DATA и указанного значения по умолчанию (последнее возвращается h[k], когдаh не имеет ключа k).

def data_to_hash(data, default)
  data.each_with_object({}) { |(k,v),h| k.each { |obj| h[obj] = v } }.
       tap { |h| h.default = default }
end

Это может использоваться следующим образом.

h = data_to_hash(DATA, 'Cat')
  #=> {"boundary box"=>"Slide", "circle"=>"GroupBox",
  #    "rectangle"=>"GroupBox", "text"=>"Text", "picture"=>"Content",
  #    "pie"=>"Content", "bar"=>"Content", "trend"=>"Content",
  #    "star"=>"Content"}
h["boundary box"]
  #=> "Slide"
h["pie"]
  #=> "Content"
h["cake"]
  #=> "Cat"

Чтобы впоследствии изменить значение по умолчанию, вы можете либо позвонить data_to_hashснова с исправленным значением по умолчанию, либо просто выполните

h.default = "Dog"

, либо заключите последний в метод.

def change_default(h, new_default)
  h.default = new_default
end

change_default(h, "Dog")
  #=> "Dog"
h["pie"]
  #=> "Content"
h["cake"]
  #=> "Dog"

Answer 3

Как насчет этого?

LABELS = {
  "boundary box" => Slide,
  "circle"       => GroupBox,
  "rectangle"    => GroupBox,
  "text"         => Text
} 

def determine_node_label(category)
  LABELS[category] || Content
end