Postgresql:为SELECT中的值划分表

Question

我想在具有UUID值的列上使用PostgreSQL的声明式表分区。我希望它通过SELECT返回的值进行分区。

CREATE TABLE foo (
id uuid NOT NULL,
type_id uuid NOT NULL,
-- other columns
PRIMARY KEY (id, type_id)
) PARTITION BY LIST (type_id);
CREATE TABLE foo_1 PARTITION OF foo
FOR VALUES IN (SELECT id FROM type_ids WHERE type_name = 'type1');
CREATE TABLE foo_2 PARTITION OF foo
FOR VALUES IN (SELECT id FROM type_ids WHERE type_name = 'type2');

我不想在FOR VALUES IN ('uuid')中使用特定的uuid，因为uuid可能因环境(dev, qa, prod)而异。然而，SELECT语法似乎不被Postgres接受。有什么建议吗?

Answer 1

我只想在表创建时对SELECT进行评估

你应该在问题里说清楚，而不是在评论里。

在这种情况下-如果这是一次性的事情，你可以使用动态SQL，例如包装成一个过程

create procedure create_partition(p_part_name text, p_type_name text)
as
$$
declare
l_sql text;
l_id uuid;
begin
select id
into l_id
from type_ids
where type_name = p_type_name;
l_sql := format('CREATE TABLE %I PARTITION OF foo FOR VALUES IN (%L)', p_part_name, l_id);
execute l_sql;
end;
$$
language plpgsql;

那么你可以这样做:

call create_partition('foo_1', 'type1');
call create_partition('foo_2', 'type2');