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"
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
}