go語言處理TCP拆包/粘包的具體實(shí)現(xiàn)

更新時(shí)間：2021年12月09日 15:43:09 作者：熊紀(jì)元

TCP的拆包/粘包也算是網(wǎng)絡(luò)編程中一個(gè)比較基礎(chǔ)的問題了，本文主要介紹了go語言處理TCP拆包/粘包，文中通過示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下

part 1

最近在學(xué)習(xí)go自帶的rpc，看完了一遍想著自己實(shí)現(xiàn)一個(gè)codec，也就是自定義消息的序列化和反序列化。消息的序列化和反序列化涉及到兩步：

1、從網(wǎng)絡(luò)中讀取數(shù)據(jù)和將數(shù)據(jù)寫到網(wǎng)絡(luò)中；

2、根據(jù)拿到的二進(jìn)制數(shù)據(jù)反序列化以及把現(xiàn)有的對(duì)象序列化成二進(jìn)制數(shù)據(jù)。而這個(gè)過程中就需要處理TCP的拆包粘包了。

TCP的拆包/粘包也算是網(wǎng)絡(luò)編程中一個(gè)比較基礎(chǔ)的問題了，具體的問題含義和解決方式也不再詳細(xì)描述了。雖然作為實(shí)現(xiàn)應(yīng)用層邏輯的程序員可能根本不需要關(guān)心這個(gè)，作為中間件的開發(fā)，同時(shí)也出于學(xué)習(xí)go語言的目的，還是稍微進(jìn)行一下練習(xí)。

part 2

TCP拆包粘包的解決思路：在讀取數(shù)據(jù)時(shí)，將讀取到的二進(jìn)制數(shù)據(jù)在正確的位置進(jìn)行分割即可。這里直接使用head+body的方式，也就是在發(fā)送數(shù)據(jù)時(shí)先將整個(gè)數(shù)據(jù)的大小附加在數(shù)據(jù)前，就像這樣：

+++++++++++++++++++++++++++++++++++++
size (2 bytes)? | body (size bytes)
+++++++++++++++++++++++++++++++++++++

這里將數(shù)據(jù)包的size作為head
注：這里舉的例子里，size占2個(gè)byte；size具體占用的byte數(shù)可以根據(jù)實(shí)際情況決定

part 3

server具體實(shí)現(xiàn)：

func doConn(conn net.Conn) {
    var (
        buffer = bytes.NewBuffer(make([]byte, 0, BUF_SIZE)) //buffer用來緩存讀取到的數(shù)據(jù)
        readBytes = make([]byte, BUF_SIZE) //readBytes用來接收每次讀取的數(shù)據(jù)，每次讀取完成之后將readBytes添加到buffer中
        isHead = true //用來標(biāo)識(shí)當(dāng)前的狀態(tài)：正在處理size部分還是body部分
        bodyLen = 0 //表示body的長(zhǎng)度
    )

    for {
        //首先讀取數(shù)據(jù)
        readByteNum, err := conn.Read(readBytes)
        if err != nil {
            log.Fatal(err)
            return
        }
        buffer.Write(readBytes[0:readByteNum])//將讀取到的數(shù)據(jù)放到buffer中
        
        // 然后處理數(shù)據(jù)
        for {
            if isHead {
                if buffer.Len() >= HEAD_SIZE {
                    isHead = false
                    head := make([]byte, HEAD_SIZE)
                    _, err = buffer.Read(head)
                    if err != nil {
                        log.Fatal(err)
                        return
                    }
                    bodyLen = int(binary.BigEndian.Uint16(head))
                } else {
                    break;
                }
            }

            if !isHead {
                if buffer.Len() >= bodyLen {
                    body := make([]byte, bodyLen)
                    _, err = buffer.Read(body[:bodyLen])
                    if err != nil {
                        log.Fatal(err)
                        return
                    }
                    fmt.Println("received body: " + string(body[:bodyLen]))
                    isHead = true
                } else {
                    break;
                }
            }
        }
    }
    
    func HandleTcp() {
 listener, err := net.Listen("tcp", ":1234")
 if err != nil {
  log.Fatal(err)
  return
 }
 log.Println("start listening on 1234")
 for {
  conn, err := listener.Accept()
  if err != nil {
   log.Fatal(err)
   return
  }
  go doConn(conn)
 }
}

client具體實(shí)現(xiàn)：

func SendStringwithTcp(arg string) error {
 conn, err := net.Dial("tcp", ":1234")
 if err != nil {
  log.Fatal(err)
  return err
 }

 head := make([]byte, server.HEAD_SIZE)
 content := []byte(arg)
 headSize := len(content)
 binary.BigEndian.PutUint16(head, uint16(headSize))

    //先寫入head部分，再寫入body部分
 _, err = conn.Write(head)
 if err != nil {
  log.Fatal(err)
  return err
 }
 _, err = conn.Write(content)
 if err != nil {
  log.Fatal(err)
  return err
 }
 return nil
}

到此這篇關(guān)于go語言處理TCP拆包/粘包的文章就介紹到這了,更多相關(guān)go語言 TCP拆包/粘包內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: