origin/network/tcpsocketserver.go

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)
	}
}