go語言處理TCP拆包/粘包的具體實現(xiàn)
part 1
最近在學習go自帶的rpc,看完了一遍想著自己實現(xiàn)一個codec,也就是自定義消息的序列化和反序列化。消息的序列化和反序列化涉及到兩步:
1、從網絡中讀取數據和將數據寫到網絡中;
2、根據拿到的二進制數據反序列化以及把現(xiàn)有的對象序列化成二進制數據。而這個過程中就需要處理TCP的拆包粘包了。
TCP的拆包/粘包也算是網絡編程中一個比較基礎的問題了,具體的問題含義和解決方式也不再詳細描述了。雖然作為實現(xiàn)應用層邏輯的程序員可能根本不需要關心這個,作為中間件的開發(fā),同時也出于學習go語言的目的,還是稍微進行一下練習。
part 2
TCP拆包粘包的解決思路:在讀取數據時,將讀取到的二進制數據在正確的位置進行分割即可。這里直接使用head+body的方式,也就是在發(fā)送數據時先將整個數據的大小附加在數據前,就像這樣:
+++++++++++++++++++++++++++++++++++++
size (2 bytes)? | body (size bytes)
+++++++++++++++++++++++++++++++++++++
這里將數據包的size作為head
注:這里舉的例子里,size占2個byte;size具體占用的byte數可以根據實際情況決定
part 3
server具體實現(xiàn):
func doConn(conn net.Conn) {
var (
buffer = bytes.NewBuffer(make([]byte, 0, BUF_SIZE)) //buffer用來緩存讀取到的數據
readBytes = make([]byte, BUF_SIZE) //readBytes用來接收每次讀取的數據,每次讀取完成之后將readBytes添加到buffer中
isHead = true //用來標識當前的狀態(tài):正在處理size部分還是body部分
bodyLen = 0 //表示body的長度
)
for {
//首先讀取數據
readByteNum, err := conn.Read(readBytes)
if err != nil {
log.Fatal(err)
return
}
buffer.Write(readBytes[0:readByteNum])//將讀取到的數據放到buffer中
// 然后處理數據
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具體實現(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
}
到此這篇關于go語言處理TCP拆包/粘包的文章就介紹到這了,更多相關go語言 TCP拆包/粘包內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家!
相關文章
GoFrame框架garray對比PHP的array優(yōu)勢
這篇文章主要為大家介紹了GoFrame框架garray對比PHP的array優(yōu)勢詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪2022-06-06
golang對etcd存取和數值監(jiān)測的實現(xiàn)
這篇文章主要介紹了golang對etcd存取和數值監(jiān)測的實現(xiàn),文中通過示例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友們下面隨著小編來一起學習學習吧2019-09-09

