PostgreSQL列表分区支持自动创建吗？如何实现插入时自动分区？

PostgreSQL 原生不支持列表分区（list partitioned table）的自动创建——当插入数据的分区键不属于任何现有分区时，默认会直接抛出错误，不会自动生成新分区。目前仅范围分区（range partitioned table）在部分场景下可结合工具实现半自动/自动分区，但列表分区需要通过自定义逻辑或第三方扩展来实现需求。

实现方案 #

1. 自定义触发器函数 #

通过编写BEFORE INSERT触发器，在数据插入前检查目标分区是否存在，若不存在则动态创建分区，再执行插入操作。

示例代码

假设我们有一个按country字段分区的主表：

-- 创建主列表分区表
CREATE TABLE users (
    id INT,
    name TEXT,
    country TEXT
) PARTITION BY LIST (country);

编写触发器函数：

CREATE OR REPLACE FUNCTION create_list_partition_if_not_exists()
RETURNS TRIGGER AS $$
DECLARE
    partition_name TEXT := 'users_' || REPLACE(NEW.country, ' ', '_'); -- 生成分区表名
BEGIN
    -- 检查分区是否存在
    IF NOT EXISTS (
        SELECT 1 FROM pg_tables 
        WHERE schemaname = 'public' AND tablename = partition_name
    ) THEN
        -- 动态创建分区
        EXECUTE format(
            'CREATE TABLE %I PARTITION OF users FOR VALUES IN (%L)',
            partition_name, NEW.country
        );
    END IF;
    RETURN NEW;
END;
$$ LANGUAGE plpgsql;

绑定触发器到主表：

CREATE TRIGGER trigger_create_users_partition
BEFORE INSERT ON users
FOR EACH ROW EXECUTE FUNCTION create_list_partition_if_not_exists();

注意事项

• 触发器会带来一定性能损耗，高并发插入场景下需谨慎评估；
• 执行触发器的数据库用户必须拥有创建表的权限；
• 需处理并发插入相同新分区键的情况（可在函数中添加异常捕获，避免重复创建报错）；
• 分区命名规则需提前规划，避免特殊字符导致的语法错误。

2. 使用pg_partman扩展 #

pg_partman是PostgreSQL生态中成熟的分区管理扩展，支持列表分区的自动创建与维护，适合生产环境使用。

实现步骤

1. 安装pg_partman（需对应PostgreSQL版本）：

   CREATE EXTENSION pg_partman;
   

2. 配置主分区表的自动分区规则：

   -- 创建主表（同前）
   CREATE TABLE users (
       id INT,
       name TEXT,
       country TEXT
   ) PARTITION BY LIST (country);
   
   -- 配置自动分区，设置分区类型为list，开启自动维护
   SELECT partman.create_parent(
       p_parent_table := 'public.users',
       p_control := 'country',
       p_type := 'list',
       p_automatic_maintenance := 'on'
   );
   

说明

• pg_partman会通过后台任务或触发器自动检测新的分区键，创建对应分区；
• 可通过partman.part_config表调整自动维护的参数（如分区命名规则、保留策略等）；
• 相比自定义触发器，pg_partman提供了更完善的分区生命周期管理（如分区删除、合并等）。

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