From de4bcd1dead50b05f716bc5718be5540bdb96783 Mon Sep 17 00:00:00 2001
From: liujiandao <274878379@qq.com>
Date: 星期日, 28 四月 2024 17:23:29 +0800
Subject: [PATCH] fix
---
api/v1/test/member.go | 44 ++++++++++++++++++++++++++++++++++----------
1 files changed, 34 insertions(+), 10 deletions(-)
diff --git a/api/v1/test/member.go b/api/v1/test/member.go
index f0af64d..a8bd70a 100644
--- a/api/v1/test/member.go
+++ b/api/v1/test/member.go
@@ -1,17 +1,24 @@
package test
import (
+ "context"
"github.com/gin-gonic/gin"
"go.uber.org/zap"
"google.golang.org/grpc"
"google.golang.org/grpc/credentials/insecure"
+ "google.golang.org/grpc/metadata"
+ "srm/constvar"
"srm/global"
"srm/model/common/request"
"srm/model/common/response"
"srm/model/test"
testReq "srm/model/test/request"
+ "srm/pkg/contextx"
+ "srm/pkg/ecode"
"srm/proto/user"
"srm/service"
+ "strings"
+ "time"
)
type MemberApi struct {
@@ -207,9 +214,25 @@
defer CloseUserConn()
cli := user.NewUserServiceClient(userConn)
- var users []*user.User
- getMemberListResponse, err := cli.SyncUser(c, &user.UserRequest{Users: users})
-
+ ctx, cancel := context.WithTimeout(context.Background(), time.Second*5)
+ defer cancel()
+ slices := strings.Split(c.GetHeader("Authorization"), " ")
+ if len(slices) != 2 {
+ ctx := new(contextx.Context).SetCtx(c)
+ ctx.Fail(ecode.JWTEmpty)
+ return
+ }
+ token := slices[1]
+ // 鍒涘缓涓�涓嚜瀹氫箟鐨� metadata
+ md := metadata.Pairs("token", token)
+ // 灏� metadata 娣诲姞鍒� context 涓�
+ ctx = metadata.NewOutgoingContext(ctx, md)
+ getMemberListResponse, err := cli.SyncAllUser(ctx, &user.AllUserRequest{SystemType: constvar.SystemTypeSrm})
+ if err != nil {
+ global.GVA_LOG.Error("user.AllUserRequest fail:", zap.Error(err))
+ response.FailWithMessage("鑾峰彇澶辫触", c)
+ return
+ }
rawMemberList := getMemberListResponse.List
memberList := make([]*test.Member, len(rawMemberList))
@@ -224,12 +247,13 @@
if err != nil || len(memberList) == 0 {
global.GVA_LOG.Error("鑾峰彇澶辫触!", zap.Error(err))
response.FailWithMessage("鑾峰彇澶辫触", c)
- } else {
- response.OkWithDetailed(response.PageResult{
- List: memberList,
- Total: int64(len(memberList)),
- Page: pageInfo.Page,
- PageSize: pageInfo.PageSize,
- }, "鑾峰彇鎴愬姛", c)
+ return
}
+ response.OkWithDetailed(response.PageResult{
+ List: memberList,
+ Total: getMemberListResponse.Total,
+ Page: pageInfo.Page,
+ PageSize: pageInfo.PageSize,
+ }, "鑾峰彇鎴愬姛", c)
+
}
--
Gitblit v1.8.0