GO語言手動處理TCP粘包

怪咖_OOP發表於2017-12-11
GoTCP

應用場景

大部分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

相關文章