Rails自定义排序：如何让ActiveRecord关联按指定国家优先且不转数组

原实现的核心问题 #

• 关于sort_by会将ActiveRecord关联转为数组的判断是完全准确的：当前实现调用sort_by时会立即触发全量数据加载，把关联转为Ruby数组做内存排序，最终返回值是Array类型，丢失了关联的链式调用能力，无法在后续流程中继续拼接筛选、分页等查询逻辑
• 内存排序的性能局限明显：当匹配的记录量较大时，全量加载到内存做排序会造成不必要的资源消耗，效率远低于数据库原生排序

调整思路 #

不需要依赖Ruby层面的内存排序，把优先国家的排序规则直接下推到数据库的ORDER BY子句中，全程保持查询的惰性加载特性，返回值始终为ActiveRecord::Relation类型，不提前加载数据。

修正后的方法实现 #

class Profile
  def self.order_user_preferences
    # 定义需要排在最前面的优先国家代码
    priority_countries = ['US']
    # 对国家代码做数据库安全转义，避免SQL注入风险
    escaped_codes = priority_countries.map { |code| connection.quote(code) }.join(',')
    # 构造数据库可识别的排序规则：命中优先列表的记录排最前
    sort_rule = Arel.sql(
      "CASE WHEN country_code IN (#{escaped_codes}) THEN 0 ELSE 1 END"
    )
    # 将排序规则追加到当前关联实例，返回新的关联对象
    order(sort_rule)
  end
end

注：如果需要扩展多国家的优先顺序，只需要调整priority_countries数组即可，比如要把US、CA、UK按顺序排在最前，直接把数组改为['US', 'CA', 'UK']，排序规则会自动适配。

实际调用效果 #

调用Profile.where(id: [1,2,3]).order_user_preferences时，生成的SQL会自动携带对应的排序规则，最终返回顺序完全匹配预期：

整个流程不会提前触发数据加载，返回值依然是标准的ActiveRecord关联对象，可以无缝继续链式调用其他查询方法，比如后续拼接字段筛选、分页等逻辑都不会中断查询链路。

内容的提问来源于stack exchange，提问作者sparkle