overflow_backup_2018_0455/overflow_gateway_websocket
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

ing

Browse Source
This commit is contained in:
crusader 2018-03-19 16:19:34 +09:00
parent 74b55942b9
commit 4c8bb08e99
2 changed files with 16 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
constants.go
View File

@ -9,9 +9,11 @@ import (
var ( var (
ServletSocketKey = cuc.ContextKey("ServletSocket") ServletSocketKey = cuc.ContextKey("ServletSocket")
ClientTypeKey = cuc.ContextKey("ClientType")
SocketIDKey = cuc.ContextKey("SocketID") GRPCMetadataKey = cuc.ContextKey("GRPCMetadata")
TargetIDKey = cuc.ContextKey("TargetID") ClientTypeKey = cuc.ContextKey("ClientType")
SocketIDKey = cuc.ContextKey("SocketID")
TargetIDKey = cuc.ContextKey("TargetID")
) )
const ( const (

16
internal/server/rpc/gateway_rpc_servlet_handlers.go
View File

@ -24,11 +24,17 @@ type GatewayRPCServletHandlers struct {
} }
func (sh *GatewayRPCServletHandlers) Invoke(servletCTX rpc.ServletContext, requestCodec protocol.RegistryCodec) (result interface{}, err error) { func (sh *GatewayRPCServletHandlers) Invoke(servletCTX rpc.ServletContext, requestCodec protocol.RegistryCodec) (result interface{}, err error) {
md := metadata.Pairs( md := servletCTX.GetAttribute(oogw.GRPCMetadataKey)
oogw.GRPCClientTypeKey, servletCTX.GetAttribute(oogw.SocketIDKey).(string), if nil == md {
oogw.GRPCSessionIDKey, servletCTX.GetAttribute(oogw.SocketIDKey).(string), md = metadata.Pairs(
oogw.GRPCTargetIDKey, servletCTX.GetAttribute(oogw.SocketIDKey).(string)) oogw.GRPCClientTypeKey, servletCTX.GetAttribute(oogw.SocketIDKey).(string),
grpcCTX := metadata.NewOutgoingContext(context.Background(), md) oogw.GRPCSessionIDKey, servletCTX.GetAttribute(oogw.SocketIDKey).(string),
oogw.GRPCTargetIDKey, servletCTX.GetAttribute(oogw.SocketIDKey).(string))
servletCTX.SetAttribute(oogw.GRPCMetadataKey, md)
}
grpcCTX := metadata.NewOutgoingContext(context.Background(), md.(metadata.MD))
params, err := requestCodec.Params() params, err := requestCodec.Params()
if nil != err { if nil != err {