重構一系列值的條件

Question

如何重構Ruby on Rails代碼的這一點？

def select_plan 
     unless params[:plan] && (params[:plan] == '1' || params[:plan] == '2' || params[:plan] == '3' || params[:plan] == '4' || params[:plan] == '5' || params[:plan] == '6' || params[:plan] == '7' || params[:plan] == '8') 
      flash[:notice] = "Please select a membership plan to register." 
      redirect_to root_url 
     end 
    end 

Answer 1

如果計劃數是在數據庫中有一個Plan模型，那麼你可以簡單地這樣說：

@plan = Plan.find_by(:id => params[:plan]) 
if(!@plan) 
    flash[:notice] = "Please select a membership plan to register." 
    redirect_to root_url 
end 

現在您可以訪問下一個視圖的完整計劃詳細信息，以便您可以向他們顯示名稱，說明，價格......以及計劃的存在僅存儲在一個位置（即數據庫）中。如果你不需要@plan，那麼你可以說：

if(!Plan.where(:id => params[:plan]).exists?) 
    ... 
end 

重要的一點是，應該只有一兩件事，知道有效的計劃是什麼，你需要知道的關於計劃的任何時候，你問這個一件事，只有一件事。

是最終調用select_plan也將使用數據庫（而不是字面值數字一至八），以獲得有效的計劃的看法：

<% Plan.order(...).each do |p| %> 
    whatever you need to display the plan as an option... 
<% end %> 

添加一個新的計劃，以數據庫和一切仍然有效。刪除/禁用計劃，一切仍然有效。您的軟件將更容易維護，缺陷更少，更容易理解，並且您將獲得新的好習慣而不是壞習慣。

Answer 2

我會這樣做。

def select_plan 
    unless params[:plan].in?('1'..'8') 
    flash[:notice] = "Please select a membership plan to register." 
    redirect_to root_url 
    end 
end 

或者像畝太短建議：請Plan一個真實的東西。這可能是一個數據庫模型或只是一個小Ruby類：

# in app/models/plan.rb 
require 'set' 
class Plan 
    VALID_PLANS = Set.new('1'..'8').freeze 

    def self.valid_plan?(plan) 
    VALID_PLANS.include?(plan) 
    end 
end 

# used like 
def select_plan 
    unless Plan.valid_plan?(params[:plan]) 
    flash[:notice] = "Please select a membership plan to register." 
    redirect_to root_url 
    end 
end