51. 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)
}
服务端运行效果
客户端运行效果

你可能感兴趣的:(51. Socket服务端和客户端使用TCP协议通讯)