Возвращать ошибки postgres в ответе API
В моем коде есть два простых метода API. Метод с конечной точкой /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
}
@Британцы, спасибо. Именно то, что я ищу
Ответы 2
Согласно комментариям, gorm предоставляет доступ к ошибкам базы данных из функции Создать запись следующим образом:
result := GetDB().Create(user)
if result.Error != nil {
// Do something with the error
}
Обратите внимание, что проверка строки ошибки, скорее всего, сделает базу данных вашего приложения специфичной (но для вас это может не быть проблемой). Ответы на этот вопрос могут оказать дополнительную помощь.
Что касается pq v1.5.2 и gorm v1.9.12
Во-первых, вам нужно определить, возвращает ли вызов метода create ошибку или нет. Вот так
err := GetDB().Create(user).Error
if err != nil {
// code to handle error
}
Затем пакет pq имеет специальный тип, который сопоставляется с ошибкой сервера postgres. Он содержит поля, которые могут помочь вам определить серьезность ошибки, ее код, таблицу, связанную с ошибкой и т. д.
Полный список идентификаторов полей psql можно найти здесь в
Поля сообщений об ошибках и уведомлениях
https://www.postgresql.org/docs/current/protocol-error-fields.html
Для решения вашей проблемы как вариант мы можем использовать поле
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
}
}
Рассматривали ли вы возможность проверить ошибку, возвращаемую
Create? (например,result := GetDB().Create(user); if result.Error != nil {...) Больше информации в документах здесь и здесь.