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

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)
}

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