Rails中使用.where查询时如何保留传入ID数组的顺序
PostgreSQL环境下Rails按传入ID数组顺序查询的解决办法
我之前在项目里也踩过这个坑!MySQL的field()函数在PostgreSQL里根本不支持,给你两个高效的解决方案,比你循环查的方法靠谱多了:
方法1:用array_position函数(PostgreSQL 9.5及以上可用)
PostgreSQL从9.5版本开始支持array_position,可以直接根据数组元素的位置来排序,代码这么写:
Page.where(company_id: company_ids).order("array_position(ARRAY[#{company_ids.join(',')}]::bigint[], company_id)")
注意要把生成的数组转成和company_id一致的数据类型(比如你的company_id是bigint,就加::bigint[]),这样就能严格按照你传入的company_ids顺序返回结果了。
方法2:用CASE WHEN语句(兼容低版本PostgreSQL)
如果你的PostgreSQL版本比较老,低于9.5,那用CASE WHEN手动指定排序优先级也很稳:
order_case = company_ids.each_with_index.map { |id, index| "WHEN #{id} THEN #{index}" }.join(' ') Page.where(company_id: company_ids).order("CASE company_id #{order_case} END")
这个逻辑是给每个传入的ID分配一个对应的排序序号,数据库会按照这个序号排序,完全匹配原数组的顺序,而且也是单次查询,效率拉满。
为啥你的临时方案效率低?
你循环调用find_by的话,每一次循环都会发一条SQL查询,数据量一大就变成N+1查询了,数据库压力会很大,上面两个方法都是一次查询搞定,性能差很多的。
内容的提问来源于stack exchange,提问作者LewlSauce
相关产品推荐
相关产品推荐

