ActiveRecord - 从每个组中选择第一条记录

时间:2013-03-04 16:58:04

标签: sql ruby-on-rails-3 postgresql activerecord

我想要每个用户发送的最新消息。这是样本数据

表格:对话

sender  receiver message                        date
============================================================
    1   2   "hi"                            "2013-03-04 09:55:01.122074"
    2   1   "hello"                         "2013-03-04 09:55:32.903975"
    1   2   "have you done with the taks?"  "2013-03-04 09:57:46.383007"
    2   1   "almost..."                     "2013-03-04 09:58:55.783219"
    2   1   "should be able to finish 2day" "2013-03-04 09:59:28.950705"
    2   3   "shall we start?"               "2013-03-04 10:01:16.842725"
    3   2   "give me a minute"              "2013-03-04 10:01:41.994589"
    3   2   "let us start"                  "2013-03-04 10:02:14.04551"

对于id为2的用户,我应该可以获得以下两个记录

1   2   "have you done with the taks?"  "2013-03-04 09:57:46.383007"        
3   2   "let us start"                  "2013-03-04 10:02:14.04551"

这是我的解决方案

型号:用户

class User < ActiveRecord::Base
has_many :chats_received, class_name: 'Conversation', foreign_key: 'receiver_id',order: "created_at DESC"
end

型号:对话

class Conversation < ActiveRecord::Base
  attr_accessible :message, :read

  belongs_to :sender, class_name: 'User'
  belongs_to :receiver, class_name: 'User'

  def most_recent_chat_received_from_connected_users
    connected_users_chats = . . . # Get all conversations which are sent to current user. e.g., user with id 2
    chats_grouped_by_senders = connected_users_chats.group_by { |conversation| conversation.sender_id }
    chats_grouped_by_senders.inject([]){|memo , (sender_id, conversations)| memo << conversations.first; memo}
  end
end

从关联用户那里获取最新消息:

user = User.find 2
user.most_recent_chat_received_from_connected_users

虽然此解决方案有效,但它为两个用户之间的每次转换选择并创建模型。此外,我觉得这不是获取所需行的方式。

我一直在使用postgresql。当我尝试在模式上使用组方法时,我遇到了以下错误。

ActiveRecord::StatementInvalid: PG::Error: ERROR:  column "conversations.id" must appear in the GROUP BY clause or be used in an aggregate function

有没有更好的方法来获得相同的结果?

1 个答案:

答案 0 :(得分:7)

我不确定如何在most_recent_chat_received_from_connected_users的实例上调用Conversation,这是User类的实例方法,而不会出现错误,但我会添加Conversation模型的自定义查找器:

class Conversation < ActiveRecord::Base
  # ...

  def self.most_recent_for(user_id)
    select('DISTINCT ON (sender_id) *').where(reciever_id: user_id).order("sender_id, created_at DESC")
  end

  # For MySQL you could have used:
  #
  # def self.most_recent_for(user_id)
  #   where(reciever_id: user_id).group("sender_id").order("created_at DESC")
  # end

  # ...
end

现在,您可以通过以下方式在控制器中获得所需的对话:

@conversations = Conversation.most_recent_for(current_user.id)