前言
一般所謂的TCP粘包是在一次接收數據不能完全地體現一個完整的消息數據。TCP通訊為何存在粘包呢？主要原因是TCP是以流的方式來處理數據，再加上網絡上MTU的往往小于在應用處理的消息數據，所以就會引發一次接收的數據無法滿足消息的需要，導致粘包的存在。處理粘包的唯一方法就是制定應用層的數據通訊協議，通過協議來規范現有接收的數據是否滿足消息數據的需要。在應用中處理粘包的基礎方法主要有兩種分別是以4節字描述消息大小或以結束符，實際上也有兩者相結合的如HTTP,redis的通訊協議等。
應用場景
大部分TCP通訊場景下，使用自定義通訊協議
粘包處理原理：通過請求頭中數據包大小，將客戶端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
總結
以上就是這篇文章的全部內容了，希望本文的內容對大家的學習或者工作具有一定的參考學習價值，如果有疑問大家可以留言交流，謝謝大家對腳本之家的支持。
您可能感興趣的文章:- Go語言服務器開發之簡易TCP客戶端與服務端實現方法
- GO語言實現簡單TCP服務的方法
- golang之tcp自動重連實現方法
- 利用Golang實現TCP連接的雙向拷貝詳解
- 6行代碼快速解決golang TCP粘包問題
- Go語言中TCP/IP網絡編程的深入講解
