经验首页 前端设计 程序设计 Java相关 移动开发 数据库/运维 软件/图像 大数据/云计算 其他经验
当前位置:技术经验 » 程序设计 » Go语言 » 查看文章
go语言处理TCP拆包/粘包的具体实现
来源:jb51  时间:2021/12/9 18:02:42  对本文有异议

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具体实现:

  1. func doConn(conn net.Conn) {
  2. var (
  3. buffer = bytes.NewBuffer(make([]byte, 0, BUF_SIZE)) //buffer用来缓存读取到的数据
  4. readBytes = make([]byte, BUF_SIZE) //readBytes用来接收每次读取的数据,每次读取完成之后将readBytes添加到buffer中
  5. isHead = true //用来标识当前的状态:正在处理size部分还是body部分
  6. bodyLen = 0 //表示body的长度
  7. )
  8.  
  9. for {
  10. //首先读取数据
  11. readByteNum, err := conn.Read(readBytes)
  12. if err != nil {
  13. log.Fatal(err)
  14. return
  15. }
  16. buffer.Write(readBytes[0:readByteNum])//将读取到的数据放到buffer中
  17. // 然后处理数据
  18. for {
  19. if isHead {
  20. if buffer.Len() >= HEAD_SIZE {
  21. isHead = false
  22. head := make([]byte, HEAD_SIZE)
  23. _, err = buffer.Read(head)
  24. if err != nil {
  25. log.Fatal(err)
  26. return
  27. }
  28. bodyLen = int(binary.BigEndian.Uint16(head))
  29. } else {
  30. break;
  31. }
  32. }
  33.  
  34. if !isHead {
  35. if buffer.Len() >= bodyLen {
  36. body := make([]byte, bodyLen)
  37. _, err = buffer.Read(body[:bodyLen])
  38. if err != nil {
  39. log.Fatal(err)
  40. return
  41. }
  42. fmt.Println("received body: " + string(body[:bodyLen]))
  43. isHead = true
  44. } else {
  45. break;
  46. }
  47. }
  48. }
  49. }
  50. func HandleTcp() {
  51. listener, err := net.Listen("tcp", ":1234")
  52. if err != nil {
  53. log.Fatal(err)
  54. return
  55. }
  56. log.Println("start listening on 1234")
  57. for {
  58. conn, err := listener.Accept()
  59. if err != nil {
  60. log.Fatal(err)
  61. return
  62. }
  63. go doConn(conn)
  64. }
  65. }

client具体实现:

  1. func SendStringwithTcp(arg string) error {
  2. conn, err := net.Dial("tcp", ":1234")
  3. if err != nil {
  4. log.Fatal(err)
  5. return err
  6. }
  7.  
  8. head := make([]byte, server.HEAD_SIZE)
  9. content := []byte(arg)
  10. headSize := len(content)
  11. binary.BigEndian.PutUint16(head, uint16(headSize))
  12.  
  13. //先写入head部分,再写入body部分
  14. _, err = conn.Write(head)
  15. if err != nil {
  16. log.Fatal(err)
  17. return err
  18. }
  19. _, err = conn.Write(content)
  20. if err != nil {
  21. log.Fatal(err)
  22. return err
  23. }
  24. return nil
  25. }

到此这篇关于go语言处理TCP拆包/粘包的文章就介绍到这了,更多相关go语言 TCP拆包/粘包内容请搜索w3xue以前的文章或继续浏览下面的相关文章希望大家以后多多支持w3xue!

 友情链接:直通硅谷  点职佳  北美留学生论坛

本站QQ群:前端 618073944 | Java 606181507 | Python 626812652 | C/C++ 612253063 | 微信 634508462 | 苹果 692586424 | C#/.net 182808419 | PHP 305140648 | 运维 608723728

W3xue 的所有内容仅供测试,对任何法律问题及风险不承担任何责任。通过使用本站内容随之而来的风险与本站无关。
关于我们  |  意见建议  |  捐助我们  |  报错有奖  |  广告合作、友情链接(目前9元/月)请联系QQ:27243702 沸活量
皖ICP备17017327号-2 皖公网安备34020702000426号