13.3 TCP 服务器

这节编写一个 TCP 服务器，给客户端返回一个当前日期和时间的网络包。在实践中，这意味着服务器在接收到一个客户端连接后，将从 Unix 系统获取时间和日期，并把这个数据返回给客户端。

这个工具命名为 TCPserver.go，并由 4 部分构成。

第一部分：

package main

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

第二部分包含以下代码：

func main() {
    arguments := os.Args
    if len(arguments) == 1 {
        fmt.Println("Please provide port number")
        return
    }

    PORT := ":" + arguments[1]
    l, err := net.Listen("tcp" , PORT)
    if err != nil {
        fmt.Println(err)
        return
    }
    defer l.Close()

net.Listen() 函数用于监听连接。如果第二个参数不包含 IP 地址，只有端口号的话，net.Listen() 将监听本地系统上所有可用的 IP 地址。

TCPserver.go 的第三段如下：

    c, err := l.Accept()
    if err != nil {
        fmt.Println(err)
        return
    }

Accept() 函数等待接收连接并返回一个通用的 Conn 变量。这个特定的 TCP 服务器有个错误是它只能接收第一个与它建立连接的 TCP 客户端，因为 Accept() 函数是在接下来的 for 循环外调用的。

TCPserver.go 的剩余代码如下：

    for {
        netData, err := bufio.NewReader(c).ReadString('\n')
        if err != nil {
            fmt.Println(err)
            return
        }
        if strings.TrimSpace(string(netData)) == "STOP" {
            fmt.Println("Exiting TCP server!")
            return
        }
        fmt.Print("-> ", string(netData))
        t := time.Now()
        myTime := t.Format(time.RFC3339) + "\n"
        c.Write([]byte(myTime))
    }
}

执行 TCPServer.go 并使用一个 TCP 客户端应用和它交互，将产生如下输出：

$ go run TCPServer.go 8001
-> HELLO
Exiting TCP server!

在客户端这边，您将看到如下输出：

$ nc 127.0.0.1 8001
HELLO
2018-05-07T14:40:05+03:00
STOP

如果这个 TCPServer.go 工具试图使用一个被其他 Unix 进程占用的 TCP 端口，您将看到下面的错误信息：

$ go run TCPserver.go 9000
listen tcp :9000: bind: address already in use

最后，如果这个 TCPServer.go 工具试图使用一个在 1-1024 范围内需要 root 权限的 TCP 端口，您将看到下面的错误信息：

$ go run TCPserver.go 80
listen tcp :80: bind: permission denied