有没有办法在 Activerecord 中为单个查询创建临时列？

Question

假设我们有一个具有以下迁移的 ChessPiece 类。

create_table :chess_pieces do |t|
  t.string  :type
  t.integer :row
  t.integer :column
  t.integer :game_id, index: true
end

如果 Bishop 想从 (0,0) 移动到 (7,7)，那么我需要检查两个位置之间的每个空间是否有其他可能阻挡我的 Bishop 的棋子。 IE 如果 (5,5) 上有一个 Pawn。

我可以在数据库中查询每个可能有阻塞块的位置，如下所示：

pieces = []
8.times do |i|
  pieces << ChessPiece.where(game_id: @game_id, row: i, column: i)
end

但我想减少查询。 或者，我可以抓住一个游戏的所有棋子并在 Ruby 中迭代它们，但这似乎也很低效。 我想要做的是告诉数据库将row和column列组合在一起形成“行列”并将其称为position 。 这样我就可以运行类似下面的查询：

ChessPiece.where(game_id: @game_id, position: ["1-1", "2-2", "3-3"])

我如何动态创建这个新列以便我可以运行类似上述查询的内容？ （假设我不能只编辑数据库模式。）

Answer 1

您可以使用.select创建您想要的任何选择语句：

@chess_pieces = ChessPiece.select(
  "chess_pieces.*", 
  "CONCAT(chess_pieces.row, '-', chess_pieces.column) AS position"
)

当您使用别名时，结果中的列将作为模型的属性提供：

@chess_pieces.first.position

您还可以在某些（PG、MySQL）但不是所有数据库的 WHERE、ORDER 和 GROUP 子句中使用列别名：

ChessPiece.select(
  "chess_pieces.*", 
  "CONCAT(chess_pieces.row, '-', chess_pieces.column) AS position"
).where('position = ?', '1-1')
 .group('position')
 .order('position ASC')

对于那些不支持它的人，你需要连接、聚合或任何你正在做的事情。

ChessPiece.order("CONCAT(chess_pieces.row, '-', chess_pieces.column)")

Answer 2

为什么不将其设为永久专栏？

添加一个整数类型的位置列。 创建一个 before save 挂钩，将位置设置为row * 10 + column

您现在可以查询位置列

ChessPiece.where(game_id: @game_id, position: [11, 22, 33])

如果你愿意，你也可以覆盖 position= 来设置列和行。 允许您也将其用于写入。

Answer 3

您可以向您的ChessPiece模型添加一个助手来简化正确查询的构建。 这样的事情应该可以解决问题：

class ChessPiece < ApplicationRecord
  def self.matches_position(*positions)
    return none if positions.empty?

    constraints = positions.map do |row, column|
      arel_table[:row].eq(row).and(arel_table[:column].eq(column))
    end

    where(constraints.reduce(:or))
  end
end

ChessPiece.where(game_id: @game_id).matches_position([1, 1], [2, 2], [3, 3])

这应该会生成 SQL（语法可能因所使用的数据库而异）：

SELECT "chess_pieces".*
FROM "chess_pieces"
WHERE "chess_pieces"."game_id" = 1
  AND ("chess_pieces"."row" = 1 AND "chess_pieces"."column" = 1
       OR "chess_pieces"."row" = 2 AND "chess_pieces"."column" = 2
       OR "chess_pieces"."row" = 3 AND "chess_pieces"."column" = 3)