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 }