是否可以将枚举转换为整数?从1开始第一个元素
3 回答
虽然您不能像 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 相信您时函数是不可变的。不要那样做。
如果您乐于使用函数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 指向其他值,因此不要在您可能希望在某些时候更改的枚举上使用它。