Retornar erros de postgres na resposta da API
Tenho dois métodos simples de API em meu código. O método com endpoind /api/user/create
cria o usuário. O campo username
é único. Quando tento criar um usuário com o mesmo nome de usuário que já existe no banco de dados, ocorre um erro no console:
(/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"
Quero mostrar esse erro em resposta ao usuário, ou de alguma forma identificar o tipo de erro em meu código, para mostrar diferentes mensagens de erro para o usuário. Eu sei apenas que se eu tiver erro user
retornará id = 0 Mas não parece uma boa mensagem para o usuário.
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
com estrutura de usuário:
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
}
Respostas
Quanto a pq v1.5.2 e gorm v1.9.12
Primeiro, você precisa identificar se a chamada do método de criação retorna um erro ou não. Igual a
err := GetDB().Create(user).Error
if err != nil {
// code to handle error
}
Então, o pacote pq tem um tipo especial que mapeia para o erro do servidor postgres. Contém campos que podem ajudá-lo a identificar a gravidade do erro, seu código, tabela que relaciona ao erro etc.
A lista completa de identificadores de campos do psql pode ser encontrada aqui em
Campos de mensagem de erro e aviso
https://www.postgresql.org/docs/current/protocol-error-fields.html
Para resolver seu problema como uma opção, podemos usar o campo
Code
Que é uma representação do tipo string do código de erro. Mas, primeiro, precisamos lançar um erro e verificar o tipo de conversão foi bem-sucedido. Igual a
err := GetDB().Create(user).Error
if err != nil {
pqErr, ok := err.(pq.Error)
if !ok {
log.Fatal(err)
}
// code to handle specific error code
}
A lista de códigos de erro pode ser encontrada na documentação oficial do postgresql https://www.postgresql.org/docs/9.3/errcodes-appendix.html
E os mapeamentos específicos go pq reais para o erro no github.com/lib/pq/error.go como para a biblioteca pq
E, finalmente, podemos tratar o erro como duplicado por código
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
}
}
De acordo com o gorm de comentários, fornece acesso a erros de banco de dados a partir da função Criar registro da seguinte maneira:
result := GetDB().Create(user)
if result.Error != nil {
// Do something with the error
}
Observe que verificar a string de erro provavelmente tornará o banco de dados do aplicativo específico (mas isso pode não ser um problema para você). As respostas a esta pergunta podem fornecer mais assistência.