diff --git a/api/member.go b/api/member.go index b274b903aaa7f88ae4025a7a5e2db49fc3f89085..e52b327991d39a4fcfc96cfcb6eb0eda861cd265 100644 --- a/api/member.go +++ b/api/member.go @@ -2,6 +2,7 @@ package api import ( "encoding/json" + "errors" "log" "net/http" "strconv" @@ -90,7 +91,7 @@ func (a *api) getMemberNum(num int, w http.ResponseWriter, req *http.Request) { var member Member err := a.db.Where("num = ?", num).First(&member).Error if err != nil { - if err.Error() == "record not found" { + if errors.Is(err, gorm.ErrRecordNotFound) { w.WriteHeader(http.StatusNotFound) return } @@ -132,7 +133,7 @@ func (a *api) UpdateMember(w http.ResponseWriter, req *http.Request) { var dbMember Member err = a.db.Where("num = ?", vars["num"]).First(&dbMember).Error if err != nil { - if err.Error() == "record not found" { + if errors.Is(err, gorm.ErrRecordNotFound) { w.WriteHeader(http.StatusNotFound) return } diff --git a/api/order.go b/api/order.go index 4a65572b8e8d679eeea4493d7ff768cde63edd14..64d7564b26808567fe504720d56713e50daef0b5 100644 --- a/api/order.go +++ b/api/order.go @@ -2,6 +2,7 @@ package api import ( "encoding/json" + "errors" "log" "net/http" "time" @@ -130,7 +131,7 @@ func (a *api) GetOrder(num int, w http.ResponseWriter, req *http.Request) { err := a.db.Preload(clause.Associations). First(&order, vars["id"]).Error if err != nil { - if err.Error() == "record not found" { + if errors.Is(err, gorm.ErrRecordNotFound) { w.WriteHeader(http.StatusNotFound) return } diff --git a/api/product.go b/api/product.go index 07a8ca3c62e4a70cf27ef5f78fa409dc3e737924..e775a59cec1f246d492520d5f554c37e737a2fec 100644 --- a/api/product.go +++ b/api/product.go @@ -2,6 +2,7 @@ package api import ( "encoding/json" + "errors" "log" "net/http" @@ -65,7 +66,7 @@ func (a *api) GetProduct(w http.ResponseWriter, req *http.Request) { var product Product err := a.db.Where("code = ?", vars["code"]).First(&product).Error if err != nil { - if err.Error() == "record not found" { + if errors.Is(err, gorm.ErrRecordNotFound) { w.WriteHeader(http.StatusNotFound) return } @@ -107,7 +108,7 @@ func (a *api) UpdateProduct(w http.ResponseWriter, req *http.Request) { var dbProduct Product err = a.db.Where("code = ?", vars["code"]).First(&dbProduct).Error if err != nil { - if err.Error() == "record not found" { + if errors.Is(err, gorm.ErrRecordNotFound) { w.WriteHeader(http.StatusNotFound) return } diff --git a/api/transaction.go b/api/transaction.go index 2c58a8594335e767daf4ae3bddb072799b64a3ff..f1bd2093b5544b4748dcea052dd99fca577615ab 100644 --- a/api/transaction.go +++ b/api/transaction.go @@ -2,6 +2,7 @@ package api import ( "encoding/json" + "errors" "fmt" "log" "net/http" @@ -54,7 +55,7 @@ func (a *api) GetTransaction(num int, role string, w http.ResponseWriter, req *h Preload(clause.Associations). First(&transaction, vars["id"]).Error if err != nil { - if err.Error() == "record not found" { + if errors.Is(err, gorm.ErrRecordNotFound) { w.WriteHeader(http.StatusNotFound) return }