使用GORM数据模型设置外键失败的问题求助

问题背景 #

使用GORM定义数据模型并设置外键，执行AutoMigrate时部分外键未生效，偶尔报错，需要生成符合以下SQL结构的Pass_Fail_Criteria_Feilds表：

CREATE TABLE Pass_Fail_Criteria_Feilds
(
  Criteria_Id SERIAL PRIMARY KEY NOT NULL,
  Category_Id INTEGER REFERENCES Pass_Fail_Categories(Category_Id) NOT NULL,
  Category_Message VARCHAR(50) REFERENCES Pass_Fail_Category_Messages(Messages) NOT NULL,
  Numeric_Operator VARCHAR(2) NOT NULL,
  Criteria_Value INTEGER NOT NULL
)

原代码问题分析 #

1. 关联模型缺少主键约束：Pass_Fail_Category_Messages中Messages字段仅设为unique，未被标记为主键或可被GORM识别的外键引用目标，导致关联失败。
2. 主键类型不符合期望：Pass_Fail_Criteria_Feilds的Criteria_Id仅标记为primary_key，未启用自增，无法生成PostgreSQL的SERIAL类型。
3. 部分模型主键缺失：User_Entity的Id字段未标记为主键，会影响关联它的Log_Settings模型外键生效。

修正后的代码 #

package main

import (
    "log"

    "database/sql/driver"
    "encoding/json"
    "errors"

    "gorm.io/driver/postgres"
    "gorm.io/gorm"
)

type JSONB map[string]interface{}

func (j JSONB) Value() (driver.Value, error) {
    if j == nil {
        return nil, nil
    }
    return json.Marshal(j)
}

func (j *JSONB) Scan(value interface{}) error {
    b, ok := value.([]byte)
    if !ok {
        return errors.New("failed to unmarshal JSONB value")
    }

    var v interface{}
    if err := json.Unmarshal(b, &v); err != nil {
        return err
    }

    *j, ok = v.(map[string]interface{})
    if !ok {
        return errors.New("failed to convert JSONB value")
    }

    return nil
}

type Tag struct {
    Tag_id   int    `gorm:"primaryKey;not null"`
    Tag_name string `gorm:"type:varchar(255);not null"`
}

type User_Entity struct {
    Id    string `gorm:"type:varchar(36);primaryKey;not null;unique"`
    Email string `gorm:"type:varchar(255);not null"`
}

type Log_Settings struct {
    Log_Setting_Id   int    `gorm:"primaryKey;not null"`
    Log_Setting_Name string `gorm:"type:varchar(20);not null"`
    Creator_Id       string `gorm:"type:varchar(36);not null"`

    User_Entity User_Entity `gorm:"foreignKey:Creator_Id"`
}

type Pass_Fail_Categories struct {
    Category_Id   int    `gorm:"primaryKey;not null"`
    Category_Name string `gorm:"type:varchar(15);not null"`
}

type Pass_Fail_Category_Messages struct {
    Category_Id      int    `gorm:"not null"`
    Messages         string `gorm:"type:varchar(50);primaryKey;not null"` // 设为主键，支持外键引用

    Pass_Fail_Categories Pass_Fail_Categories `gorm:"foreignKey:Category_Id"`
}

type Pass_Fail_Criteria_Feilds struct {
    Criteria_Id      int    `gorm:"primaryKey;autoIncrement;not null"` // 启用自增生成SERIAL类型
    Category_Id      int    `gorm:"not null"`
    Category_Message string `gorm:"type:varchar(50);not null"`
    Numeric_Operator string `gorm:"type:varchar(2);not null"`
    Criteria_Value   int    `gorm:"not null"`

    Pass_Fail_Categories        *Pass_Fail_Categories        `gorm:"foreignKey:Category_Id"`
    Pass_Fail_Category_Messages Pass_Fail_Category_Messages `gorm:"foreignKey:Category_Message;references:Messages"`
}

func main() {
    db, err := gorm.Open(postgres.Open("host=localhost user=postgres password=1234567 dbname=tester port=5432 sslmode=disable"), &gorm.Config{})
    if err != nil {
        log.Fatal("数据库连接失败:", err)
    }

    // 一次性迁移所有模型，自动处理依赖顺序
    err = db.AutoMigrate(
        &Tag{},
        &User_Entity{},
        &Pass_Fail_Categories{},
        &Pass_Fail_Category_Messages{},
        &Pass_Fail_Criteria_Feilds{},
        &Log_Settings{},
    )
    if err != nil {
        log.Fatal("自动迁移失败:", err)
    }
    log.Println("自动迁移完成")
}

关键修改说明 #

• Pass_Fail_Category_Messages模型：将Messages字段标记为primaryKey，确保可以被Pass_Fail_Criteria_Feilds的外键引用。
• Pass_Fail_Criteria_Feilds模型：给Criteria_Id添加autoIncrement标签，让PostgreSQL自动生成SERIAL类型字段。
• User_Entity模型：将Id标记为primaryKey，修复关联模型的外键生效问题。
• 迁移调用优化：一次性传入所有模型，GORM会自动处理表的创建顺序，避免因依赖关系导致的外键约束失败。

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