GORM 关联关系之 Belongs To：多对一关联的实现与实践-CSDN博客

本文链接：https://blog.csdn.net/Landcc/article/details/149047244

在 GORM 中，Belongs To 是实现多对一关联关系的核心功能，它允许一个模型实例关联到另一个模型的单个实例。本文将详细介绍 Belongs To 关联的基本用法、外键自定义、预加载技巧及外键约束配置，帮助你轻松掌握这一重要关联关系。

一、Belongs To 关联基础：多对一关系的本质

1.1 什么是 Belongs To 关联

Belongs To 表示 "属于" 关系，是多对一关联的实现方式。例如：

一个用户（User）属于一个公司（Company）
一篇文章（Article）属于一个作者（Author）
一个订单（Order）属于一个用户（User）

这种关系的特点是：多端模型实例属于一端模型的一个实例，在数据库中通过外键关联实现。

1.2 基础模型定义与关联实现

// 公司模型（一端模型）
type Company struct {
  ID   int
  Name string
  City string
}

// 用户模型（多端模型，属于一个公司）
type User struct {
  gorm.Model
  Name      string
  Age       int
  CompanyID int        // 外键，关联Company的ID
  Company   Company    // 关联的Company结构体
}

关联本质解析：

User 表中包含 CompanyID 外键字段
GORM 自动通过 CompanyID 关联到 Company 表的 ID 字段
查询 User 时，可通过 Company 字段获取关联的公司信息

1.3 基础 CRUD 操作中的关联处理

// 创建带有关联的用户
func createUserWithCompany() {
    company := Company{Name: "GORM Inc", City: "Shanghai"}
    user := User{
        Name:      "Jinzhu",
        Age:       28,
        Company:   company,
    }
    
    // GORM 会自动保存Company，并设置User的CompanyID
    db.Create(&user)
    // SQL: INSERT INTO companies (name,city) VALUES ("GORM Inc","Shanghai");
    // SQL: INSERT INTO users (name,age,company_id) VALUES ("Jinzhu",28,1);
}

// 查询用户并获取关联公司
func getUserWithCompany() {
    var user User
    db.First(&user, 1)
    
    // 直接访问关联的Company
    fmt.Println("User Company:", user.Company.Name)
    // 若Company字段为零值，GORM会自动查询数据库
}

// 更新用户的关联公司
func updateUserCompany() {
    var user User
    db.First(&user, 1)
    
    var newCompany Company
    db.First(&newCompany, 2)
    
    user.Company = newCompany
    db.Save(&user)
    // SQL: UPDATE users SET company_id = 2 WHERE id = 1;
}

二、外键与引用的自定义：灵活配置关联关系

2.1 自定义外键字段名

默认情况下，GORM 使用 关联模型名+ID 作为外键字段名（如 CompanyID），但可以通过 foreignKey 标签自定义：

type User struct {
  gorm.Model
  Name         string
  CompanyRefer int         // 自定义外键字段名
  Company      Company     `gorm:"foreignKey:CompanyRefer"` // 指定外键字段
}

type Company struct {
  ID   int
  Name string
}

应用场景：

数据库表使用非标准外键命名（如 company_ref）
多个关联需要区分不同外键（如同时属于部门和公司）

2.2 自定义引用字段（非主键关联）

默认情况下，GORM 使用关联模型的主键作为引用，但可以通过 references 标签指定其他字段：

type User struct {
  gorm.Model
  Name      string
  CompanyID string       // 外键字段，存储Company的Code
  Company   Company      `gorm:"references:Code"` // 引用Company的Code字段
}

type Company struct {
  ID   int
  Code string  // 用于关联的唯一标识
  Name string
}

注意事项：

被引用字段（如 Code）需具有唯一性
外键字段类型需与被引用字段类型一致
此模式适用于关联非主键字段的场景（如第三方系统 ID）

2.3 避免关联歧义的配置

当关联模型的字段名与外键字段名冲突时，需显式配置 references：

type User struct {
  gorm.Model
  Name      string
  CompanyID int         // 外键字段
  Company   Company     `gorm:"references:CompanyID"` // 引用Company的CompanyID字段
}

type Company struct {
  CompanyID int    // 自定义主键字段
  Code      string
  Name      string
}

三、预加载关联数据：提升查询效率的关键

3.1 使用 Preload 预加载关联数据

默认情况下，GORM 不会自动加载关联数据，需要使用 Preload 显式预加载：

// 单条查询预加载关联公司
var user User
db.Preload("Company").First(&user, 1)
// SQL: SELECT * FROM users WHERE id = 1;
// SQL: SELECT * FROM companies WHERE id = 1;

// 批量查询预加载所有关联公司
var users []User
db.Preload("Company").Find(&users)
// SQL: SELECT * FROM users;
// SQL: SELECT * FROM companies WHERE id IN (1,2,3);

3.2 使用 Joins 预加载（单 SQL 查询）

Joins 方式通过 JOIN 语句在单个 SQL 中获取关联数据：

// 使用Joins预加载关联公司
var users []User
db.Joins("Company").Find(&users)
// SQL: SELECT users.*, companies.* FROM users LEFT JOIN companies ON users.company_id = companies.id;

// 带条件的Joins预加载
db.Joins("Company").Where("companies.city = ?", "Beijing").Find(&users)
// SQL: SELECT users.*, companies.* FROM users LEFT JOIN companies ON users.company_id = companies.id WHERE companies.city = "Beijing";

3.3 预加载性能对比与选择策略

预加载方式	优点	缺点	适用场景
Preload	自动处理 N+1 查询，代码简洁	多次数据库查询	大多数场景，尤其是复杂关联
Joins	单个 SQL 查询，性能更高	结果集可能重复，需处理去重	简单关联，追求极致性能

四、外键约束：数据一致性的保障

4.1 配置外键约束

通过 constraint 标签配置外键约束，GORM 在迁移时会创建对应的约束：

type User struct {
  gorm.Model
  Name      string
  CompanyID int
  Company   Company `gorm:"constraint:OnUpdate:CASCADE,OnDelete:SET NULL;"`
}

type Company struct {
  ID   int
  Name string
}

常见约束选项：

OnUpdate: CASCADE：当关联的 Company 更新主键时，User 的外键自动更新
OnDelete: SET NULL：当关联的 Company 被删除时，User 的外键设为 NULL
OnDelete: CASCADE：当关联的 Company 被删除时，User 也被级联删除
OnDelete: RESTRICT：禁止删除有相关 User 的 Company

4.2 外键约束的应用场景

// 场景1：公司删除时设置用户的CompanyID为NULL
// 配置：OnDelete: SET NULL
db.Delete(&company)
// SQL: UPDATE users SET company_id = NULL WHERE company_id = 1;
// 好处：保留用户数据，避免孤儿记录

// 场景2：公司删除时级联删除用户
// 配置：OnDelete: CASCADE
db.Delete(&company)
// SQL: DELETE FROM users WHERE company_id = 1;
// 注意：使用时需谨慎，避免意外删除大量数据

// 场景3：禁止删除有用户的公司
// 配置：OnDelete: RESTRICT
db.Delete(&company)
// 若存在关联用户，删除会失败并返回错误