go语言处理TCP拆包/粘包的具体实现

 更新时间:2022-01-27 14:15:57   作者:佚名   我要评论(0)

目录part 1part 2part 3part 1


最近在学习go自带的rpc,看完了一遍想着自己实现一个codec,也就是自定义消息的序列化和反序列化。消息的序

part 1

最近在学习go自带的rpc,看完了一遍想着自己实现一个codec,也就是自定义消息的序列化和反序列化。消息的序列化和反序列化涉及到两步:

1、从网络中读取数据和将数据写到网络中;

2、根据拿到的二进制数据反序列化以及把现有的对象序列化成二进制数据。而这个过程中就需要处理TCP的拆包粘包了。

TCP的拆包/粘包也算是网络编程中一个比较基础的问题了,具体的问题含义和解决方式也不再详细描述了。虽然作为实现应用层逻辑的程序员可能根本不需要关心这个,作为中间件的开发,同时也出于学习go语言的目的,还是稍微进行一下练习。

part 2

TCP拆包粘包的解决思路:在读取数据时,将读取到的二进制数据在正确的位置进行分割即可。这里直接使用head+body的方式,也就是在发送数据时先将整个数据的大小附加在数据前,就像这样:

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

这里将数据包的size作为head
注:这里举的例子里,size占2个byte;size具体占用的byte数可以根据实际情况决定

part 3

server具体实现:

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 //用来标识当前的状态:正在处理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具体实现:

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拆包/粘包内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

您可能感兴趣的文章:
  • GO语言如何手动处理TCP粘包详解

相关文章

  • go语言处理TCP拆包/粘包的具体实现

    go语言处理TCP拆包/粘包的具体实现

    目录part 1part 2part 3part 1 最近在学习go自带的rpc,看完了一遍想着自己实现一个codec,也就是自定义消息的序列化和反序列化。消息的序
    2022-01-27
  • 使用gin框架搭建简易服务的实现方法

    使用gin框架搭建简易服务的实现方法

    go语言web框架挺多的,各有各的特点和风格。我之所以在项目中使用gin框架,是因为项目一开始是用的martini,一个设计得很好的框架,但是存在
    2022-01-27
  • Gin 框架快速创建静态文件下载Web服务

    Gin 框架快速创建静态文件下载Web服务

    目录介绍安装快速开始1.创建 boot.yaml 2.创建 main.go 3.文件夹结构 4.验证 从 pkger 读取文件 (嵌入式静态文件)1.下载 pkger 命令行
    2022-01-27
  • golang配制高性能sql.DB的使用

    golang配制高性能sql.DB的使用

    目录开放和空闲连接SetMaxOpenConns方法SetMaxIdleConns方法SetConnMaxLifetime方法连接数量超出总结有很多教程是关于Go的sql.DB类型和如何使
    2022-01-27
  • golang构建HTTP服务的实现步骤

    golang构建HTTP服务的实现步骤

    目录HTTPHandlerServeMuxServer创建HTTP服务http注册路由开启监听处理请求参考:???一个go最简单的Http服务器程序 package main import
    2022-01-27
  • golang?db事务的统一封装的实现

    golang?db事务的统一封装的实现

    目录简单封装偷懒第一步 偷懒第二步 偷懒第三步 进一步封装事务处理的流程示例 database := db.DB tx, err := database.Beg
    2022-01-27
  • 浅谈Gin框架中bind的使用

    浅谈Gin框架中bind的使用

    目录概述Binding接口context.Bindcnotext.MustBindWithShouldBindWithcontext.BindJSONcontext.ShouldBindJSONcontext.ShouldBindUri()conte
    2022-01-27
  • Go并发编程中sync/errGroup的使用

    Go并发编程中sync/errGroup的使用

    目录一.序二.errGroup2.1 函数签名 三.源码3.1 Group 3.2 WaitContext 3.3 Go 3.4 Wait 四. 案例五. 参考一.序 这一篇算是并发编程的
    2022-01-27
  • Go语言原子操作及互斥锁的区别

    Go语言原子操作及互斥锁的区别

    目录增或减比较并交换(Compare And Swap)载入与存储交换原子值原子操作与互斥锁的区别原子操作就是不可中断的操作,外界是看不到原子操作的
    2022-01-27
  • Go语言并发爬虫的具体实现

    Go语言并发爬虫的具体实现

    目录写在前面1. 单线程爬虫2. 多线程爬虫2.1 channel main函数2.2 sync.WaitGroup3. 源码地址写在前面 这篇文章主要让大家明白多线程爬虫,
    2022-01-27

最新评论