如何用ActiveRecord内置功能实现任意N组(a,b)条件的单查询?
批量匹配多组(a,b)条件的单次查询实现
问题背景
假设有一个包含a、b列的Model,需要获取满足(a: 0 b: 3)、(a: 1 b: 4)、(a: 2 b: 5)的三组记录。原写法会执行三次独立查询:
as = [0,1,2] bs = [3,4,5] Model.where(a: as[0], b: bs[0]) Model.where(a: as[1], b: bs[1]) Model.where(a: as[2], b: bs[2])
使用.or可以合并为一次查询,但仅适用于固定数量的条件:
Model.where(a: as[0], b: bs[0]).or(Model.where(a: as[1], b: bs[1])).or(Model.where(a: as[2], b: bs[2]))
当需要匹配的记录数为任意N(as和bs数组长度为N)时,如何通过Rails内置功能实现单次查询,而非手动拼接SQL字符串?
解决方案
方法1:Rails 6+ 直接传入多组条件数组
Rails 6及以上版本支持直接向where传入包含多组键值对的数组,框架会自动生成OR连接的复合条件:
as = [0,1,2] bs = [3,4,5] # 把a和b数组配对,生成[{a:0, b:3}, {a:1, b:4}, {a:2, b:5}] conditions = as.zip(bs).map { |a_val, b_val| { a: a_val, b: b_val } } # 执行单次查询 Model.where(conditions)
生成的SQL示例:
SELECT "models".* FROM "models" WHERE (("models"."a" = 0 AND "models"."b" = 3) OR ("models"."a" = 1 AND "models"."b" = 4) OR ("models"."a" = 2 AND "models"."b" = 5))
方法2:用reduce链式拼接.or(兼容低版本Rails)
如果你的Rails版本低于6,可以用reduce方法自动把所有子查询用.or连接,无需手动编写N个链式调用:
as = [0,1,2] bs = [3,4,5] # 先生成每个条件对应的查询对象 queries = as.zip(bs).map { |a_val, b_val| Model.where(a: a_val, b: b_val) } # 用reduce把所有查询用.or拼接 queries.reduce(&:or)
注意:该方法要求所有子查询的SELECT字段保持一致,Rails会自动校验这一点。
内容的提问来源于stack exchange,提问作者gnk
相关产品推荐
相关产品推荐

