This commit is contained in:
crusader 2018-03-29 18:13:34 +09:00
parent 90c785e9e0
commit 286a33acdb

View File

@ -7,6 +7,7 @@ import (
cwf "git.loafle.net/commons_go/websocket_fasthttp" cwf "git.loafle.net/commons_go/websocket_fasthttp"
oocmp "git.loafle.net/overflow/overflow_commons_go/modules/probe" oocmp "git.loafle.net/overflow/overflow_commons_go/modules/probe"
probeM "git.loafle.net/overflow/overflow_commons_go/modules/probe/model" 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" oogweg "git.loafle.net/overflow/overflow_gateway_websocket/external/grpc"
"git.loafle.net/overflow/overflow_gateway_websocket/rpc" "git.loafle.net/overflow/overflow_gateway_websocket/rpc"
"github.com/valyala/fasthttp" "github.com/valyala/fasthttp"
@ -56,6 +57,9 @@ func (sh *GatewayRPCHandlers) Handshake(socketCTX cwf.SocketContext, ctx *fastht
return "", nil return "", nil
} }
socketCTX.SetAttribute(oogw.ClientTypeKey, oogw.PROBE)
socketCTX.SetAttribute(oogw.TargetIDKey, probeKey)
return probeKey, nil return probeKey, nil
} }