使用GORM检查数据库用户是否存在时如何避免无匹配记录的错误日志

问题原因 #

你当前的业务逻辑是正确的，日志打印是GORM的默认行为导致的：GORM默认日志配置会将gorm.ErrRecordNotFound识别为错误级别的日志输出，而创建新用户时必然会触发「查询不到用户」的分支，所以正常业务逻辑被误打印为错误日志。

可选解决方案 #

方案1：使用Exists查询（最推荐） #

直接查询是否存在对应记录，不需要加载完整用户结构体，性能更高，同时不会触发ErrRecordNotFound错误：

var exists bool
// 只查是否存在，不返回完整用户数据
err := connection.DB.Model(&models.User{}).
    Select("count(*) > 0").
    Where("username = ?", user.Username).
    Find(&exists).Error
if err != nil {
    return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
        "error": "internal server error",
    })
}

// 用户名已存在
if exists {
    return c.Status(fiber.StatusConflict).JSON(fiber.Map{
        "error": "username already exists",
    })
}

// 不存在则创建用户
if err := connection.DB.Create(&user).Error; err != nil {
    return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
        "error": "internal server error",
    })
}
return c.Status(fiber.StatusCreated).JSON(fiber.Map{
    "message": "user created",
})

方案2：全局配置GORM日志忽略该错误 #

如果项目中大量使用First判断记录存在，不想逐个修改查询逻辑，可以在初始化数据库连接时配置日志，直接忽略ErrRecordNotFound类错误：

import (
    "log"
    "os"
    "gorm.io/gorm/logger"
)

// 初始化自定义日志
customLogger := logger.New(
    log.New(os.Stdout, "\r\n", log.LstdFlags),
    logger.Config{
        IgnoreRecordNotFoundError: true, // 开启忽略记录未找到错误
        LogLevel:                  logger.Info,
    },
)

// 初始化DB时传入自定义日志配置
db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{
    Logger: customLogger,
})

方案3：使用FirstOrCreate一步完成操作 #

GORM自带的FirstOrCreate方法可以同时完成「查询存在则返回，不存在则创建」的逻辑，代码更简洁：

result := models.User{}
// 按username查询，不存在则创建传入的user对象
err := connection.DB.Where("username = ?", user.Username).FirstOrCreate(&result, user).Error
if err != nil {
    return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
        "error": "internal server error",
    })
}

// 已存在的记录不会触发创建操作，RowsAffected为0，以此判断是否是新创建的用户
if connection.DB.RowsAffected == 0 {
    return c.Status(fiber.StatusConflict).JSON(fiber.Map{
        "error": "username already exists",
    })
}

return c.Status(fiber.StatusCreated).JSON(fiber.Map{
    "message": "user created",
})

注意事项 #

无论使用哪种方案，都建议给users表的username字段添加唯一索引，避免高并发场景下出现重复插入的问题。

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