新增tcpsocket协议

network/tcpsocketserver.go

@@ -0,0 +1,203 @@
+package network
+
+import (
+	"bufio"
+	"encoding/binary"
+	"github.com/duanhf2012/origin/util"
+	"github.com/duanhf2012/origin/service"
+	"io"
+	"net"
+
+	"os"
+	"time"
+)
+
+type ITcpSocketServerReciver interface {
+	OnConnected(pClient *SClient)
+	OnDisconnect(pClient *SClient)
+	OnRecvMsg(pClient *SClient, pPack *MsgBasePack)
+}
+
+
+type SClient struct {
+	id uint64
+	conn net.Conn
+
+	recvPack util.SyncQueue
+	sendPack util.SyncQueue
+	tcpserver *TcpSocketServer
+	remoteip string
+	starttime int64
+}
+
+type TcpSocketServer struct {
+	listenAddr string //ip:port
+	mapClient util.Map
+
+	MaxRecvPackSize uint16
+	MaxSendPackSize uint16
+	iReciver ITcpSocketServerReciver
+}
+
+type MsgBasePack struct {
+	packsize uint16
+	packtype uint16
+	body []byte
+	StartTime time.Time
+}
+
+func (slf *TcpSocketServer) Register(listenAddr string,iReciver ITcpSocketServerReciver){
+	slf.listenAddr = listenAddr
+	slf.iReciver = iReciver
+}
+
+
+func (slf *TcpSocketServer) Start(){
+	slf.MaxRecvPackSize = 2048
+	slf.MaxSendPackSize = 40960
+
+	util.Go(slf.listenServer)
+}
+
+func (slf *TcpSocketServer) listenServer(){
+	listener, err := net.Listen("tcp", slf.listenAddr)
+	if err != nil {
+		service.GetLogger().Printf(service.LEVER_FATAL, "TcpSocketServer Listen error %+v",err)
+		os.Exit(1)
+	}
+
+	var clientId uint64
+	for {
+		conn, aerr := listener.Accept()
+		if aerr != nil {
+			service.GetLogger().Printf(service.LEVER_FATAL, "TcpSocketServer accept error %+v",aerr)
+			continue
+		}
+
+		for {
+			clientId += 1
+			if slf.mapClient.Get(clientId)!= nil {
+				continue
+			}
+
+			sc :=&SClient{id:clientId,conn:conn,tcpserver:slf,remoteip:conn.RemoteAddr().String(),starttime:time.Now().UnixNano()}
+			slf.iReciver.OnConnected(sc)
+			util.Go(sc.listendata)
+			//收来自客户端数据
+			util.Go(sc.onrecv)
+			//发送数据队列
+			util.Go(sc.onsend)
+			slf.mapClient.Set(clientId,sc)
+
+			break
+		}
+	}
+}
+
+
+func (slf *SClient) listendata(){
+	defer func() {
+		slf.tcpserver.iReciver.OnDisconnect(slf)
+		slf.conn.Close()
+		slf.tcpserver.mapClient.Del(slf.id)
+	}()
+
+	//获取一个连接的reader读取流
+	reader := bufio.NewReader(slf.conn)
+
+	//临时接受数据的buff
+	var buff []byte //tmprecvbuf
+	var tmpbuff []byte
+	var buffDataSize uint16
+	tmpbuff = make([]byte,2048)
+
+	//解析包数据
+	var pack MsgBasePack
+	for {
+		n,err := reader.Read(tmpbuff)
+		if err != nil || err == io.EOF {
+			service.GetLogger().Printf(service.LEVER_INFO, "clent id %d is disconnect  %+v",slf.id,err)
+			return
+		}
+		buff = append(buff,tmpbuff[:n]...)
+		buffDataSize += uint16(n)
+		if buffDataSize> slf.tcpserver.MaxRecvPackSize {
+			service.GetLogger().Print(service.LEVER_WARN,"recv client id %d data size %d is over %d",slf.id,buffDataSize,slf.tcpserver.MaxRecvPackSize)
+			return
+		}
+
+		fillsize,bfillRet := pack.FillData(buff,buffDataSize)
+		if bfillRet == true {
+			slf.recvPack.Push(pack)
+			pack = MsgBasePack{}
+		}
+		buff = append(buff[fillsize:])
+		buffDataSize -= fillsize
+	}
+}
+
+
+func (slf *MsgBasePack) Bytes() (bRet []byte){
+	binary.BigEndian.PutUint16(bRet,slf.packsize)
+	binary.BigEndian.PutUint16(bRet,slf.packtype)
+	bRet = append(bRet,slf.body...)
+
+	return
+}
+
+//返回值：填充多少字节，是否完成
+func (slf *MsgBasePack) FillData(bdata []byte,datasize uint16) (uint16,bool) {
+	var fillsize uint16
+	//解包头
+	if slf.packsize ==0 {
+		if datasize < 4 {
+			return 0,false
+		}
+
+		slf.packsize= binary.BigEndian.Uint16(bdata[:2])
+		slf.packtype= binary.BigEndian.Uint16(bdata[2:2])
+		fillsize += 4
+	}
+
+	//解包体
+	if slf.packsize>0 && slf.packsize>=datasize {
+		slf.body = append(slf.body, bdata[fillsize:slf.packsize]...)
+		fillsize += (datasize - fillsize)
+		return fillsize,true
+	}
+
+	return fillsize,false
+}
+
+func (slf *MsgBasePack) Clear() {
+}
+
+func (slf *SClient) Send(pack *MsgBasePack){
+	slf.sendPack.Push(pack)
+}
+
+func (slf *SClient) onsend(){
+	for {
+		pack := slf.sendPack.Pop()
+		if pack == nil {
+			continue
+		}
+		pPackData := pack.(*MsgBasePack)
+		slf.conn.Write(pPackData.Bytes())
+	}
+}
+
+func (slf *SClient) onrecv(){
+	for {
+		pack := slf.recvPack.Pop()
+		if pack == nil {
+			continue
+		}
+
+		pMsg := pack.(MsgBasePack)
+		slf.tcpserver.iReciver.OnRecvMsg(slf,&pMsg)
+	}
+}
+
+
+