ruby-on-rails 如何从多态关联中的关联模型中提取不同的列值?

7lrncoxx  于 2023-04-08  发布在  Ruby
关注(0)|答案(4)|浏览(158)

如果我有3个模型之间的多态关联:
评论

belongs_to :book, :class_name => 'Book', :foreign_key => 'ref_id', conditions: "comments.ref_type = 'Book'"
belongs_to :article, :class_name => 'Article', :foreign_key => 'ref_id', conditions: "comments.ref_type = 'Article'"
belongs_to :ref, :polymorphic => true

对于给定的注解列表,如何从BookArticle模型的Title列中选择不同的值?
例如,如果我必须列出在某个时间段内有评论的书籍和文章的标题,那么我该怎么做呢?我可以很容易地选择评论列表,但我如何从BookArticle中选择相关的唯一标题呢?
例如:

Book
+--------------+
| Id |  Title  |
+----+---------+
| 1  | 'Book1' | 
| 2  | 'Book2' |
| 3  | 'Book3' |
+--------------+

Article
+-----------------+
| Id |   Title    |
+----+------------+
| 1  | 'Article1' |
| 2  | 'Article2' |
+-----------------+

Comments
+--------------------------------------+
| Id |  comment   | ref_id | ref_type  |
+----+------------+--------+-----------+
| 1  | 'comment1' |   1    |   Book    | 
| 2  | 'comment2' |   1    |   Book    | 
| 3  | 'comment3' |   1    |   Article | 
| 4  | 'comment4' |   3    |   Book    | 
+--------------------------------------+

我需要的标题列表是'Book1''Book3''Article1'

ddrv8njm

ddrv8njm1#

简单方法:

Comment.all.includes(:ref).map { |comment| comment.ref.title }.uniq

获取所有注解,eager加载它们的引用,并返回一个包含唯一标题的数组。eager加载部分不是严格必要的,但它可能会执行得更好。执行3个查询,一个针对注解,一个针对每种引用。您可以将 all 替换为任何范围。请注意,这将获取所有注解及其引用,并使用ruby将其转换为数组,而不是SQL。这非常有效。但是性能可能会受到影响。通常最好使用 distinct 来获得唯一值,然后使用 pluck 来获得这些值的数组。
这种方法适用于所有类型的ref。因此,如果我们引入第三种ref,例如 Post,它将自动包含在此查询中。

复用方式:

class Comment < ApplicationRecord
  belongs_to :book, class_name: 'Book', foreign_key: 'ref_id'
  belongs_to :article, class_name: 'Article', foreign_key: 'ref_id'
  belongs_to :ref, polymorphic: true

  scope :with_ref_titles, lambda {
    book_titles = select('comments.*, books.title').joins(:book)
    article_titles = select('comments.*, articles.title').joins(:article)
    union = book_titles.union(article_titles)
    comment_table = arel_table
    from(comment_table.create_table_alias(union, :comments).to_sql)
  }
end

此作用域使用arel和子查询的UNION来获取标题。它基本上将引用的标题添加到注解对象。由于作用域应该是可链接的,因此它返回ActiveRecord Relation而不是数组。要获取不同的标题,请追加 distinct.pluck(:title)

comments = Comment.with_ref_titles
comments.distinct.pluck(:title) # => ["Article1", "Book1", "Book3"]
comments.where('title LIKE "Book%"').distinct.pluck(:title) # => ["Book1", "Book3"]

此作用域生成的SQL查询如下所示:

SELECT DISTINCT "title" FROM ( SELECT comments.*, books.title FROM "comments" INNER JOIN "books" ON "books"."id" = "comments"."ref_id" UNION SELECT comments.*, articles.title FROM "comments" INNER JOIN "articles" ON "articles"."id" = "comments"."ref_id" ) "comments"

使用rails 5.1.2和sqlite进行测试。

vxf3dgd4

vxf3dgd42#

我认为最简单的方法是在BookArticle上定义.titles_for_comments作用域,当给定comments的集合时,可以从每个表中选择不同的标题,而不直接使用Arel API或迭代数组。然后为Comment定义一个.distinct_titles作用域,该作用域使用[Book|Article].titles_for_comments。例如,下面的模型和作用域定义允许您通过调用Comment.distinct_titles为任何给定的Comment::ActiveRecord_Relation示例查找不同的Book和Article标题

class Book < ActiveRecord::Base
  has_many :comments, as: :ref

  def self.titles_for_comments(comment_ids)
    joins(:comments).where(comments: { id: comment_ids }).distinct.pluck(:title)
  end
end

class Article < ActiveRecord::Base
  has_many :comments, as: :ref

  def self.titles_for_comments(comment_ids)
    joins(:comments).where(comments: { id: comment_ids }).distinct.pluck(:title)
  end
end

class Comment < ActiveRecord::Base
  belongs_to :ref, polymorphic: true

  def self.distinct_titles
    comment_ids = ids
    Article.titles_for_comments(comment_ids) + Book.titles_for_comments(comment_ids)
  end
end

您可以下载此Gist并使用ruby polymorphic_query_test.rb运行它,测试应该通过https://gist.github.com/msimonborg/907eb513fdde9ab48ee881d43ddb8378

cnwbcb6i

cnwbcb6i3#

试试这个

titles_from_books = Comment.joins('INNER JOIN books on comments.ref_id = books.id').where('comments.ref_type = ?','Book').pluck('books.title')

titles_from_articles = Comment.joins('INNER JOIN articles on comments.ref_id = article.id').where('comments.ref_type = ?','Article').pluck('articles.title')

final_titles = (titles_from_books + titles_from_articles).uniq
0vvn1miw

0vvn1miw4#

我建议你看看这个教程:https://web.archive.org/web/20180928102044/http://karimbutt.github.io:80/blog/2015/01/03/step-by-step-guide-to-polymorphic-associations-in-rails/,看看它是否适合你。
如果您控制模型代码,我会将注解设置为belong_to一个通用的可注解对象。示例:

class Comment < ActiveRecord::Base
  belong_to :commentable, :polymorphic => true
end

然后呢

class Book < ActiveRecord::Base
  has_many :comments, as: commentable
end

class Article < ActiveRecord::Base
  has_many :comments, as: commentable
end

然后给出任何一组注解

Comments.each do |comment|
  comment.commentable.title
end.uniq

目前,仅仅获取标题似乎需要做很多工作,但如果你坚持这个项目,我希望书籍和文章能够分享 * 很多 * 这种类型的代码,以及将来可能添加其他可注解的对象。总体而言,拥有一个通用对象将保存大量工作。

相关问题