feat(queries.getprofile): add condition to get comment reward

ShanghaitechGeekPie · Apr 9, 2024 · 8c10138 · 8c10138
1 parent 88ce7f3
commit 8c10138
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pkg/queries/user.go b/pkg/queries/user.go
@@ -349,6 +349,10 @@ func GetProfile(db *gorm.DB, id uint, uid uint) (models.ProfileResponse, error)
 		r := models.ProfileResponse{ID: id, Email: user.Email, Year: user.Year, Grade: user.Grade, NickName: user.NickName, RealName: user.RealName, IsAnonymous: user.IsAnonymous, Avatar: avatar, IsAdmin: user.IsAdmin, IsCommunityAdmin: user.IsCommunityAdmin, Reward: user.Reward}
 		if id == uid {
 			r.InvitationCode = user.InvitationCode
+			r.Reward = user.Reward
+		}
+		if user.IsAdmin || user.IsCommunityAdmin {
+			r.Reward = user.Reward
 		}
 		return r, nil
 	}