21

是否可以将枚举转换为整数?从1开始第一个元素

4

3 回答 3

34

虽然您不能像 Catcall 解释的那样将枚举转换为整数,但您可以使用 PostgreSQL 特定且可能与版本不兼容的pg_enum系统目录表来获得序数表示。

regress=# CREATE TYPE happiness AS ENUM ('happy', 'very happy', 'ecstatic');

regress=# select enumsortorder, enumlabel from pg_catalog.pg_enum 
regress-# WHERE enumtypid = 'happiness'::regtype ORDER BY enumsortorder;
 enumsortorder | enumlabel  
---------------+------------
             1 | happy
             2 | very happy
             3 | ecstatic
(3 rows)

这看起来很容易,但事实并非如此。观察:

regress=# ALTER TYPE happiness ADD VALUE 'sad' BEFORE 'happy';
regress=# ALTER TYPE happiness ADD VALUE 'miserable' BEFORE 'very happy';
regress=# SELECT * FROM pg_enum ;
 enumtypid | enumsortorder | enumlabel  
-----------+---------------+------------
    185300 |             1 | happy
    185300 |             2 | very happy
    185300 |             3 | ecstatic
    185300 |             0 | sad
    185300 |           1.5 | miserable
(5 rows)

从这里你可以看到它enumsortorder提供了排序,但没有固定的“距离”。如果添加了对从枚举中删除值的支持,它也可能会在序列中创建“漏洞”。

要获取枚举位置,您需要使用row_number()窗口函数来获取排序,并pg_typeof获取枚举类型的oid( regtype)。当有多个具有相同标签的枚举时,您需要它来确保返回正确的序数。

这个函数完成了这项工作:

CREATE OR REPLACE FUNCTION enum_to_position(anyenum) RETURNS integer AS $$
SELECT enumpos::integer FROM (
        SELECT row_number() OVER (order by enumsortorder) AS enumpos,
               enumsortorder,
               enumlabel
        FROM pg_catalog.pg_enum
        WHERE enumtypid = pg_typeof($1)
    ) enum_ordering
    WHERE enumlabel = ($1::text);
$$ LANGUAGE 'SQL' STABLE STRICT;

笔记:

  • STABLE不是IMMUTABLE,因为从枚举中添加(或者如果稍后添加 Pg 中的支持,则删除)值会改变排序并破坏依赖于排序的索引;所以
  • 您不能在索引表达式中使用它;和
  • 这是STRICT因为它应该为 null 输入返回 null

您现在可以使用此函数将CREATE CAST特定枚举用于integer. 您不能为所有枚举创建通用转换integer,因为anyenum伪类型不能用于转换。例如,如果我想让演示happiness转换为整数,我会写:

CREATE CAST (happiness AS integer) WITH FUNCTION enum_to_position(anyenum);

之后我可以成功执行:

regress=# SELECT ('happy'::happiness)::integer;
 int4 
------
    2
(1 row)

请注意,这可能是一件疯狂的事情,不受支持,而且很可能是一个糟糕的主意。您的代码必须知道,当您添加或(如果以后支持)从枚举中删除一个值时,序数值会发生变化。

如果您更改枚举的定义(除非在其末尾附加新值),那么基于此强制转换创建的索引(仅当函数定义为不可变时才可能)将开始产生疯狂和错误的结果,因为当您说 PostgreSQL 相信您时函数是不可变的。不要那样做。

于 2012-09-10T08:17:13.857 回答
22

如果您乐于使用函数enum_range()并计算其结果,则可以这样做。

如果您将枚举值作为 enum_range() 函数的第二个参数传递,并且 NULL 是第一个参数,您将获得一个数组,其中包含枚举可以占用的所有值。然后你只需要用 array_length 对它们进行计数,你就会得到一个表示枚举的整数。

这是一个例子。这是我的枚举:

content=# select enum_range(null::content_state);
                        enum_range                          
--------------------------------------------------------------
 {created,deleted,preview,draft,submitted,approved,published}

这是我找出“草稿”值的 int :

content=# select array_length(enum_range(NULL, 'draft'::content_state), 1);
 array_length 
--------------
            4

警告:从枚举中删除值将使您的 int 指向其他值,因此不要在您可能希望在某些时候更改的枚举上使用它。

于 2013-04-16T18:53:16.777 回答
4

您不能将枚举转换为整数。

也许可以编写一个自定义运算符来提取与值关联的数字,但我发现很难相信这是值得的。

如果我需要这种信息,我会建立一个表并为其设置外键引用,而不是使用枚举。

于 2012-09-10T03:36:56.517 回答