处理 gorm 错误返回时，有一些错误是没有办法直接使用 errors.Is 来进行判断的，比如主键冲突的错误，直接使用 errors.Is(err, gorm.ErrDuplicatedKey) 是无法判断出主键冲突的错误返回的。
如果没有办法进行判断，为什么 gorm 要给这样一个 error ，但又不能使用呢？
gorm.io/driver/mysql 包中有一个 error_translator 的 go 文件
package mysql
import (
"github.com/go-sql-driver/mysql"
"gorm.io/gorm")
// The error codes to map mysql errors to gorm errors, here is the mysql error codes reference https://dev.mysql.com/doc/mysql-errors/8.0/en/server-error-reference.html.
var errCodes = map[uint16]error{
1062: gorm.ErrDuplicatedKey,
1451: gorm.ErrForeignKeyViolated,
1452: gorm.ErrForeignKeyViolated,
}
func (dialector Dialector) Translate(err error) error {
if mysqlErr, ok := err.(*mysql.MySQLError); ok {
if translatedErr, found := errCodes[mysqlErr.Number]; found {
return translatedErr
}
return mysqlErr
}
return err
}
我们可以看到这个文件将 mysql 的几种错误码进行了枚举，使用 Translate 函数会将对应的 mysql error 转化为 gorm error
那这里的 Translate 函数，是谁进行使用了呢？在什么时候进行使用了呢？
主键冲突的错误一定是出现在插入的时候，我们顺着 gorm 的 Create 方法向下找，可以发现它调用了一个 AddError 的函数，如下
// AddError add error to dbfunc (db *DB) AddError(err error) error {
if err != nil {
if db.Config.TranslateError {
if errTranslator, ok := db.Dialector.(ErrorTranslator); ok {
err = errTranslator.Translate(err)
}
}
if db.Error == nil {
db.Error = err
} else {
db.Error = fmt.Errorf("%v; %w", db.Error, err)
}
}
return db.Error
}
这里有一行很关键，db.Dialector.(ErrorTranslator) 对 Dialector 接口进行了断言，断言成功，就调用对应 Dialector 的 Translate 函数，而当这里的 Dialector 是上面 gorm.io/driver/mysql 中的 Dialector 时，就可以运行上面的翻译逻辑，将 mysql 的 error 转换为 gorm 的 error。
那么，我们下一步就是找到方法，把这里使用的 Dialector 替换成 gorm.io/driver/mysql 包下的这个 Dialector，这样我们就可以使用 errors.Is(err, gorm.ErrDuplicatedKey) 对插入冲突进行判断了。
gorm 中 DB 对象的结构是这样的
// DB GORM DB definition
type DB struct {
*Config
Error error
RowsAffected int64
Statement *Statement
clone int
}
这里的 Config 中就包含了 Dialector 接口，我们只需要在创建 gorm.DB 的时候，将接口的实例（gorm.io/driver/mysql 包下的这个）传入进去，就可以让 gorm 在之后的 error 判断时，对 mysql 的 error 进行翻译。
到这儿，原理部分就明明白白了，接下来简单改写一下 gorm.DB 的 init 过程即可！⬇️
import "gorm.io/driver/mysql"
import "gorm.io/gorm"
connUrl := "数据库连接地址"
db, err := gorm.Open(
mysql.Open(connUrl).(*mysql.Dialector),
TranslateError: true, // 开启 mysql 方言翻译 (开启后 duplicatedKey err 判断才能生效)
)
以上！
参考链接：https://gorm.io/docs/error_handling.html#Dialect-Translated-Errors