From 286a33acdb3293f14127804ff33554f31af252cb Mon Sep 17 00:00:00 2001 From: crusader Date: Thu, 29 Mar 2018 18:13:34 +0900 Subject: [PATCH] ing --- servlet/data/gateway_rpc_handlers.go | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/servlet/data/gateway_rpc_handlers.go b/servlet/data/gateway_rpc_handlers.go index 2ba374c..c9e7b37 100644 --- a/servlet/data/gateway_rpc_handlers.go +++ b/servlet/data/gateway_rpc_handlers.go @@ -7,6 +7,7 @@ import ( cwf "git.loafle.net/commons_go/websocket_fasthttp" oocmp "git.loafle.net/overflow/overflow_commons_go/modules/probe" probeM "git.loafle.net/overflow/overflow_commons_go/modules/probe/model" + oogw "git.loafle.net/overflow/overflow_gateway_websocket" oogweg "git.loafle.net/overflow/overflow_gateway_websocket/external/grpc" "git.loafle.net/overflow/overflow_gateway_websocket/rpc" "github.com/valyala/fasthttp" @@ -56,6 +57,9 @@ func (sh *GatewayRPCHandlers) Handshake(socketCTX cwf.SocketContext, ctx *fastht return "", nil } + socketCTX.SetAttribute(oogw.ClientTypeKey, oogw.PROBE) + socketCTX.SetAttribute(oogw.TargetIDKey, probeKey) + return probeKey, nil }