dobyte / due Goto Github PK

err4 := r.Proxy().Push(4, route.UploadUserPosition, &pb.PlayerPositionUploadReq{
	UserId: msg.GetUserId(),
	Data: &pb.PositionData{
		Pos: &pb.Vector3{
			X: 1,
			Y: 1,
			Z: 1,
		},
		Rotation:  &pb.Vector3{
			X: 1,
			Y: 1,
			Z: 1,
		},
		Speed:  &pb.Vector3{
			X: 4,
			Y: 3,
			Z: 2,
		},
	},
})
if err4 != nil {
	log.Errorf("upload player position error: %v", err4)
}


err4 = r.Proxy().Push(5, route.UploadUserPosition, &pb.PlayerPositionUploadReq{
	UserId: msg.GetUserId(),
	Data: &pb.PositionData{
		Pos: &pb.Vector3{
			X: 3,
			Y: 3,
			Z: 3,
		},
		Rotation:  &pb.Vector3{
			X: 3,
			Y: 3,
			Z: 4,
		},
		Speed:  &pb.Vector3{
			X: 4,
			Y: 3,
			Z: 2,
		},
	},
})
if err4 != nil {
	log.Errorf("upload player position error: %v", err4)
}


err4 = r.Proxy().Push(6, route.UploadUserPosition, &pb.PlayerPositionUploadReq{
	UserId: msg.GetUserId(),
	Data: &pb.PositionData{
		Pos: &pb.Vector3{
			X: 6,
			Y: 7,
			Z: 8,
		},
		Rotation:  &pb.Vector3{
			X: 23,
			Y: 32,
			Z: 24,
		},
		Speed:  &pb.Vector3{
			X: 4,
			Y: 3,
			Z: 2,
		},
	},
})
if err4 != nil {
	log.Errorf("upload player position error: %v", err4)
}

err4 = r.Proxy().Push(7, route.UploadUserPosition, &pb.PlayerPositionUploadReq{
	UserId: msg.GetUserId(),
	Data: &pb.PositionData{
		Pos: &pb.Vector3{
			X: 16,
			Y: 27,
			Z: 18,
		},
		Rotation:  &pb.Vector3{
			X: 3,
			Y: 3,
			Z: 2,
		},
		Speed:  &pb.Vector3{
			X: 4,
			Y: 3,
			Z: 2,
		},
	},
})
if err4 != nil {
	log.Errorf("upload player position error: %v", err4)
}

n.Proxy().Router().Group(func(group *node.RouterGroup) {
		group.Middleware(middleware.RecordMetric1)
		group.Middleware(middleware.RecordMetric2)
		group.AddRouteHandler(int32(pb_lobby.ROUTE_Echo1), false, EchoHandler)
})

// RecordMetric1 指标
func RecordMetric1(ctx *node.Context) {
	logger.Info("RecordMetric1 in")
	start := time.Now()
	ctx.Middleware.Next(ctx)

	logger.Info("RecordMetric1 out", time.Since(start))
}

// RecordMetric2 指标
func RecordMetric2(ctx *node.Context) {
	logger.Info("RecordMetric2 in")
	start := time.Now()
	ctx.Middleware.Next(ctx)

	logger.Info("RecordMetric2 out", time.Since(start))
} 

func EchoHandler(ctx *node.Context) {
	logger.Info("EchoHandler in")
	time.Sleep(2 * time.Second)
	logger.Info("EchoHandler out")
}

得到的日志
 [RecordMetric1 in]
 [RecordMetric2 in]
 [RecordMetric2 out 128ns]
 [RecordMetric1 out 121.798µs]
 [EchoHandler in]
 [EchoHandler out]

期望得到
 [RecordMetric1 in]
 [RecordMetric2 in]
 [EchoHandler in]
 [EchoHandler out]
 [RecordMetric2 out 128ns]
 [RecordMetric1 out 121.798µs]

 type Middleware struct {
	index        int
	middlewares  []MiddlewareHandler
+	routeHandler RouteHandler
 }

 // 重置中间件
-func (m *Middleware) reset(middlewares []MiddlewareHandler) {
+func (m *Middleware) reset(middlewares []MiddlewareHandler, routeHandler RouteHandler) {
 	m.index = -1
 	m.middlewares = middlewares
+	m.routeHandler = routeHandler
 }

func (m *Middleware) Skip(ctx *Context, skip int) {
	m.index += skip
	if m.isFinished() {
+		m.routeHandler(ctx)
		return
	}

 	if ok {
 		if len(route.middlewares) > 0 {
-			ctx.Middleware.reset(route.middlewares)
+			ctx.Middleware.reset(route.middlewares, route.handler)
 			ctx.Middleware.Next(ctx)
-
-			if ctx.Middleware.isFinished() {
-				route.handler(ctx)
-			}
 		} else {

read message failed: 5 read tcp [10.0.4.3:3553](http://10.0.4.3:3553/)->[171.216.69.26:5632](http://171.216.69.26:5632/): read: connection reset by peer

websocket upgrade error: websocket: the client is not using the websocket protocol: 'upgrade' token not found in 'Connection' header