如何在rails中获取关联名称？

Question

我的 user.rb 模型文件中有以下关联代码

class User < ActiveRecord::Base
  has_many :sent_messages, class_name: 'ChatMessage', foreign_key: 'sender_id'
  has_many :received_messages, class_name: 'ChatMessage', foreign_key: 'receiver_id'
end

我想要 ChatMessage 模型中的一个方法，该方法应由以下触发

current_user.sent_messages
current_user.received_messages

该方法应该返回被调用的关联的名称。 例如：

class ChatMessage < ActiveRecord::Base
  after_find :get_association_name
  def get_association_name
    self.association_name // this should return sent_message or received_message depending on whether current_user.sent_messages or current_user.received_messages was called
  end
end

有没有办法在 Rails 中获取此关联名称？ 任何帮助深表感谢。谢谢

Answer 1

我不确定你在找什么，但是

CurrentUser.reflect_on_all_associations(:has_many)

将给出一个包含所有 has_many 关联的数组。

Answer 2

我没有在这种情况下使用 AR 关联扩展，但你应该可以这样做：

has_many :sent_messages, class_name: 'ChatMessage', foreign_key: 'sender_id' do
  def get_association_name; 'sent_messages'; end
  # or, to make this more generic,
  # def get_association_name; proxy_association.reflection.name.to_s; end
end

并且该方法应该可以从您的关系中访问。如果您使用的是 Rails 4，您可以将通用版本提取到一个单独的模块中，以更简洁地扩展您的关联。见http://guides.rubyonrails.org/association_basics.html#association-extensions。

编辑

试试：

has_many :sent_messages, class_name: 'ChatMessage', foreign_key: 'sender_id' do
  def and_set_type
    proxy_association.target.each do |msg|
      msg.update_attribute(:type, 'sent')
    end
    scoped
  end
end

然后使用current_user.sent_messages.and_set_type 访问您的sent_messages。