⼀、使用Golang创建一个TCP连接

1.tcp服务端处理理流程

  a.监听端口
  b.接受客户端的链接

  c.创建Goroutine,处理这个链接

  1. package main
  2. import (
  3. "fmt"
  4. "net"
  5. )
  6. func main() {
  7. //1.建立监听端口
  8. listen, err := net.Listen("tcp", "0.0.0.0:20000")
  9. if err != nil {
  10. fmt.Println("listen failed, err:", err)
  11. return
  12. }
  13. fmt.Println("listen Start...:")
  14. for {
  15. //2.接收客户端的链接
  16. conn, err := listen.Accept()
  17. if err != nil {
  18. fmt.Printf("accept failed, err:%v\n", err)
  19. continue
  20. }
  21. //3.开启一个Goroutine,处理链接
  22. go process(conn)
  23. }
  24. }
  25. //处理请求,类型就是net.Conn
  26. func process(conn net.Conn) {
  27. //处理结束后关闭链接
  28. defer conn.Close()
  29. for {
  30. var buf [128]byte
  31. n, err := conn.Read(buf[:])
  32. if err != nil {
  33. fmt.Printf("read from conn failed, err:%v", err)
  34. break
  35. }
  36. fmt.Printf("recv from client, content:%v\n", string(buf[:n]))
  37. }
  38. }

2.tcp客户端处理流程

a.和服务端建立一个链接
b.进行数据的收发
c.关闭链接

  1. package main
  2. import (
  3. "bufio"
  4. "fmt"
  5. "net"
  6. "os"
  7. "strings"
  8. )
  9. func main() {
  10. //1.建立一个链接(Dial拨号)
  11. conn, err := net.Dial("tcp", "0.0.0.0:20000")
  12. if err != nil {
  13. fmt.Printf("dial failed, err:%v\n", err)
  14. return
  15. }
  16. fmt.Println("Conn Established...:")
  17. //读入输入的信息
  18. reader := bufio.NewReader(os.Stdin)
  19. for {
  20. data, err := reader.ReadString('\n')
  21. if err != nil {
  22. fmt.Printf("read from console failed, err:%v\n", err)
  23. break
  24. }
  25. data = strings.TrimSpace(data)
  26. //传输数据到服务端
  27. _, err = conn.Write([]byte(data))
  28. if err != nil {
  29. fmt.Printf("write failed, err:%v\n", err)
  30. break
  31. }
  32. }
  33. }

二、使用Golang创建一个UDP连接

UDP协议(User Datagram Protocol)中文名称是用户数据报协议,是OSI(Open System Interconnection,开放式系统互联)参考模型中一种无连接的传输层协议,不需要建立连接就能直接进行数据发送和接收,属于不可靠的、没有时序的通信,但是UDP协议的实时性比较好,通常用于视频直播相关领域。

1.udp服务端处理理流程

  1. package main
  2. import (
  3. "fmt"
  4. "net"
  5. )
  6. func main() {
  7. listen,err := net.ListenUDP("udp",&net.UDPAddr{
  8. IP: net.ParseIP("127.0.0.1"),
  9. Port: 9000,
  10. })
  11. if err != nil {
  12. fmt.Println("conn failed",err)
  13. return
  14. }
  15. for {
  16. //接受数据
  17. var buf []byte
  18. n,addr,err := listen.ReadFromUDP(buf[:])
  19. if err != nil {
  20. fmt.Printf("接受数据失败,err is %v",err)
  21. return
  22. }
  23. fmt.Println("接受的数据为%v:%v\n",addr,string(buf[:n]))
  24. //回复数据
  25. nn,err = listen.WriteToUDP([]byte("好的"),addr)
  26. if err != nil {
  27. fmt.Printf("回复数据失败,err is %v",err)
  28. return
  29. }
  30. fmt.
  31. }
  32. }

2.udp客户端处理流程

  1. package main
  2. import (
  3. "fmt"
  4. "net"
  5. )
  6. func main() {
  7. conn,err := net.Dial("udp","127.0.0.1:9000")
  8. if err != nil{
  9. fmt.Println("连接失败",err)
  10. return
  11. }
  12. fmt.Println("连接成功")
  13. //发送数据
  14. n,err :=conn.Write([]byte("约吗"))
  15. if err != nil {
  16. fmt.Println("发送数据失败",err)
  17. return
  18. }
  19. //接受数据
  20. var buf []byte
  21. n,err := conn.Read(buf[:])
  22. if err != nil {
  23. fmt.Println("接受数据失败",err)
  24. return
  25. }
  26. fmt.Println("接受的数据为%v",string(buf[:n]))
  27. }

三、TCP粘包

主要原因就是tcp数据传递模式是流模式,在保持长连接的时候可以进行多次的收和发。
“粘包”可发生在发送端也可发生在接收端:

  1. 由Nagle算法造成的发送端的粘包:Nagle算法是一种改善网络传输效率的算法。简单来说就是当我们提交一段数据给TCP发送时,TCP并不立刻发送此段数据,而是等待一小段时间看看在等待期间是否还有要发送的数据,若有则会一次把这两段数据发送出去。
  2. 接收端接收不及时造成的接收端粘包:TCP会把接收到的数据存在自己的缓冲区中,然后通知应用层取数据。当应用层由于某些原因不能及时的把TCP的数据取出来,就会造成TCP缓冲区中存放了几段数据。

    解决办法

    出现”粘包”的关键在于接收方不确定将要传输的数据包的大小,因此我们可以对数据包进行封包和拆包的操作。
    封包:封包就是给一段数据加上包头,这样一来数据包就分为包头和包体两部分内容了(过滤非法包时封包会加入”包尾”内容)。包头部分的长度是固定的,并且它存储了包体的长度,根据包头长度固定以及包头中含有包体长度的变量就能正确的拆分出一个完整的数据包。
    我们可以自己定义一个协议,比如数据包的前4个字节为包头,里面存储的是发送的数据的长度。 ```go // socket_stick/proto/proto.go package proto

import ( “bufio” “bytes” “encoding/binary” )

// Encode 将消息编码 func Encode(message string) ([]byte, error) { // 读取消息的长度,转换成int32类型(占4个字节) var length = int32(len(message)) var pkg = new(bytes.Buffer) // 写入消息头 err := binary.Write(pkg, binary.LittleEndian, length) if err != nil { return nil, err } // 写入消息实体 err = binary.Write(pkg, binary.LittleEndian, []byte(message)) if err != nil { return nil, err } return pkg.Bytes(), nil }

// Decode 解码消息 func Decode(reader *bufio.Reader) (string, error) { // 读取消息的长度 lengthByte, _ := reader.Peek(4) // 读取前4个字节的数据 lengthBuff := bytes.NewBuffer(lengthByte) var length int32 err := binary.Read(lengthBuff, binary.LittleEndian, &length) if err != nil { return “”, err } // Buffered返回缓冲中现有的可读取的字节数。 if int32(reader.Buffered()) < length+4 { return “”, err }

  1. // 读取真正的消息数据
  2. pack := make([]byte, int(4+length))
  3. _, err = reader.Read(pack)
  4. if err != nil {
  5. return "", err
  6. }
  7. return string(pack[4:]), nil

} ```