go語言處理TCP拆包/粘包的具體實(shí)現(xiàn)
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)文章希望大家以后多多支持腳本之家!
相關(guān)文章
GoFrame框架garray對(duì)比PHP的array優(yōu)勢(shì)
這篇文章主要為大家介紹了GoFrame框架garray對(duì)比PHP的array優(yōu)勢(shì)詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2022-06-06Go語言中關(guān)于set的實(shí)現(xiàn)思考分析
Go?開發(fā)過程中有時(shí)我們需要集合(set)這種容器,但?Go?本身未內(nèi)置這種數(shù)據(jù)容器,故常常我們需要自己實(shí)現(xiàn),下面我們就來看看具體有哪些實(shí)現(xiàn)方法吧2024-01-01golang實(shí)現(xiàn)簡(jiǎn)單的tcp數(shù)據(jù)傳輸
這篇文章主要為大家介紹了golang實(shí)現(xiàn)簡(jiǎn)單的tcp數(shù)據(jù)傳輸,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2022-12-12Golang結(jié)構(gòu)化日志包log/slog的使用詳解
官方提供的用于打印日志的包是標(biāo)準(zhǔn)庫中的 log 包,該包雖然被廣泛使用,但是缺點(diǎn)也很多,所以Go 1.21新增的 log/slog 完美解決了以上問題,下面我們就來看看log/slog包的使用吧2023-09-09golang對(duì)etcd存取和數(shù)值監(jiān)測(cè)的實(shí)現(xiàn)
這篇文章主要介紹了golang對(duì)etcd存取和數(shù)值監(jiān)測(cè)的實(shí)現(xiàn),文中通過示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2019-09-09