dlt645-server/server/client.go

package server

import (
	"github.com/gogf/gf/net/gtcp"
	"github.com/gogf/gf/os/glog"
	"io"
	"net"
	"strings"
	"syscall"
	"time"
)

type Client struct {
	Id           string
	srv          *Server
	conn         *gtcp.Conn
	sendChan     chan []byte
	closeChan    chan struct{}
	closeHandler func(id string, c *Client)
	nc            *gtcp.Conn
	lastHeartBeat time.Time
	done          chan struct{}
	gatewayId     uint16

}


func (c *Client) ReadLoop() {
	defer c.srv.grWG.Done()
	for {
		buf, err := c.nc.RecvTil([]byte{0x16})
		if err != nil {
			c.readError(err)
			return
		}
		if len(buf) > 0 {
			err = c.srv.message.Decode(buf)
			if err != nil {
				glog.Errorf("解析报文失败:%s", err.Error())
			}
		}
	}
}


func (c *Client) SetId(id string) {
	c.Id = id
}


func (c *Client) readError(err error) {
	defer c.closeConnection()
	if err == io.EOF || isErrConnReset(err) {
		return
	}
	glog.Errorf("读取数据发生错误：%s", err.Error())
}

func (c *Client) closeConnection() {
	c.nc.Close()
	c.nc = nil
	close(c.done)
	if c.closeHandler != nil {
		c.closeHandler(c.Id, c)
	}
}

// isErrConnReset read: connection reset by peer
func isErrConnReset(err error) bool {
	if ne, ok := err.(*net.OpError); ok {
		return strings.Contains(ne.Err.Error(), syscall.ECONNRESET.Error())
	}
	return false
}

func (c *Client) send(buf []byte) error {
	if c.nc == nil {
		return nil
	}
	err := c.nc.Send(buf)
	if err != nil {
		glog.Error(err)
		c.closeConnection()
		return err
	}
	return nil
}