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

Question

I have something like this and I would like to check if email already exist in DB:

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",
})
}

With this code, every time is telling me that: Email already exist, only works for the first time.

Answer 1

plase read the document: https://gorm.io/docs/query.html

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

Answer 2

Since, your struct object is not a slice. You should use ErrRecordNotFound .

Note: ErrRecordNotFound only works with First , Last , Take which is expected to return some result. And RecordNotFound is removed in 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
}

OR

If you want to avoid the ErrRecordNotFound error, you could use Find like db.Limit(1).Find(&user), the Find method accepts both struct and slice data. And check like this:

result.RowsAffected // returns count of records found

For better understanding refer the link here: https://gorm.io/docs/v2_release_note.html#ErrRecordNotFound and https://gorm.io/docs/query.html

And, If you want to add record in DB though email exist then you should remove unique constraint and also check the error while creating the record. If record successfully created then return success response else return the appropriate error message.

Answer 3

1. you should validate the input after binding and before db queries
2. alter your email column to be unique
3. try to insert the validated data to db
   • if success => 200 (there was no similar email)
   • if err => check err code

for example:

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

For more Information, you should look GoDoc pg lib and Possible Error Codes
and then, then you can return a suitable error code

btw. hopefully you don't save clear passwords to db:D

Answer 4

ITS FINALLY WORKING ! Thanks for everyone thats answered !