github/nps
1
0
Fork 0
mirror of https://github.com/ehang-io/nps.git synced 2025-09-03 12:36:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

module

Browse Source
This commit is contained in:
unknown
2019-10-15 13:59:08 +08:00
parent 8f45d86cee
commit f4ae503982
2 changed files with 14 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
server/common/common_inet_proxy_handle.go
View File

@@ -2,7 +2,6 @@ package common
import (
"context"
"github.com/cnlh/nps/bridge"
"github.com/cnlh/nps/core"
"net"
)
@@ -20,25 +19,20 @@ func (proxy *Proxy) GetConfigName() *core.NpsConfigs {
func (proxy *Proxy) Run(ctx context.Context, config map[string]string) error {
proxy.clientConn = proxy.GetClientConn(ctx)
proxy.ctx = ctx
bg := ctx.Value(core.BRIDGE)
if bg == nil {
return core.BRIDGE_NOT_EXIST
}
clientCtxConn := ctx.Value(core.CLIENT_CONNECTION)
if clientCtxConn == nil {
return core.CLIENT_CONNECTION_NOT_EXIST
}
clientId := ctx.Value(core.CLIENT_ID)
if clientId == nil {
return core.CLIENT_ID_NOT_EXIST
}
clientId := proxy.GetClientId(ctx)
brg := bg.(*bridge.Bridge)
severConn, err := brg.GetConnByClientId(clientId.(int))
brg := proxy.GetBridge(ctx)
severConn, err := brg.GetConnByClientId(clientId)
if err != nil {
return err
}
go core.CopyBuffer(severConn, clientCtxConn.(net.Conn))
core.CopyBuffer(clientCtxConn.(net.Conn), severConn)
return nil