Golang实现简单tcp服务器04 -- 服务器的粘包处理

前端之家收集整理的这篇文章主要介绍了Golang实现简单tcp服务器04 -- 服务器的粘包处理前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

服务器的粘包处理

什么是粘包

一个完成的消息可能会被TCP拆分成多个包进行发送,也有可能把多个小的包封装成一个大的数据包发送,这个就是TCP的拆包和封包问题

TCP粘包和拆包产生的原因

  1. 应用程序写入数据的字节大小大于套接字发送缓冲区的大小

  2. 进行MSS大小的TCP分段。MSS是最大报文段长度的缩写。MSS是TCP报文段中的数据字段的最大长度。数据字段加上TCP首部才等于整个的TCP报文段。所以MSS并不是TCP报文段的最大长度,而是:MSS=TCP报文段长度-TCP首部长度

  3. 以太网的payload大于MTU进行IP分片。MTU指:一种通信协议的某一层上面所能通过的最大数据包大小。如果IP层有一个数据包要传,而且数据的长度比链路层的MTU大,那么IP层就会进行分片,把数据包分成托干片,让每一片都不超过MTU。注意,IP分片可以发生在原始发送端主机上,也可以发生在中间路由器上。

TCP粘包和拆包的解决策略

  1. 消息定长。例如100字节。
  2. 在包尾部增加回车或者空格符等特殊字符进行分割,典型的如FTP协议
  3. 将消息分为消息头和消息尾。
  4. 其它复杂的协议,如RTMP协议等。

参考(http://www.jb51.cc/article/p-eokjffzs-g.html)

我们的处理方式

解决粘包问题有多种多样的方式,我们这里的做法是:

  • 发送方在每次发送消息时将消息长度写入一个int32作为包头一并发送出去,我们称之为Encode
  • 接受方则先读取一个int32的长度的消息长度信息,再根据长度读取相应长的byte数据,称之为Decode

在实验环境中的主文件夹内,新建一个名为codec的文件夹在其之下新建一个文件codec.go,将我们的Encode和Decode方法写入其中,这里给出Encode与Decode相应的代码:

codec.go

package codec

import (
	"bufio"
	"bytes"
	"encoding/binary"
)

func Encode(message string) ([]byte,error) {
	// 读取消息的长度
	var length int32 = int32(len(message))
	var pkg *bytes.Buffer = new(bytes.Buffer)
	// 写入消息头
	err := binary.Write(pkg,binary.LittleEndian,length)
	if err != nil {
		return nil,err
	}
	// 写入消息实体
	err = binary.Write(pkg,[]byte(message))
	if err != nil {
		return nil,err
	}

	return pkg.Bytes(),nil
}

func Decode(reader *bufio.Reader) (string,error) {
	// 读取消息的长度
	lengthByte,_ := reader.Peek(4)
	lengthBuff := bytes.NewBuffer(lengthByte)
	var length int32
	err := binary.Read(lengthBuff,&length)
	if err != nil {
		return "",err
	}
	if int32(reader.Buffered()) < length+4 {
		return "",err
	}
	
	// 读取消息真正的内容
	pack := make([]byte,int(4+length))
	_,err = reader.Read(pack)
	if err != nil {
		return "",err
	}
	return string(pack[4:]),nil
}

这里就不帖服务器与客户端的调用代码了,同学们自己动手试试~

相关源码: https://git.oschina.net/victoriest/go-simple-tcp-server.git

原文链接:https://www.f2er.com/go/190381.html

猜你在找的Go相关文章