活动记录 - 具有多列的IN

时间:2016-05-12 09:33:06

标签: ruby-on-rails postgresql activerecord

我有一个查询需要从满足两列要求的表中获取。因此,如果我的users表包含列agescore

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

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

我正在研究postgres数据库

6 个答案:

答案 0 :(得分: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)

答案 1 :(得分:4)

另一种方法,用于根据多列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})")

答案 2 :(得分: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)

答案 3 :(得分: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查询。
  4. 最后,您将有一个Arel查询,它返回符合您条件的所有用户......并且它应该适用于Rails(Arel)支持的任何数据库。

答案 4 :(得分: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注入的纯ActiveRecord。

答案 5 :(得分: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)