Go语言实现TCP通信TCP通信

Go语言实现TCP通信TCP通信

client端

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

1. 向指定ip和端口拨号，请求建立连接（Tcp或Udp）
2. 进行数据收发（从终端和从服务端读数据）
3. 关闭链接

package main

import (
   "bufio"
   "fmt"
   "net"
   "os"
   "strings"
)

func main() {

   conn, err := net.Dial("tcp", "127.0.0.1:8888")
   if err != nil {
      fmt.Println("client dial err=", err)
      return
   }

   reader := bufio.NewReader(os.Stdin)

   for {

      //从终端读取数据（会以文件的形式）
      line, err := reader.ReadString('\n')
      if err != nil {
         fmt.Println("readstring err=", err)
      }
      //如多用户输入exit就退出
      line = strings.Trim(line, "\r\n")
      if line == "exit" {
         fmt.Println("客户端退出")
         return

      }
      //将line发给服务器
      n, err := conn.Write([]byte(line + "\n"))
      if err != nil {
         fmt.Println("conn.Write err=", err)
      }
      fmt.Printf("发送了%d个字节\n",n)

      //创建切片
      buf := make([]byte, 1024)

      //1 等待客户端通过conn发送信息
      //2 如果没有writer发送就一直阻塞在这
      n, err = conn.Read(buf)
      if err != nil {
         fmt.Println("服务器read err=", err) //出错退出
         return
      }
      //3. 显示读取内容到终端

      fmt.Print(string(buf[:n]))

   }

}

server端

server端TCP服务端程序的处理流程：

1. 监听端口
2. 接收客户端请求建立链接
3. 创建goroutine处理链接（数据的收发）。

package main

import (
   "bufio"
   "fmt"
   "net"
   "os"
   "strings"
)

func process(conn net.Conn) {

   //循环接受客户端发送数据,不让协程退出
   defer conn.Close() //关闭conn连接
   for {

      //创建切片
      buf := make([]byte, 1024)

      //1 等待客户端通过conn发送信息
      //2 如果没有writer发送就一直阻塞
      n, err := conn.Read(buf)
      if err != nil {
         fmt.Println("服务器read err=", err) //出错退出
         return
      }
      //3. 显示读取内容到终端

      fmt.Print(string(buf[:n]))

      //发送消息给客户端
      reader := bufio.NewReader(os.Stdin)
      //从终端读取数据（会以文件的形式）
      line, err := reader.ReadString('\n')
      if err != nil {
         fmt.Println("readstring err=", err)
      }
      //如多用户输入exit就退出
      line = strings.Trim(line, "\r\n")
      if line == "exit" {
         fmt.Println("客户端退出")
         return

      }
      //将line发给服务器

      _, err = conn.Write([]byte(line + "\n"))
      if err != nil {
         fmt.Println("conn.Write err=", err)
      }

   }

}

func main() {

   fmt.Println("服务器开始监听------")
   listen, err := net.Listen("tcp", "0.0.0.0:8888")
   if err != nil {
      fmt.Println("listen err=", err)
      return
   }

   defer listen.Close() //延时关闭listen

   for {

      fmt.Println("-----等待客户连接----")
      conn, err := listen.Accept()
      if err != nil {
         fmt.Println("Accept err=", err)
      } else {
         fmt.Printf("Accept success con类型%T,客户ip%v\n", conn, conn.RemoteAddr().String())
      }

      //这里要起一个协程
      go process(conn)
   }

}

上面的client和server只是对tcp通信进行了简单的模拟，旨在联系tcp的通信过程。

代码还有个bug，就是client和server都只能收一句、发一句、收一句...交替进行，因为程序中收发是顺序进行的，例如client，发送一次数据之后，程序就阻塞到了 n, err := conn.Read(buf)这个位置，所以只有再server发送来数据并读取后，才可以再次在终端输入数据发送。