1、增加钱包余额

apps/user/internal/logic/user/userinfologic.go

@@ -32,27 +32,30 @@ func (l *UserInfoLogic) UserInfo(in *user.UserInfoReq) (*user.UserInfoResp, erro
 		return nil, errors.New("用户不存在")
 	}
 
-	// 查询企业信息
-	enterprise, err := l.svcCtx.EnterpriseModel.FindOneByUserId(l.ctx, users.Id)
-	if err != nil && !errors.Is(err, sql.ErrNoRows) {
-		return nil, errors.New("failed to query enterprise auth info")
-	}
-
-	if enterprise == nil {
+	if users.AuthStatus == "approved" {
+		enterprise, findEntErr := l.svcCtx.EnterpriseModel.FindOneByUserId(l.ctx, users.Id)
+		if findEntErr != nil && !errors.Is(findEntErr, sql.ErrNoRows) {
+			return nil, errors.New("failed to query enterprise auth info")
+		}
+		wallet, findWalletErr := l.svcCtx.WalletsModel.FindOne(l.ctx, in.UserId)
+		if findWalletErr != nil {
+			return nil, findWalletErr
+		}
+		// 正常返回用户和企业信息
 		return &user.UserInfoResp{
 			Username:             users.Username,
 			Phone:                users.Phone,
 			EnterpriseAuthStatus: users.AuthStatus,
+			EnterpriseName:       enterprise.EnterpriseName,
+			CreditCode:           enterprise.CreditCode,
+			LegalPerson:          enterprise.LegalPerson,
+			Balance:              wallet.Balance,
 		}, nil
 	}
-
-	// 正常返回用户和企业信息
 	return &user.UserInfoResp{
 		Username:             users.Username,
 		Phone:                users.Phone,
 		EnterpriseAuthStatus: users.AuthStatus,
-		EnterpriseName:       enterprise.EnterpriseName,
-		CreditCode:           enterprise.CreditCode,
-		LegalPerson:          enterprise.LegalPerson,
 	}, nil
+
 }