在 Postgresql 中模拟 MySQL 的 ORDER BY FIELD()

Question

第一次尝试 PostgreSQL，来自 MySQL。在我们的 Rails 应用程序中，我们有几个使用 SQL 的位置，如下所示：

SELECT * FROM `currency_codes` ORDER BY FIELD(code, 'GBP', 'EUR', 'BBD', 'AUD', 'CAD', 'USD') DESC, name ASC

很快就发现 PostgreSQL 不支持/不允许这样做。

有谁知道如何在 PostgreSQL 中模拟这种行为，还是我们必须在代码中进行排序？

Answer 1

ilgam's answer 自 Rails 6.1 起将无法使用，将引发 ActiveRecord::UnknownAttributeReference 错误：https://api.rubyonrails.org/classes/ActiveRecord/UnknownAttributeReference.html

推荐的方法是使用 Arel 而不是原始 SQL。

除了ilgam's answer，这里是Rails 6.1的解决方案：

def self.order_by_ids(ids)
  t = User.arel_table
  condition = Arel::Nodes::Case.new(t[:id])
  ids.each_with_index do |id, index|
    condition.when(id).then(index)
  end
  order(condition)
end

Answer 2

也可以使用数组unnest 和WITH ORDINALITY 功能进行此排序：

--- Table and data setup ...
CREATE TABLE currency_codes (
     code text null,
     name text
);
INSERT INTO currency_codes
  (code)
VALUES
  ('USD'), ('BBD'), ('GBP'), ('EUR'), ('AUD'), ('CAD'), ('AUD'), ('AUD');

-- ...and the Query
SELECT
  c.*
FROM
  currency_codes c
JOIN
  unnest('{"GBP", "EUR", "BBD", "AUD", "CAD", "USD"}'::text[])
  WITH ORDINALITY t(code, ord)
  USING (code)
ORDER BY t.ord DESC, c.name ASC;

Answer 3

SELECT * FROM (VALUES ('foo'), ('bar'), ('baz'), ('egg'), ('lol')) t1(name)
ORDER BY ARRAY_POSITION(ARRAY['foo', 'baz', 'egg', 'bar'], name)

这个怎么样？以上一次获取如下：

foo
baz
egg
bar
lol

正如你已经知道的那样，如果一个元素不在数组中，那么它会回到后面。

Answer 4

在mysql中排序：

> ids = [11,31,29]
=> [11, 31, 29]
> User.where(id: ids).order("field(id, #{ids.join(',')})")

在 postgres 中：

def self.order_by_ids(ids)
  order_by = ["CASE"]
  ids.each_with_index do |id, index|
    order_by << "WHEN id='#{id}' THEN #{index}"
  end
  order_by << "END"
  order(order_by.join(" "))
end

User.where(id: [3,2,1]).order_by_ids([3,2,1]).map(&:id) 
#=> [3,2,1]

Answer 5

使用此函数创建迁移

CREATE OR REPLACE FUNCTION field(anyelement, VARIADIC anyarray) RETURNS bigint AS $$
  SELECT n FROM (
    SELECT row_number() OVER () AS n, x FROM unnest($2) x)
      numbered WHERE numbered.x = $1;
$$ LANGUAGE SQL IMMUTABLE STRICT;

那就这样做吧

sequence = [2,4,1,5]
Model.order("field(id,#{sequence.join(',')})")

瞧！

Answer 6

当我回答 here 时，我刚刚发布了一个 gem (order_as_specified)，它允许您像这样进行本机 SQL 排序：

CurrencyCode.order_as_specified(code: ['GBP', 'EUR', 'BBD', 'AUD', 'CAD', 'USD'])

它返回一个 ActiveRecord 关系，因此可以与其他方法链接，它适用于我测试过的每个 RDBMS。

Answer 7

如果您经常运行此操作，请添加一个新列和一个预插入/更新触发器。然后根据此触发器设置新列中的值并按此字段排序。您甚至可以在此字段上添加索引。

Answer 8

只需定义FIELD 函数并使用它。这很容易实现。以下应该在 8.4 中工作，因为它具有 unnest 和像 row_number 这样的窗口函数：

