Improve error handling of non-existing rows

fabiante · Sep 24, 2023 · 0fa517c · 0fa517c
1 parent f44523e
commit 0fa517c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/app/service_user.go b/app/service_user.go
@@ -35,6 +35,8 @@ func (s *UserService) GetUser(email string) (*models.User, error) {
 
 	err := s.db.Take(user, "email = ?", email).Error
 	switch {
+	case errors.Is(err, gorm.ErrRecordNotFound):
+		return nil, ErrNotFound
 	case err != nil:
 		return nil, mapDBError(err)
 	default:
@@ -79,6 +81,8 @@ func (s *UserService) GetUserKey(value string) (*models.UserKey, error) {
 
 	err := s.db.Take(key, "value = ?", value).Error
 	switch {
+	case errors.Is(err, gorm.ErrRecordNotFound):
+		return nil, ErrNotFound
 	case err != nil:
 		return nil, mapDBError(err)
 	default: