Go语言从入门到精通-04｜ Socket客户端与服务端

请从github下载配套代码： https://github.com/shuningzhang/learn_golang
电子书下载地址： https://u19702000.ctfile.com/dir/19702000-33357795-bb2f31/

大家都知道Web服务是基于HTTP协议的，而HTTP协议又是TCP协议上的应用层协议。因此我们在开发一个Web服务之前，我们先介绍一下如何使用Go语言开发一个基于TCP的客户端与服务端程序（关于TCP协议的原理可以参考本号之前的文章《从socket到TCP协议，透彻理解网络编程》）。

程序比较简单，服务端用于等待客户端的连接，连接建立成功后，如果接到客户端的数据，则原封不动的返回相同的数据。客户端建立与服务端的连接，连接建立成功后向服务端发送字符串数据，并等待接收服务端的返回。

服务端

前面已经介绍，服务端用于等待客户端的连接，连接建立成功后，如果接到客户端的数据，则原封不动的返回相同的数据。如下是服务端的代码，监听的端口为8080。

package main

import (
    "fmt"
    "net"  //网络通信的库
    "os"
)

func main() {
	/* 构建服务端的地址，端口 */
    service := ":8080"
    tcpAddr, err := net.ResolveTCPAddr("tcp4", service)
    checkError(err)

    /* 建立对端口的监听 */
    listener, err := net.ListenTCP("tcp", tcpAddr)
    checkError(err)
    for {
        fmt.Println("begin accept")
        /* 等待客户端的连接 */
        conn, err := listener.Accept()
        if err != nil {
            continue
        }

        buf := make([]byte,1024)

        /* 读取客户端发送的数据，并打印 */
        result, err := conn.Read(buf)
        fmt.Println(result, string(buf))

        /* 将数据原封返回给客户端 */
        conn.Write(buf) 
        conn.Close()   
    }
}
func checkError(err error) {
    if err != nil {
        fmt.Fprintf(os.Stderr, 
                    "Fatal error: %s", err.Error())
        os.Exit(1)
    }
}

客户端

客户端建立与服务端的连接，连接建立成功后向服务端发送字符串数据，并等待接收服务端的返回。客户端收到服务端的数据后会在终端打印出来。

package main

import (
    "fmt"
    "io/ioutil"
    "net"
    "os"
)

func main() {
    if len(os.Args) != 2 {
        fmt.Fprintf(os.Stderr, 
        			"Usage: %s host:port ", os.Args[0])
        os.Exit(1)
    }

    /* 这里解析从命令行输入的参数，包括地址和端口 */
    service := os.Args[1]
    tcpAddr, err := net.ResolveTCPAddr("tcp4", service)
    checkError(err)

    /* 建立与服务端的连接 */
    conn, err := net.DialTCP("tcp", nil, tcpAddr)
    checkError(err)

    /* 向服务端发送数据 */
    _, err = conn.Write([]byte("Hello World!\r\n"))
    checkError(err)

    /* 接收服务端发送的数据，并在终端打印           */
    result, err := ioutil.ReadAll(conn)
    checkError(err)
    fmt.Println(string(result))
    os.Exit(0)
}
func checkError(err error) {
    if err != nil {
        fmt.Fprintf(os.Stderr, 
                    "Fatal error: %s", err.Error())
        os.Exit(1)
    }
}

测试验证

测试验证也比较简单，首先需要注意观察一下服务端的ip地址，然后在服务端启动服务端的程序。假设服务端的地址为192.168.142.128，此时在客户端的中断输入如下命令：

./client 192.168.142.128:8080

此时可以看到有Hello World！字符串输出。说明整个通信过程是正确的。如果会用Wireshark或者tcpdump等网络监测工具，可以看一下整个交互过程。