golang网络socket粘包问题的解决方法
程序员文章站
2022-08-29 08:30:32
本文实例讲述了golang网络socket粘包问题的解决方法。分享给大家供大家参考,具体如下:
看到很多人问这个问题, 今天就写了个例子, 希望能帮助大家
首先说一下什...
本文实例讲述了golang网络socket粘包问题的解决方法。分享给大家供大家参考,具体如下:
看到很多人问这个问题, 今天就写了个例子, 希望能帮助大家
首先说一下什么是粘包:百度上比较通俗的说法是指tcp协议中,发送方发送的若干包数据到接收方接收时粘成一包,从接收缓冲区看,后一包数据的头紧接着前一包数据的尾。
解决方案如下:
服务端:
复制代码 代码如下:
package main
import (
"bytes"
"encoding/binary"
"fmt"
"io"
"net"
)
func main() {
// 监听端口
ln, err := net.listen("tcp", ":6000")
if err != nil {
fmt.printf("listen error: %s\n", err)
return
}
// 监听循环
for {
// 接受客户端链接
conn, err := ln.accept()
if err != nil {
fmt.printf("accept error: %s\n", err)
continue
}
// 处理客户端链接
go handleconnection(conn)
}
}
func handleconnection(conn net.conn) {
// 关闭链接
defer conn.close()
// 客户端
fmt.printf("client: %s\n", conn.remoteaddr())
// 消息缓冲
msgbuf := bytes.newbuffer(make([]byte, 0, 10240))
// 数据缓冲
databuf := make([]byte, 4096)
// 消息长度
length := 0
// 消息长度uint32
ulength := uint32(0)
// 数据循环
for {
// 读取数据
n, err := conn.read(databuf)
if err == io.eof {
fmt.printf("client exit: %s\n", conn.remoteaddr())
}
if err != nil {
fmt.printf("read error: %s\n", err)
return
}
fmt.println(databuf[:n])
// 数据添加到消息缓冲
n, err = msgbuf.write(databuf[:n])
if err != nil {
fmt.printf("buffer write error: %s\n", err)
return
}
// 消息分割循环
for {
// 消息头
if length == 0 && msgbuf.len() >= 4 {
binary.read(msgbuf, binary.littleendian, &ulength)
length = int(ulength)
// 检查超长消息
if length > 10240 {
fmt.printf("message too length: %d\n", length)
return
}
}
// 消息体
if length > 0 && msgbuf.len() >= length {
fmt.printf("client messge: %s\n", string(msgbuf.next(length)))
length = 0
} else {
break
}
}
}
}
import (
"bytes"
"encoding/binary"
"fmt"
"io"
"net"
)
func main() {
// 监听端口
ln, err := net.listen("tcp", ":6000")
if err != nil {
fmt.printf("listen error: %s\n", err)
return
}
// 监听循环
for {
// 接受客户端链接
conn, err := ln.accept()
if err != nil {
fmt.printf("accept error: %s\n", err)
continue
}
// 处理客户端链接
go handleconnection(conn)
}
}
func handleconnection(conn net.conn) {
// 关闭链接
defer conn.close()
// 客户端
fmt.printf("client: %s\n", conn.remoteaddr())
// 消息缓冲
msgbuf := bytes.newbuffer(make([]byte, 0, 10240))
// 数据缓冲
databuf := make([]byte, 4096)
// 消息长度
length := 0
// 消息长度uint32
ulength := uint32(0)
// 数据循环
for {
// 读取数据
n, err := conn.read(databuf)
if err == io.eof {
fmt.printf("client exit: %s\n", conn.remoteaddr())
}
if err != nil {
fmt.printf("read error: %s\n", err)
return
}
fmt.println(databuf[:n])
// 数据添加到消息缓冲
n, err = msgbuf.write(databuf[:n])
if err != nil {
fmt.printf("buffer write error: %s\n", err)
return
}
// 消息分割循环
for {
// 消息头
if length == 0 && msgbuf.len() >= 4 {
binary.read(msgbuf, binary.littleendian, &ulength)
length = int(ulength)
// 检查超长消息
if length > 10240 {
fmt.printf("message too length: %d\n", length)
return
}
}
// 消息体
if length > 0 && msgbuf.len() >= length {
fmt.printf("client messge: %s\n", string(msgbuf.next(length)))
length = 0
} else {
break
}
}
}
}
客户端:
复制代码 代码如下:
package main
import (
"bytes"
"encoding/binary"
"fmt"
"net"
"time"
)
func main() {
// 链接服务器
conn, err := net.dial("tcp", "127.0.0.1:6000")
if err != nil {
fmt.printf("dial error: %s\n", err)
return
}
// 客户端信息
fmt.printf("client: %s\n", conn.localaddr())
// 消息缓冲
msgbuf := bytes.newbuffer(make([]byte, 0, 1024))
// 消息内容
message := []byte("我是utf-8的消息")
// 消息长度
messagelen := uint32(len(message))
// 消息总长度
mlen := 4 + len(message)
// 写入5条消息
for i := 0; i < 10; i++ {
binary.write(msgbuf, binary.littleendian, messagelen)
msgbuf.write(message)
}
// 单包发送一条消息
conn.write(msgbuf.next(mlen))
time.sleep(time.second)
// 单包发送三条消息
conn.write(msgbuf.next(mlen * 3))
time.sleep(time.second)
// 发送不完整的消息头
conn.write(msgbuf.next(2))
time.sleep(time.second)
// 发送消息剩下部分
conn.write(msgbuf.next(mlen - 2))
time.sleep(time.second)
// 发送不完整的消息体
conn.write(msgbuf.next(mlen - 6))
time.sleep(time.second)
// 发送消息剩下部分
conn.write(msgbuf.next(6))
time.sleep(time.second)
// 多段发送
conn.write(msgbuf.next(mlen + 2))
time.sleep(time.second)
conn.write(msgbuf.next(-2 + mlen - 8))
time.sleep(time.second)
conn.write(msgbuf.next(8 + 1))
time.sleep(time.second)
conn.write(msgbuf.next(-1 + mlen + mlen))
time.sleep(time.second)
// 关闭链接
conn.close()
}
import (
"bytes"
"encoding/binary"
"fmt"
"net"
"time"
)
func main() {
// 链接服务器
conn, err := net.dial("tcp", "127.0.0.1:6000")
if err != nil {
fmt.printf("dial error: %s\n", err)
return
}
// 客户端信息
fmt.printf("client: %s\n", conn.localaddr())
// 消息缓冲
msgbuf := bytes.newbuffer(make([]byte, 0, 1024))
// 消息内容
message := []byte("我是utf-8的消息")
// 消息长度
messagelen := uint32(len(message))
// 消息总长度
mlen := 4 + len(message)
// 写入5条消息
for i := 0; i < 10; i++ {
binary.write(msgbuf, binary.littleendian, messagelen)
msgbuf.write(message)
}
// 单包发送一条消息
conn.write(msgbuf.next(mlen))
time.sleep(time.second)
// 单包发送三条消息
conn.write(msgbuf.next(mlen * 3))
time.sleep(time.second)
// 发送不完整的消息头
conn.write(msgbuf.next(2))
time.sleep(time.second)
// 发送消息剩下部分
conn.write(msgbuf.next(mlen - 2))
time.sleep(time.second)
// 发送不完整的消息体
conn.write(msgbuf.next(mlen - 6))
time.sleep(time.second)
// 发送消息剩下部分
conn.write(msgbuf.next(6))
time.sleep(time.second)
// 多段发送
conn.write(msgbuf.next(mlen + 2))
time.sleep(time.second)
conn.write(msgbuf.next(-2 + mlen - 8))
time.sleep(time.second)
conn.write(msgbuf.next(8 + 1))
time.sleep(time.second)
conn.write(msgbuf.next(-1 + mlen + mlen))
time.sleep(time.second)
// 关闭链接
conn.close()
}
希望本文所述对大家go语言程序设计有所帮助。
下一篇: golang解析xml的方法