db: use gorm function to check `ErrRecordNotFound` (#6060)

1 files changed, 2 insertions, 2 deletions

diff --git a/internal/db/users.go b/internal/db/users.go
index cadd106c..006e2089 100644
--- a/internal/db/users.go
+++ b/internal/db/users.go
@@ -117,7 +117,7 @@ func (db *users) GetByID(id int64) (*User, error) {
 	user := new(User)
 	err := db.Where("id = ?", id).First(user).Error
 	if err != nil {
-		if err == gorm.ErrRecordNotFound {
+		if gorm.IsRecordNotFoundError(err) {
 			return nil, ErrUserNotExist{args: map[string]interface{}{"userID": id}}
 		}
 		return nil, err
@@ -129,7 +129,7 @@ func (db *users) GetByUsername(username string) (*User, error) {
 	user := new(User)
 	err := db.Where("lower_name = ?", strings.ToLower(username)).First(user).Error
 	if err != nil {
-		if err == gorm.ErrRecordNotFound {
+		if gorm.IsRecordNotFoundError(err) {
 			return nil, ErrUserNotExist{args: map[string]interface{}{"name": username}}
 		}
 		return nil, err