在Ruby on Rails中重构一些模型方法

时间:2015-01-06 14:06:21

标签: ruby-on-rails

我的模型中有一些方法用于访问视图中存储的文本哈希值:

class CarSpec < ActiveRecord::Base

    @@fuel_type_select_data = Hash[(1..5).to_a.zip(['Petrol', 'Diesel', 'Gas/petrol', 'Hybrid', 'Electric'])]
    @@mileage_type_select_data = Hash[(1..2).to_a.zip(['km', 'miles'])]
    @@transmission_select_data = Hash[(1..3).to_a.zip(['Manual', 'Automatic', 'Tiptronic'])]
    @@wheel_drive_data = Hash[(1..3).to_a.zip(['Front', 'Rear', 'All'])]
    @@color_data = Hash[(1..8).to_a.zip(['black', 'white', 'beige', 
              'blue', 'yellow', 'green', 'red', 'silver'])]

    def text_for_fuel_type
       @@fuel_type_select_data[fuel_type] 
    end

    def text_for_mileage_type
       @@mileage_type_select_data[mileage_type] 
    end

    def text_for_transmission
       @@transmission_select_data[transmission] 
    end

    def text_for_wheel_drive
       @@wheel_drive_data[wheel_drive] 
    end

    def text_for_color
       @@color_data[color]
    end

    def text_for_interior_color
       @@color_data[interior_color] 
    end

目前,我需要为每个字段编写一个新方法。如何重构这些方法,以便我不需要为每个字段编写新方法?请包括如何在视图中调用新方法。

2 个答案:

答案 0 :(得分:1)

在你的情况下,使用常量而不是类变量会好得多。

一个例子:

class CarSpec < ActiveRecord::Base
  WHEEL_DRIVE_DATA = {'Front' => 1}

  # remaining class code
end

查看示例代码:

<%= CarSpec::WHEEL_DRIVE_DATA['Front'] %>

答案 1 :(得分:1)

同意@Humza:常量优于类变量。

当前的一组方法可以动态定义 - 使用一些元编程魔法 - 如下:

fields = %w{ fuel_type mileage_type transmission wheel_drive interior_color }
fields.each do |field|
  define_method("text_for_#{field}") do
    CarSpec.class_variable_get("@@#{field}_select_data").send(:[], eval(field))
  end 
end 

注意: 为了使上述工作,所有字段的类变量必须与@@#{field}_select_data名称一致;因此@@wheel_drive_data应更改为@@wheel_drive_select_data等。