活动记录 - 具有多列的IN

问题描述 投票:5回答:6

我有一个查询需要从一个完全满足两列要求的表中获取。所以,如果我有users表与列,agescore

SELECT * FROM users where (age, score) IN ((5,6), (9,12), (22,44)..)

在我的网络应用程序中,我从ajax请求得到这对,并且数量可能非常大。如何为此构建Active Record查询?

我正在研究postgres数据库

ruby-on-rails postgresql activerecord
6个回答
5
投票

理想情况下,我们将根据输入构建一个查询字符串。例如

ages_and_scores = [ [5, 6], [9, 12], [22, 44] ]
query_string = ages_and_scores.map do |pair|
  "(age = #{pair[0]} AND score = #{pair[1]})"
end.join(" OR ")
# => (age = 5 AND score = 6) OR (age = 9 AND score = 12) OR (age = 22 AND score = 44)

最后,您的查询将是

User.where(query_string)

您可以更正如何构建查询字符串的逻辑,因为ages_and_scores与我的示例的格式不同。

起色

ages_and_scores = [ [5, 6], [9, 12], [22, 44] ]
query_params = []

query_template = ages_and_scores.map{ |_| "(age = ? AND score = ?)" }.join(" OR ")
# => (age = ? AND score = ?) OR (age = ? AND score = ?) OR (age = ? AND score = ?)

User.where(query_template, *ages_and_scores.flatten)

3
投票

另一种方法,用于基于多列IN子句重构OP使用的确切查询。

一种更紧凑的SQL形式,具有可论证的更好的语义:

ages_and_scores = [ [5, 6], [9, 12], [22, 44] ]
query_string    = ages_and_scores.map { |pair| "(#{pair[0]},#{pair[1]})" }.join(",")
User.where("(age, score) IN (#{query_string})")

1
投票

嘿,你可以在mysql中尝试这种方式:

ages_and_scores = [ [5, 6], [9, 12], [22, 44] ]

User.where("CONCAT(age,',', score) in (?)",ages_and_scores.map{|b| "#{b[0]},#{b[1]}"})

在PG数据库中,您可以直接使用Concat:

   (age || ' ,' || score)

1
投票

另一个解决方案是使用Arel,这将有助于保持数据库不可知。

ages_and_scores = [ [5, 6], [9, 12], [22, 44] ]
first_age, first_score = ages_and_scores.shift
t = User.arel_table
users = ages_and_scores.inject(User.where(t[:age].eq(first_age)).where(t[:score].eq(first_score))) { |query, age_score| query.or(User.where(t[:age].eq(age_score[0])).where(t[:score].eq(age_score[1]))) }

基本上这是如何工作的:

  1. 将第一个值对与数组分开
  2. 获取arel表以便您可以构建查询
  3. 使用Ruby的inject模块中的Enumerable方法迭代遍历所有or条件的数组的其余部分...但是从第一个where查询中的初始两个值开始。

最后,您将获得一个Arel查询,该查询返回符合您条件的所有用户...并且它应该适用于Rails(Arel)支持的任何数据库。


0
投票

在rails 5中,您可以使用OR,这样您就可以:

ages_and_scores = [ [5, 6], [9, 12], [22, 44] ]
ages_and_scores.each do |age, score|
  User.where(age: age).where(score: score)
end.reduce(&:or)
# => should produce something like:
# SELECT * FROM users WHERE (`users`.`age` = 5 AND `users`.`score` = 6 OR `users`.`age` = 9 AND `users`.`score` = 12 OR `users`.`age` = 22 AND `users`.`score` = 44)

我相信这是sql-injection免费和纯ActiveRecord。


0
投票

基于Yuki Inoue的评论。我已经修改了一点,它工作顺利

ages_and_scores = [ [5, 6], [9, 12], [22, 44] ]
ages_and_scores.each do |age, score|
  User.where(age: age).map(score: score)
end.reduce(&:or)
# => should produce something like:
# SELECT * FROM users WHERE (`users`.`age` = 5 AND `users`.`score` = 6 OR `users`.`age` = 9 AND `users`.`score` = 12 OR `users`.`age` = 22 AND `users`.`score` = 44)
© www.soinside.com 2019 - 2024. All rights reserved.