GO語言怎么手動處理TCP粘包

小編給大家分享一下GO語言怎么手動處理TCP粘包，相信大部分人都還不怎么了解，因此分享這篇文章給大家參考一下，希望大家閱讀完這篇文章后大有收獲，下面讓我們一起去了解一下吧！

前言

一般所謂的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實現。

以上是“GO語言怎么手動處理TCP粘包”這篇文章的所有內容，感謝各位的閱讀！相信大家都有了一定的了解，希望分享的內容對大家有所幫助，如果還想學習更多知識，歡迎關注億速云行業資訊頻道！