Ma/origin
1
0
Fork 0
mirror of https://github.com/duanhf2012/origin.git synced 2026-02-04 06:54:45 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

新增TcpGateWay服务，支持通过配置进行路由转发

Browse Source
This commit is contained in:
boyce
2020-10-07 17:25:40 +08:00
parent 0c55961c44
commit a4d2a0c4ac
1 changed files with 33 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
sysservice/tcpgateway/TcpGateService.go
View File

@@ -32,12 +32,6 @@ type TcpGateService struct {
func (slf *TcpGateService) OnInit() error {
slf.OnLoad()
//获取安装好了的TcpService对象
slf.tcpService = node.GetService("TcpService").(*tcpservice.TcpService)
//新建内置的protobuf处理器，您也可以自定义路由器，比如json，后续会补充
slf.processor = processor.NewPBRawProcessor()
//注册监听客户连接断开事件
slf.processor.SetDisConnectedHandler(slf.router.OnDisconnected)
//注册监听客户连接事件
@@ -52,21 +46,50 @@ func (slf *TcpGateService) OnInit() error {
}
func (slf *TcpGateService) OnLoad() {
slf.loadBalance = &LoadBalance{}
slf.router = NewRouter(slf.loadBalance,slf,slf.GetServiceCfg())
//设置默认LoadBalance
if slf.loadBalance == nil {
slf.loadBalance = &LoadBalance{}
}
//设置默认Router
if slf.router == nil {
slf.router = NewRouter(slf.loadBalance,slf,slf.GetServiceCfg())
}
//新建内置的protobuf处理器，您也可以自定义路由器，比如json
if slf.processor == nil {
slf.processor = processor.NewPBRawProcessor()
}
//加载路由
slf.router.Load()
//设置默认的TcpService服务
if slf.tcpService == nil {
slf.tcpService = node.GetService("TcpService").(*tcpservice.TcpService)
}
if slf.tcpService == nil {
panic("TcpService is not installed!")
}
}
func (slf *TcpGateService) SetupLoadBalance(loadBalance ILoadBalance){
func (slf *TcpGateService) SetLoadBalance(loadBalance ILoadBalance){
slf.loadBalance = loadBalance
}
func (slf *TcpGateService) SetupRouter(router IRouter){
func (slf *TcpGateService) SetRouter(router IRouter){
slf.router = router
}
func (slf *TcpGateService) SetRawProcessor(processor processor.IRawProcessor){
slf.processor = processor
}
func (slf *TcpGateService) SetTcpGateService(tcpService *tcpservice.TcpService){
slf.tcpService = tcpService
}
func (slf *TcpGateService) RPC_Dispatch(replyMsg *ReplyMessage) error {
for _,id := range replyMsg.ClientList {
err := slf.tcpService.SendRawMsg(id,replyMsg.Msg)