使用 golang、gin 和 gorm 以及 email 等 postgres 进行验证已经存在于数据库中

Question

我有这样的事情，我想检查 DB 中是否已经存在 email：

func RegisterUser(c *gin) {
var user models.User
if err := c.ShouldBindJSON(&user); err != nil {
    c.JSON(http.StatusBadRequest, gin.H{
        "messAge": err.Error(),
        "data":    "",
    })
    return
}
// **I TRIED SOEMTHING LIKE THIS**
err := database.DB.Find(&user.Email).Error
if err != nil {
    c.JSON(401, gin.H{"MESSAGE": "Email ALREADY exist",
    return
}
// **but is not working, because ANY mail it give me error**


if !strings.Contains(user.Email, "@") {
    c.JSON(400, gin.H{"MESSAGE": utils.ErrEmailWrong})
    return
}

if len(user.Password) < 4 {
    c.JSON(400, gin.H{"MESSAGE": utils.ErrPasswordLength})
    return
}

database.DB.Create(&user)
c.JSON(200, gin.H{
    "MESSAGE": "CREATED",
})
}

有了这段代码，每次都在告诉我：Email 已经存在，只适用于第一次。

Answer 1

请阅读文档： https://gorm.io/docs/query.html

var userFind models.User
database.DB.Where("email = ?", user.Email).First(&userFind)

Answer 2

因为，您的结构 object 不是切片。 您应该使用ErrRecordNotFound 。

注意： ErrRecordNotFound仅适用于First ， Last ， Take预计会返回一些结果。 并且RecordNotFound在 V2 中被删除。

if err != nil {
     if errors.Is(err, gorm.ErrRecordNotFound){
           c.JSON(401, gin.H{"MESSAGE": "Email Not Found",
           return

     }
    c.JSON(401, gin.H{"MESSAGE": "Your Message",
    return
}

或者

如果你想避免ErrRecordNotFound错误，你可以像db.Limit(1).Find(&user),一样使用Find ， Find方法接受结构和切片数据。 并像这样检查：

result.RowsAffected // returns count of records found

为了更好地理解，请参阅此处的链接： https://gorm.io/docs/v2_release_note.html#ErrRecordNotFound和https://gorm.io/docs/query.ZFC35FDC70D5FC69D2569883

而且，如果您想在 DB 中添加记录，尽管 email 存在，那么您应该删除唯一约束并在创建记录时检查错误。 如果成功创建记录，则返回成功响应，否则返回相应的错误消息。

Answer 3

1. 您应该在绑定之后和数据库查询之前验证输入
2. 将您的 email 列更改为唯一
3. 尝试将经过验证的数据插入到 db
   • 如果成功 => 200（没有类似的电子邮件）
   • if err => 检查错误码

例如：

func IsUniqueContraintViolation(err error) bool {
        if pgError, ok := err.(*pgconn.PgError); ok && errors.Is(err, pgError) {
            if pgError.Code == "23505" {
                return true
            }
        }
    
        return false
    }

有关更多信息，您应该查看GoDoc pg lib和可能的错误代码
然后，然后您可以返回一个合适的错误代码

顺便提一句。 希望您不要将清晰的密码保存到 db:D

Answer 4

它终于工作了！ 谢谢大家回答！