如何在GORM中优雅处理多数据库下的唯一约束错误？

1. 封装通用的唯一约束错误判断函数 #

不同数据库的唯一约束冲突会返回各自特定的错误码，我们可以封装一个函数，把这些数据库相关的错误转换成通用判断逻辑，不用上层代码关心具体数据库类型：

import (
  "errors"
  "gorm.io/gorm"
  sqliteErr "gorm.io/driver/sqlite"
  mysqlErr "gorm.io/driver/mysql"
  pgErr "gorm.io/driver/postgres"
)

// 定义通用的唯一约束错误
var ErrUniqueConstraintViolation = errors.New("unique constraint violation")

// 判断是否为唯一约束冲突错误
func IsUniqueConstraintError(err error) bool {
  if err == nil {
    return false
  }

  var gormErr *gorm.DBError
  if errors.As(err, &gormErr) {
    switch driverErr := gormErr.Unwrap().(type) {
    // SQLite：唯一约束冲突对应错误码19，扩展码2067
    case *sqliteErr.Error:
      return driverErr.Code == 19 && driverErr.ExtendedCode == 2067
    // MySQL：唯一约束冲突错误码1062
    case *mysqlErr.MySQLError:
      return driverErr.Number == 1062
    // PostgreSQL：唯一约束冲突错误码23505
    case *pgErr.PgError:
      return driverErr.Code == "23505"
    // 新增其他数据库时，在这里补充判断逻辑即可
    default:
      return false
    }
  }

  return false
}

使用时直接调用这个函数判断：

if err := db.Create(&user).Error; err != nil {
  if IsUniqueConstraintError(err) {
    // 处理唯一约束冲突，比如返回用户友好提示
    // 例：fmt.Println("该值已存在，请更换后重试")
  } else {
    // 处理其他类型错误
  }
}

2. 全局统一转换错误（可选） #

如果想让所有GORM操作的错误都自动转换成通用错误，可以注册一个全局错误回调，把数据库特定的唯一约束错误替换成我们定义的通用错误：

func init() {
  gorm.ErrCallback = func(db *gorm.DB) {
    if db.Error != nil && IsUniqueConstraintError(db.Error) {
      db.Error = ErrUniqueConstraintViolation
    }
  }
}

这样上层代码只需要判断错误是否为ErrUniqueConstraintViolation即可，完全不用关心底层数据库。

3. 扩展注意事项 #

• 后续新增支持的数据库时，只需要在IsUniqueConstraintError函数里补充对应数据库的错误码判断逻辑
• 这种方式不需要提前了解所有数据模型的约束，只针对数据库返回的错误做判断，符合你的需求
• 注意部分数据库可能有多个错误码对应唯一约束场景，需要确认驱动返回的具体错误信息后再添加判断

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