gpt4 book ai didi

postgresql - 在API响应中返回Postgres错误

转载 作者:行者123 更新时间:2023-12-03 10:08:55 24 4
gpt4 key购买 nike

我的代码中有两个简单的api方法。使用endpoind /api/user/create的方法创建用户。字段username是唯一的。当我尝试使用数据库中已经存在的相同用户名创建用户时,控制台出现错误:

(/home/andrej/go/src/go_contacts/models/users.go:19) 
[2020-12-23 22:03:10] pq: duplicate key value violates unique constraint "users_username_key"
我想显示此错误以响应用户,或者以某种方式在我的代码中标识错误的类型,以向用户显示不同的错误消息。我只知道如果我有错误 user会向我返回id = 0。但这对用户来说似乎不是一个好消息。 main.go
package main

import (
"fmt"
"go_contacts/controllers"
"net/http"
"os"

"github.com/gorilla/mux"
"github.com/joho/godotenv"
)

func main() {
godotenv.Load(".env")


router := mux.NewRouter()
router.HandleFunc("/", controllers.ReturnHello).Methods("GET")
router.HandleFunc("/api/user/create", controllers.CreateUser).Methods("POST")

port := os.Getenv("PORT")
if port == "" {
port = "8000"
}

err := http.ListenAndServe(":"+port, router)
if err != nil {
fmt.Print(err)
}
}
带有用户结构的 models.go:
package models

import (
u "go_contacts/utils"

"github.com/jinzhu/gorm"
)

// User base model
type User struct {
gorm.Model
Username string `json:"username" gorm:"unique"`
Password string `json:"password"`
Email string `json:"email"`
}

// Create new user
func (user *User) Create() map[string]interface{} {
GetDB().Create(user)

if user.ID <= 0 {
return u.Message(false, "Failed to create user, connection error.")
}

response := u.Message(true, "Account has been created")
response["user"] = user
return response
}

最佳答案

至于pq v1.5.2和gorm v1.9.12
首先,您需要确定create方法调用是否返回错误。像这样

err := GetDB().Create(user).Error
if err != nil {
// code to handle error
}
然后pq包具有特殊的类型,它映射到postgres服务器错误。它包含可以帮助您识别错误严重性的字段,代码,与错误相关的表等。
psql字段标识符的完整列表可在此处找到

Error and Notice Message Fields


https://www.postgresql.org/docs/current/protocol-error-fields.html
为了解决您的问题,我们可以使用field
Code
这是错误代码的字符串类型表示。但是首先,我们需要强制转换错误并检查类型转换是否成功。像这样
err := GetDB().Create(user).Error
if err != nil {
pqErr, ok := err.(pq.Error)
if !ok {
log.Fatal(err)
}

// code to handle specific error code
}
错误代码列表可以在官方的PostgreSQL文档 https://www.postgresql.org/docs/9.3/errcodes-appendix.html中找到
和实际的go pq特定映射到github.com/lib/pq/error.go中的错误,就像pq库一样
最后我们可以通过代码来处理错误
err := GetDB().Create(user).Error
if err != nil {
pqErr, ok := err.(pq.Error)
if !ok {
log.Fatal(err)
}

// note that type casting here is redundant and used for showing specific
// string type from pq library
if pqErr.Code == pq.ErrorCode("23505") { // 23505 is unique_violation error code for psql
// now you can create error and write it to a message to return it for
// a client
}
}

关于postgresql - 在API响应中返回Postgres错误,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/65429761/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com