0

我在我的应用程序中为两种不同类型的用户使用设计。他们被称为用户和专业人员。

我目前有一个简单的基于资源的控制器,称为 MessagesController,它可以像这样为当前专业人士提取消息

class MessagesController < ApplicationController
    def index
       @messages = Message.find_all_by_profession_id(current_professional.id)
    end
end

我想找到保留此控制器的最佳方式,但根据登录的用户类型更改查询。我希望资源中的所有操作(索引、新建、创建、更新等)都发生相同的情况

我知道我能做到

if current_user
    @messages = Message.find_all_by_user_id(current_user.id)
else
    @messages = Message.find_all_by_profession_id(current_professional.id)
end

但这对所有操作来说都是庞大而混乱的。我相信一定有更好的方法。这样做最像rails的方式是什么?我应该创建一个全新的控制器来处理基于用户的消息吗?

4

2 回答 2

1

我可以想到两种方法:

您可以将代码放入initialize控制器的方法中:

def initialize
  if current_user
      @messages = Message.find_all_by_user_id(current_user.id)
  else
      @messages = Message.find_all_by_profession_id(current_professional.id)
  end
  super
end

或者您可以创建一个before_filter

class MessagesController < ApplicationController
  before_filter :get_messages
  private
  def get_messages
     if current_user
       @messages = Message.find_all_by_user_id(current_user.id)
     else
       @messages = Message.find_all_by_profession_id(current_professional.id)
     end
  end
end
于 2012-12-27T10:29:57.483 回答
1

恕我直言,我认为您可以将这段代码移动到模型中,因此控制器只调用传递用户参数并从模型中获取所有消息。

# messsages_controller.rb
@messages = Message.find_all_messages(current_user, current_professional)

# message.rb
def self.find_all_messages(user, professional)
  if user
    self.find_all_by_user_id(user.id)
  else
    self.find_all_by_profession_id(professional.id)
  end
end

我认为最好在您的模型上使用这种代码。当然,您可以改进 if/else 代码,但我现在没有想法。

于 2012-12-27T10:30:19.213 回答