Rails DRY 问题：控制器和视图中需要相同的代码

Question

我的页面中有以下登录检查：

class LoungeController < ApplicationController
  before_filter :confirm_logged_in
  def index
  end 
end

同时在此处定义了 confirm_logged_in：

class ApplicationController < ActionController::Base
  protect_from_forgery

  protected

  def confirm_logged_in
    return true if current_user
    redirect_to(:controller => 'access', :action => 'login')
    return false   # halts the before_filter
  end

  def current_user
    return false unless session[:user_id]
    user = User.find(session[:user_id])
    return false unless user
    (user.display_name == session[:user_display_name]) ? user : nil
  end  
end

现在，我想在 app/views/layouts/application.html 中也使用 confirm_logged_in .erb：

<% if confirm_logged_in %>
   <div id="logged_in_as">You are logged in as <%= session[:user_display_name] %></div>
   <div id="logout"><%= link_to("Logout", {:controller => "access", :action => "logout"}, :id => "logout_link") %></div>
<% end %>

您建议如何解决这个问题？我应该在哪里定义 confirm_logged_in ？

Answer 1

您可以使用 helper_method

controller.rb

helper_method :confirm_logged_in, :current_user

protected

def confirm_logged_in
  # code...
end

def current_user
  # code...
end