From 0bdaa141bd9f98ad146d41f506a2d02111e583f2 Mon Sep 17 00:00:00 2001
From: liangzai <2440983361@qq.com>
Date: Wed, 18 Jun 2025 02:40:25 +0800
Subject: [PATCH] fix revenuei

---
 .../logic/agent/getagentrevenueinfologic.go   | 26 +++++++++++++++++--
 1 file changed, 24 insertions(+), 2 deletions(-)

diff --git a/app/main/api/internal/logic/agent/getagentrevenueinfologic.go b/app/main/api/internal/logic/agent/getagentrevenueinfologic.go
index 544edae..6d1626b 100644
--- a/app/main/api/internal/logic/agent/getagentrevenueinfologic.go
+++ b/app/main/api/internal/logic/agent/getagentrevenueinfologic.go
@@ -31,9 +31,31 @@ func NewGetAgentRevenueInfoLogic(ctx context.Context, svcCtx *svc.ServiceContext
 }
 
 func (l *GetAgentRevenueInfoLogic) GetAgentRevenueInfo(req *types.GetAgentRevenueInfoReq) (resp *types.GetAgentRevenueInfoResp, err error) {
-	userID, err := ctxdata.GetUidFromCtx(l.ctx)
+	claims, err := ctxdata.GetClaimsFromCtx(l.ctx)
 	if err != nil {
-		return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取代理奖励, %v", err)
+		return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取代理信息, %v", err)
+	}
+	userID := claims.UserId
+	userType := claims.UserType
+	if userType == model.UserTypeTemp {
+		return &types.GetAgentRevenueInfoResp{
+			Balance:       0,
+			TotalEarnings: 0,
+			FrozenBalance: 0,
+			DirectPush: types.DirectPushReport{
+				TotalCommission: 0,
+				TotalReport:     0,
+				Today:           types.TimeRangeReport{},
+				Last7D:          types.TimeRangeReport{},
+				Last30D:         types.TimeRangeReport{},
+			},
+			ActiveReward: types.ActiveReward{
+				TotalReward: 0,
+				Today:       types.ActiveRewardData{},
+				Last7D:      types.ActiveRewardData{},
+				Last30D:     types.ActiveRewardData{},
+			},
+		}, nil
 	}
 	agentModel, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, userID)
 	if err != nil {