如何在PostgreSQL中对非主键列哈希分区并适配Django迁移？

问题背景 #

要将现有非分区表替换为基于vendor_id的哈希分区表，要求新表schema与原表完全一致，且能通过Django迁移正常管理。首次创建分区表时触发PostgreSQL约束报错，尝试复合主键或无主键的方案又无法满足Django对单主键的强制要求。

报错原因 #

PostgreSQL硬性规定：分区表的主键/唯一约束必须包含所有分区列。你首次创建的表主键仅为id（或想要保留的row_index），未包含分区键vendor_id，因此触发错误。而Django不支持复合主键，直接使用(row_index, vendor_id)作为主键会导致Django认知的schema与实际数据库schema不一致。

可行解决方案 #

通过数据库层复合主键+全局唯一约束配合Django模型配置，同时满足PostgreSQL分区要求和Django单主键规则：

1. 创建符合要求的哈希分区表 #

先创建分区表结构，将复合键(row_index, vendor_id)设为数据库主键，同时给row_index添加全局唯一约束，保证其与原表主键逻辑一致：

CREATE TABLE partition_test (
    row_index bigint NOT NULL,
    id character varying(128),
    vendor_id character varying(128) NOT NULL,
    -- 满足PostgreSQL分区要求的复合主键
    PRIMARY KEY (row_index, vendor_id),
    -- 保证row_index全局唯一，对齐原表主键逻辑
    UNIQUE (row_index)
) PARTITION BY HASH (vendor_id);

再创建具体的哈希分区（示例为4个分区，可根据业务需求调整数量）：

-- 创建4个哈希分区
CREATE TABLE partition_test_p1 PARTITION OF partition_test FOR VALUES WITH (MODULUS 4, REMAINDER 0);
CREATE TABLE partition_test_p2 PARTITION OF partition_test FOR VALUES WITH (MODULUS 4, REMAINDER 1);
CREATE TABLE partition_test_p3 PARTITION OF partition_test FOR VALUES WITH (MODULUS 4, REMAINDER 2);
CREATE TABLE partition_test_p4 PARTITION OF partition_test FOR VALUES WITH (MODULUS 4, REMAINDER 3);

2. 配置Django模型 #

在Django模型中，将row_index设为单主键，同时添加唯一约束与数据库层对应，确保schema完全一致：

from django.db import models

class PartitionTest(models.Model):
    row_index = models.BigIntegerField(primary_key=True)
    id = models.CharField(max_length=128, blank=True, null=True)
    vendor_id = models.CharField(max_length=128)

    class Meta:
        db_table = 'partition_test'
        constraints = [
            models.UniqueConstraint(fields=['row_index'], name='unique_row_index'),
        ]

3. 迁移适配 #

如果是从原有非分区表迁移数据，可先导出原表数据，创建分区表后导入，再通过Django的makemigrations和migrate同步模型。注意：迁移时需确保Django不会尝试修改主键结构，可通过自定义迁移文件或先禁用自动迁移再手动调整。

方案合理性说明 #

• 数据库层：复合主键(row_index, vendor_id)满足PostgreSQL分区表的约束要求，UNIQUE(row_index)保证了原表主键的唯一性逻辑不受影响。
• Django层：模型将row_index设为单主键，与原表schema完全对齐，唯一约束的配置也和数据库层匹配，不会出现迁移冲突。

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