Rails 7.2+pg_party创建分区表复合主键配置问题求助

如果你在Rails 7.2里用pg_party的create_range_partition创建范围分区表，想设置id + 分区字段的复合主键，但开了create_with_primary_key仍没生效，structured.sql里看不到主键声明，还导致没法用belongs_to关联，试试下面的解决方法：

正确配置步骤 #

1. 模型层明确复合主键 #

先在模型里用Rails 7.2原生的composite_primary_key指定复合主键：

class Order < ApplicationRecord
  composite_primary_key :id, :order_date
  # 现在belongs_to关联可以正常使用了
  belongs_to :customer
end

2. 迁移文件显式声明主键 + 开启pg_party主键配置 #

迁移里不能只靠pg_party自动识别，必须手动在分区表块里指定复合主键，同时确保create_with_primary_key为true：

class CreateOrdersRangePartition < ActiveRecord::Migration[7.2]
  def change
    create_range_partition :orders, partition_key: :order_date, type: :date do |t|
      t.bigint :id, null: false
      t.date :order_date, null: false
      t.string :customer_name
      # 其他业务字段...

      # 关键：显式指定复合主键
      t.primary_key [:id, :order_date]
    end

    # 按需创建初始分区
    add_range_partition :orders, start: '2024-01-01', end: '2024-02-01', name: 'orders_2024_01'
  end

  # 全局开启pg_party创建主键的配置（也可以放到initializer里）
  def self.config
    pg_party_config = super
    pg_party_config[:create_with_primary_key] = true
    pg_party_config
  end
end

3. 验证结果 #

执行rails db:migrate后，再跑rails db:structure:dump，检查db/structure.sql，应该能看到PostgreSQL的主键约束声明：

ALTER TABLE ONLY orders ADD CONSTRAINT orders_pkey PRIMARY KEY (id, order_date);

常见踩坑点 #

• 漏写t.primary_key：pg_party不会自动读取模型的复合主键定义，必须在迁移里手动声明
• 配置位置错误：create_with_primary_key要在迁移类的self.config里设置，或者全局初始化文件config/initializers/pg_party.rb里配置：

  PgParty.configure do |config|
    config.create_with_primary_key = true
  end
  

• 字段未加null约束：复合主键的所有字段必须设null: false，PostgreSQL不允许主键字段为null

兜底方案 #

如果上述方法还是不生效，直接在迁移后手动执行SQL添加主键约束：

class AddCompositePkToOrders < ActiveRecord::Migration[7.2]
  def change
    execute <<~SQL
      ALTER TABLE orders ADD PRIMARY KEY (id, order_date);
    SQL
  end
end

这种方式直接绕开pg_party的逻辑，确保复合主键生效，同时belongs_to关联也能正常工作。

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