提交origin2.0版本

sysservice/tcpservice.go

@@ -0,0 +1,137 @@
+package sysservice
+
+import (
+	"fmt"
+	"github.com/duanhf2012/originnet/event"
+	"github.com/duanhf2012/originnet/log"
+	"github.com/duanhf2012/originnet/network"
+	"github.com/duanhf2012/originnet/service"
+	"sync"
+)
+
+type TcpService struct {
+	tcpServer network.TCPServer
+	service.Service
+
+	tcpService *TcpService
+	mapClientLocker sync.RWMutex
+	mapClient map[uint64] *Client
+	initClientId uint64
+	process network.Processor
+}
+
+const Default_MaxConnNum = 3000
+const Default_PendingWriteNum = 10000
+const Default_LittleEndian = false
+const Default_MinMsgLen = 2
+const Default_MaxMsgLen = 65535
+
+
+
+func (slf *TcpService) OnInit() error{
+	iConfig := slf.GetServiceCfg()
+	if iConfig == nil {
+		return fmt.Errorf("%s service config is error!",slf.GetName())
+	}
+	tcpCfg := iConfig.(map[string]interface{})
+	addr,ok := tcpCfg["ListenAddr"]
+	if ok == false {
+		return fmt.Errorf("%s service config is error!",slf.GetName())
+	}
+	slf.tcpServer.Addr = addr.(string)
+	slf.tcpServer.MaxConnNum = Default_MaxConnNum
+	slf.tcpServer.PendingWriteNum = Default_PendingWriteNum
+	slf.tcpServer.LittleEndian = Default_LittleEndian
+	slf.tcpServer.MinMsgLen = Default_MinMsgLen
+	slf.tcpServer.MaxMsgLen = Default_MaxMsgLen
+	MaxConnNum,ok := tcpCfg["MaxConnNum"]
+	if ok == true {
+		slf.tcpServer.MaxConnNum = int(MaxConnNum.(float64))
+	}
+	PendingWriteNum,ok := tcpCfg["PendingWriteNum"]
+	if ok == true {
+		slf.tcpServer.PendingWriteNum = int(PendingWriteNum.(float64))
+	}
+	LittleEndian,ok := tcpCfg["LittleEndian"]
+	if ok == true {
+		slf.tcpServer.LittleEndian = LittleEndian.(bool)
+	}
+	MinMsgLen,ok := tcpCfg["MinMsgLen"]
+	if ok == true {
+		slf.tcpServer.MinMsgLen = uint32(MinMsgLen.(float64))
+	}
+	MaxMsgLen,ok := tcpCfg["MaxMsgLen"]
+	if ok == true {
+		slf.tcpServer.MaxMsgLen = uint32(MaxMsgLen.(float64))
+	}
+	slf.tcpServer.NewAgent =slf.NewClient
+	slf.tcpServer.Start()
+	//加载配置
+	return nil
+}
+
+func (slf *TcpService) SetProcessor(process network.Processor){
+	slf.process = process
+}
+
+func (slf *TcpService) NewClient(conn *network.TCPConn) network.Agent {
+	slf.mapClientLocker.Lock()
+	defer slf.mapClientLocker.Unlock()
+
+	for {
+		slf.initClientId+=1
+		_,ok := slf.mapClient[slf.initClientId]
+		if ok == true {
+			continue
+		}
+
+		pClient := &Client{tcpConn:conn}
+		slf.mapClient[slf.initClientId] = pClient
+		return pClient
+	}
+
+	return nil
+}
+
+
+
+type Client struct {
+	id uint64
+	tcpConn *network.TCPConn
+	tcpService *TcpService
+}
+
+type TcpPack struct {
+	ClientId uint64
+	Data interface{}
+}
+
+func (slf *Client) GetId() uint64 {
+	return slf.id
+}
+
+func (slf *Client) Run() {
+	slf.tcpService.GetEventReciver().NotifyEvent(&event.Event{Type:event.Sys_Event_Tcp_Connected,Data:&TcpPack{ClientId:slf.id}})
+	for{
+		bytes,err := slf.tcpConn.ReadMsg()
+		if err != nil {
+			log.Debug("read client id %d is error",err)
+			break
+		}
+		data,err:=slf.tcpService.process.Unmarshal(bytes)
+		if err != nil {
+			log.Debug("process.Unmarshal is error:%+v",err)
+			continue
+		}
+
+
+		slf.tcpService.GetEventReciver().NotifyEvent(&event.Event{Type:event.Sys_Event_Tcp_RecvPack,Data:&TcpPack{ClientId:slf.id,Data:data}})
+	}
+}
+
+func (slf *Client) OnClose(){
+	slf.tcpService.GetEventReciver().NotifyEvent(&event.Event{Type:event.Sys_Event_Tcp_DisConnected,Data:&TcpPack{ClientId:slf.id}})
+	slf.tcpService.mapClientLocker.Lock()
+	defer slf.tcpService.mapClientLocker.Unlock()
+	delete (slf.tcpService.mapClient,slf.GetId())
+}