将SQL触发器转换为Gorm BeforeSave触发器：不存在时插入字段至另一表

直接执行SQL触发器语句容易因数据库语法限制（比如MySQL需要修改语句分隔符）导致预编译失败，更推荐用Gorm的钩子方法替代数据库触发器，实现业务逻辑：

1. 定义模型结构 #

先声明User和Chat的Gorm模型，确保users表的phone字段唯一：

type User struct {
    ID    uint   `gorm:"primaryKey"`
    Phone string `gorm:"unique;not null"`
}

type Chat struct {
    ID     uint   `gorm:"primaryKey"`
    Phone  string `gorm:"not null"`
    // 其他业务字段...
}

2. 实现Chat的BeforeSave钩子 #

给Chat结构体绑定BeforeSave方法，在插入/更新Chat数据前自动检查并创建用户：

func (c *Chat) BeforeSave(tx *gorm.DB) error {
    var user User
    // 查询当前phone是否已存在于users表
    result := tx.Where("phone = ?", c.Phone).First(&user)
    
    switch {
    case result.Error == gorm.ErrRecordNotFound:
        // 不存在则插入新用户
        return tx.Create(&User{Phone: c.Phone}).Error
    case result.Error != nil:
        // 处理查询异常
        return result.Error
    default:
        // 用户已存在，无需操作
        return nil
    }
}

3. 批量插入优化（可选） #

如果是批量插入Chat数据，逐个触发钩子会影响性能，可改用批量检查逻辑：

func BatchCreateChats(tx *gorm.DB, chats []Chat) error {
    // 提取所有待插入的phone
    phones := make([]string, len(chats))
    for i, chat := range chats {
        phones[i] = chat.Phone
    }

    // 查询已存在的phone
    var existingPhones []string
    tx.Model(&User{}).Where("phone IN ?", phones).Pluck("phone", &existingPhones)
    existingMap := make(map[string]bool)
    for _, p := range existingPhones {
        existingMap[p] = true
    }

    // 收集需要新建的用户
    var newUsers []User
    for _, p := range phones {
        if !existingMap[p] {
            newUsers = append(newUsers, User{Phone: p})
            existingMap[p] = true // 避免重复添加同一phone
        }
    }

    // 批量插入新用户
    if len(newUsers) > 0 {
        if err := tx.Create(&newUsers).Error; err != nil {
            return err
        }
    }

    // 批量插入Chat数据
    return tx.Create(&chats).Error
}

原SQL触发器失败原因 #

不同数据库对触发器语法要求不同，比如MySQL需要先执行DELIMITER //修改语句分隔符，但Gorm的Exec方法默认以分号拆分语句，会把触发器代码拆成多个无效片段，导致预编译报错。用Gorm钩子可以避开数据库层面的语法差异，逻辑更易维护。

内容的提问来源于stack exchange，提问作者O'Brien