TCP协议

TCP/IP:传输控制协议/网间协议,是一种面向连接(连接导向)的、可靠的、基于字节流的传输层(Transport layer)通信协议,因为是面向连接的协议,数据像水流一样传输,会存在黏包问题。

TCP服务端

一个TCP服务端可以同时连接很多个客户端,例如世界各地的用户使用自己电脑上的浏览器访问淘宝网。因为Go语言中创建多个goroutine实现并发非常方便和高效,所以可以每建立一次链接就创建一个goroutine去处理。

TCP服务端程序的处理流程:

  1. 1.监听端口
  2. 2.接收客户端请求建立链接
  3. 3.创建goroutine处理链接。

使用Go语言的net包实现的TCP服务端代码如下:

  1. // tcp/server/main.go
  2. // TCP server端
  3. // 处理函数
  4. func process(conn net.Conn) {
  5. defer conn.Close() // 关闭连接
  6. for {
  7. reader := bufio.NewReader(conn)
  8. var buf [128]byte
  9. n, err := reader.Read(buf[:]) // 读取数据
  10. if err != nil {
  11. fmt.Println("read from client failed, err:", err)
  12. break
  13. }
  14. recvStr := string(buf[:n])
  15. fmt.Println("收到client端发来的数据:", recvStr)
  16. conn.Write([]byte(recvStr)) // 发送数据
  17. }
  18. }
  19. func main() {
  20. listen, err := net.Listen("tcp", "127.0.0.1:20000")
  21. if err != nil {
  22. fmt.Println("listen failed, err:", err)
  23. return
  24. }
  25. for {
  26. conn, err := listen.Accept() // 建立连接
  27. if err != nil {
  28. fmt.Println("accept failed, err:", err)
  29. continue
  30. }
  31. go process(conn) // 启动一个goroutine处理连接
  32. }
  33. }

将上面的代码保存之后编译成server或server.exe可执行文件。

TCP客户端

一个TCP客户端进行TCP通信的流程如下:

  1. 1.建立与服务端的链接
  2. 2.进行数据收发
  3. 3.关闭链接

使用Go语言的net包实现的TCP客户端代码如下:

  1. // tcp/client/main.go
  2. // 客户端
  3. func main() {
  4. conn, err := net.Dial("tcp", "127.0.0.1:20000")
  5. if err != nil {
  6. fmt.Println("err :", err)
  7. return
  8. }
  9. defer conn.Close() // 关闭连接
  10. inputReader := bufio.NewReader(os.Stdin)
  11. for {
  12. input, _ := inputReader.ReadString('\n') // 读取用户输入
  13. inputInfo := strings.Trim(input, "\r\n")
  14. if strings.ToUpper(inputInfo) == "Q" { // 如果输入q就退出
  15. return
  16. }
  17. _, err = conn.Write([]byte(inputInfo)) // 发送数据
  18. if err != nil {
  19. return
  20. }
  21. buf := [512]byte{}
  22. n, err := conn.Read(buf[:])
  23. if err != nil {
  24. fmt.Println("recv failed, err:", err)
  25. return
  26. }
  27. fmt.Println(string(buf[:n]))
  28. }
  29. }

将上面的代码编译成client或client.exe可执行文件,先启动server端再启动client端,在client端输入任意内容回车之后就能够在server端看到client端发送的数据,从而实现TCP通信。