add ws 链接确认

services/percutaneous/internal/logic/getWsCodeLogic.go

@@ -30,17 +30,21 @@ func NewGetWsCodeLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetWsCo
 }
 
 func (l *GetWsCodeLogic) GetWsCode(in *percutaneous.GetWsCodeRequest) (*percutaneous.Response, error) {
-	adminInfo := utils.GetUserFromCtx(l.ctx)
-	if adminInfo.ID < utils.NumberOne {
-		return l.fail(utils.ErrorNoLoginInfo)
+	id := in.UserId
+	if id < utils.NumberOne {
+		adminInfo := utils.GetUserFromCtx(l.ctx)
+		if adminInfo.ID < utils.NumberOne {
+			return l.fail(utils.ErrorNoLoginInfo)
+		}
+		id = adminInfo.ID
 	}
 
-	Type := "app"
-	if in.Type == utils.NumberOne {
-		Type = "operation"
+	Type := WsTypeApp
+	if in.Type == utils.NumberTwo {
+		Type = WsTypeOperation
 	}
 
-	key := utils.MD5Encrypt(fmt.Sprintf("%d%s%s", adminInfo.ID, dao.WsCode, Type))
+	key := utils.MD5Encrypt(fmt.Sprintf("%d%s%s", id, dao.WsCode, Type))
 	redisKey := pkgws.RedisKeyQRSession + key
 
 	payload, err := jsoniter.MarshalToString(pkgws.Session{