PostgreSQL能否仅向分区插入数据？如何修改分区触发器脚本

你的问题核心是触发器未正确阻止原INSERT操作写入父表，同时原脚本存在语法错误和性能隐患，以下是修改后的完整方案：

修改后的脚本 #

create or replace function create_partition_and_insert_to_partition_bundle() returns trigger as
$$
declare
    partition text;
    dt_constraint text;
    start_date date;
    end_date date;
begin
    -- 生成季度标识，修复原脚本的语法错误
    dt_constraint := format( 'y%sq%s', 
        date_part('year', new.created)::int, 
        ceil(date_part('month', new.created)::float / 3)::int
    );
    partition := format( 'bundle_%s', dt_constraint);
    
    -- 计算当前季度的起止日期，用于添加分区约束
    start_date := make_date(
        date_part('year', new.created)::int,
        ((ceil(date_part('month', new.created)::float / 3)::int - 1) * 3) + 1,
        1
    );
    end_date := (start_date + interval '3 months')::date;
    
    -- 检查分区是否存在，不存在则创建并添加CHECK约束
    if not exists(select relname from pg_class where relname = partition) then
        execute format('
            create table %I (like bundle including all) inherits (bundle);
            alter table %I add constraint %I_check_created 
            check (created >= %L and created < %L);
        ', partition, partition, partition, start_date, end_date);
    end if;
    
    -- 将数据插入对应分区
    execute format('insert into %I values ($1.*)', partition) using new;
    
    -- 返回NULL，终止原INSERT操作，避免写入父表
    return null;
end
$$ language plpgsql;

-- 重建触发器（先删除旧触发器，避免冲突）
drop trigger if exists create_insert_partition_bundle on bundle;
create trigger create_insert_partition_bundle 
before insert on bundle 
for each row execute procedure create_partition_and_insert_to_partition_bundle();

-- 开启约束排除，优化分区查询性能
set constraint_exclusion = partition;

-- 给父表添加不可插入约束，彻底防止数据写入父表
alter table bundle add constraint bundle_no_direct_insert check (false);

关键修改说明 #

• 修复语法错误：原脚本中format的第二个参数未加括号，导致函数无法正常运行，修改后直接使用表达式生成季度标识，同时转为整数让命名更规范。
• 添加分区CHECK约束：每个分区都会被限定只能存储对应季度的数据，既提升查询时的约束排除性能，又防止数据错插分区。
• 安全的SQL拼接：使用format的%I占位符处理表名和约束名，避免SQL注入风险。
• 阻止父表写入：通过返回NULL终止原INSERT操作，同时给父表添加check(false)约束，即使触发器失效，直接插入父表也会报错，从根源杜绝父表存数据。

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