CREATE OR REPLACE FUNCTION field(text, VARIADIC text[]) RETURNS bigint AS $$
SELECT n FROM (
    SELECT row_number() OVER () AS n, x FROM unnest($2) x
) numbered WHERE numbered.x = $1;
$$ LANGUAGE 'SQL' IMMUTABLE STRICT;

您还可以定义另一个带有签名的副本：

CREATE OR REPLACE FUNCTION field(anyelement, VARIADIC anyarray) RETURNS bigint AS $$

如果您想为任何数据类型支持field()，则使用相同的正文。

Answer 9

其实 postgres 8.1 的版本是另一个优势。

调用 postgres 函数时，您不能向其传递超过 100 个参数，因此您最多可以对 99 个元素进行排序。

使用使用数组作为第二个参数的函数，而不是使用可变参数，只需删除此限制。

Answer 10

你可以这样做……

SELECT 
   ..., code
FROM 
   tablename
ORDER BY 
   CASE 
      WHEN code='GBP' THEN 1
      WHEN code='EUR' THEN 2
      WHEN code='BBD' THEN 3
      ELSE 4
   END

但是为什么要将这些硬编码到查询中——支持表不是更合适吗？

--

编辑：按照 cmets 翻转它

Answer 11

更新，充实了@Tometzky 的绝妙建议。

这应该在 pg 8.4 下给你一个 MySQL FIELD()-alike 函数：

-- SELECT FIELD(varnames, 'foo', 'bar', 'baz')
CREATE FUNCTION field(anyelement, VARIADIC anyarray) RETURNS numeric AS $$
  SELECT
    COALESCE(
     ( SELECT i FROM generate_subscripts($2, 1) gs(i)
       WHERE $2[i] = $1 ),
     0);
$$ LANGUAGE SQL STABLE

Mea culpa，但我现在无法在 8.4 上验证上述内容；但是，我可以向后工作到适用于我面前的 8.1 实例的“道德”等效版本：

-- SELECT FIELD(varname, ARRAY['foo', 'bar', 'baz'])
CREATE OR REPLACE FUNCTION field(anyelement, anyarray) RETURNS numeric AS $$
  SELECT
    COALESCE((SELECT i
              FROM generate_series(1, array_upper($2, 1)) gs(i)
              WHERE $2[i] = $1),
             0);
$$ LANGUAGE SQL STABLE

更尴尬的是，您仍然可以便携地使用（可能派生的）货币代码排名表，如下所示：

pg=> select cc.* from currency_codes cc
     left join
       (select 'GBP' as code, 0 as rank union all
        select 'EUR', 1 union all
        select 'BBD', 2 union all
        select 'AUD', 3 union all
        select 'CAD', 4 union all
        select 'USD', 5) cc_weights
     on cc.code = cc_weights.code
     order by rank desc, name asc;
 code |           name
------+---------------------------
 USD  | USA bits
 CAD  | Canadian maple tokens
 AUD  | Australian diwallarangoos
 BBD  | Barbadian tridents
 EUR  | Euro chits
 GBP  | British haypennies
(6 rows)

Answer 12

这是我认为最简单的方法：

create temporary table test (id serial, field text);
insert into test(field) values
  ('GBP'), ('EUR'), ('BBD'), ('AUD'), ('CAD'), ('USD'),
  ('GBP'), ('EUR'), ('BBD'), ('AUD'), ('CAD'), ('USD');
select * from test
order by field!='GBP', field!='EUR', field!='BBD',
  field!='AUD', field!='CAD', field!='USD';
 id | field 
----+-------
  1 | GBP
  7 | GBP
  2 | EUR
  8 | EUR
  3 | BBD
  9 | BBD
  4 | AUD
 10 | AUD
  5 | CAD
 11 | CAD
  6 | USD
 12 | USD
(12 rows)

在 PostgreSQL 8.4 中，您还可以使用 function with variable number of arguments（可变函数）来移植 field 函数。

Answer 13

啊，gahooa 这么近：

SELECT * FROM currency_codes
  ORDER BY
  CASE
    WHEN code='USD' THEN 1
    WHEN code='CAD' THEN 2
    WHEN code='AUD' THEN 3
    WHEN code='BBD' THEN 4
    WHEN code='EUR' THEN 5
    WHEN code='GBP' THEN 6
    ELSE 7
  END,name;