fix(queries.Register): handle error from updating inviter's reward

ShanghaitechGeekPie · May 23, 2024 · 570468a · 570468a
1 parent bccfdda
commit 570468a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pkg/queries/user.go b/pkg/queries/user.go
@@ -131,7 +131,10 @@ func Register(db *gorm.DB, u *models.User, invitation_code string) error {
 		u.InvitedByUserID = inviter.ID
 		// TODO: only once for the inviter?
 		inviter.Reward += 100
-		db.Select("reward").Save(inviter)
+		err = db.Select("reward").Save(inviter).Error
+		if err != nil {
+			return errors.Wrap(err, errors.DatabaseError)
+		}
 	}
 
 	// 检查邮箱是否已存在