GO语言手动处理TCP粘包

应用场景

大部分TCP通讯场景下，使用自定义通讯协议

image.png
粘包处理原理：通过请求头中数据包大小，将客户端N次发送的数据缓冲到一个数据包中
例如：
请求头占3个字节（指令头1字节、数据包长度2字节），版本占1个字节，指令占2个字节
协议规定一个数据包最大是512字节，请求头中数据包记录是1300字节，完整的数据包是1307个字节，此时服务器端需要将客户端3次发送数据进行粘包处理

代码示例

package server

import (
    "net"
    "bufio"
    "ftj-data-synchro/protocol"
    "golang.org/x/text/transform"
    "golang.org/x/text/encoding/simplifiedchinese"
    "io/ioutil"
    "bytes"
    "ftj-data-synchro/logic"
    "fmt"
    "strconv"
)

/*
 客户端结构体
 */
type Client struct {
    DeviceID string        //客户端连接的唯标志
    Conn     net.Conn      //连接
    reader   *bufio.Reader //读取
    writer   *bufio.Writer //输出
    Data     []byte        //接收数据
}

func NewClient(conn *net.TCPConn) *Client {
    reader := bufio.NewReaderSize(conn, 10240)
    writer := bufio.NewWriter(conn)
    c := &Client{Conn:conn, reader:reader, writer:writer}
    return c

}

/**
    数据读取（粘包处理）
 */
func (this *Client)read() {
    for {
        var data []byte
        var err error
        //读取指令头 返回输入流的前4个字节，不会移动读取位置
        data, err = this.reader.Peek(4)
        if len(data) == 0 || err != nil {
            continue
        }
        //返回缓冲中现有的可读取的字节数
        var byteSize = this.reader.Buffered()
        fmt.Printf("读取字节长度：%d\n", byteSize)
        //生成一个字节数组，大小为缓冲中可读字节数
        data = make([]byte, byteSize)
        //读取缓冲中的数据
        this.reader.Read(data)
        fmt.Printf("读取字节：%d\n", data)
        //保存到新的缓冲区
        for _, v := range data {
            this.Data = append(this.Data, v)
        }

        if len(this.Data) < 4 {
            //数据包缓冲区清空
            this.Data = []byte{}
            fmt.Printf("非法数据，无指令头...\n")
            continue
        }

        data, err = protocol.HexBytesToBytes(this.Data[:4])
        instructHead, _ := strconv.ParseUint(string(data), 16, 16)
        //指令头效验
        if uint16(instructHead) != 42330 {
            fmt.Printf("非法数据\n")
            //数据包缓冲区清空
            this.Data = []byte{}
            continue
        }

        data = this.Data[:protocol.HEADER_SIZE]

        var p = protocol.Decode(data)
        fmt.Printf("消息体长度：%d\n", p.Len)

        var bodyLength = len(this.Data)
        
      /**
            判断数据包缓冲区的大小是否小于协议请求头中数据包大小
            如果小于，等待读取下一个客户端数据包，否则对数据包解码进行业务逻辑处理
         */
        if int(p.Len) > len(this.Data) - protocol.HEADER_SIZE {
            fmt.Printf("body体长度：%d,读取的body体长度：%d\n", p.Len, bodyLength)
            continue
        }
        fmt.Printf("实际处理字节：%v\n", this.Data)
        p = protocol.Decode(this.Data)
        //逻辑处理
        go this.logicHandler(p)
        //数据包缓冲区清空
        this.Data = []byte{}
    }
}

待优化部分：

type Client struct {
    DeviceID string        //客户端连接的唯标志
    Conn     net.Conn      //连接
    reader   *bufio.Reader //读取
    writer   *bufio.Writer //输出
    Data     []byte        //接收数据
}

结构体中Data属性可考虑使用bytes.Buffer实现。

Golang标准库文档：https://studygolang.com/pkgdoc

以上属于原创文章，转载请注明作者@怪咖
QQ:208275451