pkg/service/bundle/bundleBalance.go

@@ -131,10 +131,12 @@ func GetToBeComfirmedWorks(c *gin.Context) {
 }
 
 func GetUserBalance(c *gin.Context) {
-	userInfo := login.GetUserInfoFromC(c)
-	res, err := service.BundleProvider.GetBundleBalanceByUserId(context.Background(), &bundle.GetBundleBalanceByUserIdReq{
-		UserId: int32(userInfo.ID),
-	})
+	var req bundle.GetBundleBalanceByUserIdReq
+	if err := c.ShouldBindJSON(&req); err != nil {
+		service.Error(c, err)
+		return
+	}
+	res, err := service.BundleProvider.GetBundleBalanceByUserId(context.Background(), &req)
 	if err != nil {
 		service.Error(c, err)
 		return