当前位置 博文首页 > Golang 实现Socket服务端和客户端使用TCP协议通讯

    Golang 实现Socket服务端和客户端使用TCP协议通讯

    作者:冷月醉雪 时间:2021-02-09 06:27

    Socket服务器是网络服务中常用的服务器。使用go语言实现这个业务场景是很容易的。

    这样的网络通讯,需要一个服务端和至少一个客户端。

    我们计划构建一个这样的通讯工程。服务端启动后等待客户端的访问。客户端发送一段信息给服务端。服务端接收到信息后,再回馈给客户端一段信息。

    首先要建立服务端。服务端最先要做的事情就是"建立Socket端口监听"。

    netListen, err := net.Listen("tcp", "localhost:1024")

    上面的代码,表名监听的是本机端口1024,而使用的通讯协议是TCP。

    当监听结束,模块任务完成后,最后要close这个netListen。

    defer netListen.Close()

    使用日志功能,让服务端窗口能看到服务端已经运行了。

    Log("Waiting for clients ...")

    之后使用一个for循环,无尽的等待那些不知何时来访问的客户端信息。

    for循环体内,要监听netListen的信息接收情况:

    conn, err := netListen.Accept()

    当有来自客户端的访问时,接受访问。并在服务端的日志记录已经有客户端连接成功了。

    Log(conn.RemoteAddr().String(), "tcp connect success")

    conn.RemoteAddr().String()表示的就是远程客户端。

    然后,我们开启一个goroutine处理连接任务。

    go handleConnection(conn)

    处理过程就是接收客户端信息和反馈给客户端信息。

    n, err := conn.Read(buffer)

    conn.Write([]byte(strTemp))

    服务端代码示例

    package main 
    import (
      "net"
      "fmt"
      "os"
      "log"
      "time"
    )
     
    func main() {
      //建立socket端口监听
      netListen, err := net.Listen("tcp", "localhost:1024")
      CheckError(err) 
      defer netListen.Close() 
      Log("Waiting for clients ...")
     
      //等待客户端访问
      for{
        conn, err := netListen.Accept()   //监听接收
        if err != nil{
          continue    //如果发生错误,继续下一个循环。
        } 
        Log(conn.RemoteAddr().String(), "tcp connect success") //tcp连接成功
        go handleConnection(conn)
      }
    }
     
    //处理连接
    func handleConnection(conn net.Conn) {
      buffer := make([]byte, 2048)    //建立一个slice
      for{
        n, err := conn.Read(buffer)   //读取客户端传来的内容
        if err != nil{
          Log(conn.RemoteAddr().String(), "connection error: ", err)
          return   //当远程客户端连接发生错误(断开)后,终止此协程。
        } 
        Log(conn.RemoteAddr().String(), "receive data string:\n", string(buffer[:n]))
     
        //返回给客户端的信息
        strTemp := "CofoxServer got msg \""+string(buffer[:n])+"\" at "+time.Now().String()
        conn.Write([]byte(strTemp))
      }
    }
     
    //日志处理
    func Log(v ...interface{}) {
      log.Println(v...)
    }
     
    //错误处理
    func CheckError(err error) {
      if err != nil{
        fmt.Fprintf(os.Stderr, "Fatal error: %s", err.Error())
      }
    }

    客户端的业务逻辑是,发送信息给服务端,然后接收服务端的反馈。

    conn, err := net.DialTCP("tcp", nil, tcpAddr)

    用TCP协议拨号(Dial)到服务端。如果没有发生错误,就说明拨通了。于是在客户端日志记录连接成功

    fmt.Println("connection success")

    然后在这个已经通畅的连接里,进行发送和接收信息的任务。conn.Write([]byte(words))是发送信息;conn.Read(buffer)是接收信息。如果接收发生错误,就记录错误:

    Log(conn.RemoteAddr().String(), "waiting server back msg error: ", err)

    并且中断进程。

    如果没有发生错误,酒吧接收到的信息在日志中记录。

    Log(conn.RemoteAddr().String(), "receive server back msg: ", string(buffer[:n]))

    客户端代码示例

    package main 
    import (
      "net"
      "fmt"
      "log"
      "os"
    )
    //发送信息
    func sender(conn net.Conn) {
      words := "Hello Server!"
      conn.Write([]byte(words))
      fmt.Println("send over")
     
      //接收服务端反馈
      buffer := make([]byte, 2048) 
      n, err := conn.Read(buffer)
      if err != nil {
        Log(conn.RemoteAddr().String(), "waiting server back msg error: ", err)
        return
      }
      Log(conn.RemoteAddr().String(), "receive server back msg: ", string(buffer[:n]))
     
    }
    //日志
    func Log(v ...interface{}) {
      log.Println(v...)
    }
     
    func main() {
      server := "127.0.0.1:1024"
      tcpAddr, err := net.ResolveTCPAddr("tcp4", server)
      if err != nil {
        fmt.Fprintf(os.Stderr, "Fatal error: %s", err.Error())
        os.Exit(1)
      }
      conn, err := net.DialTCP("tcp", nil, tcpAddr)
      if err != nil {
        fmt.Fprintf(os.Stderr, "Fatal error: %s", err.Error())
        os.Exit(1)
      }
     
      fmt.Println("connection success")
      sender(conn)
    }

    以上为个人经验,希望能给大家一个参考,也希望大家多多支持站长博客。如有错误或未考虑完全的地方,望不吝赐教。

    js