13.5 UDP 服务器

这节开发的 UDP 服务器的目的是给它的 UDP 客户端返回从 1 到 1,000 的随机数。 这个程序命名为 UDPserver.go， 并分为 4 个部分。

UDPserver.go 的第一部分如下：

package main

import(
    "fmt"
    "math/rand"
    "net"
    "os"
    "strconv"
    "strings"
    "time"
)

func random(min, max, int) int {
    return rand.Intn(max-min) + min
}

UDPserver.go 的第二部分如下：

func main() {
    arguments := os.Args
    if len(arguments) == 1 {
        fmt.Println("Please provide a port number!")
        return
    }
    PORT := ":" + arguments[1]

    s, err := net.ResolveUDPAddr("udp4", PORT)
    if err != nil {
        fmt.Println(err)
        return
    }

UDPServer.go 的第三部分如下：

    connection, err := net.ListenUDP("udp4", s)
    if err != nil {
        fmt.Println(err)
        return
    }

    defer connection.Close()
    buffer := make([]byte, 1024)
    rand.Seed(time.Now().Unix())

对于 UDP 网络，net.ListenUDP() 函数作用类似于 net.ListenTCP()。

UDPserver.go 余下代码如下：

    for {
        n, addr, err := connection.ReadFromUDP(buffer)
        fmt.Print("->", string(buffer[0:n-1]))

        if strings.TrimSpace(string(buffer[0:n])) == "STOP" {
            fmt.Println("Exiting UDP server!")
            return
        }

        data := []byte(strconv.Itoa(random[1, 1001]))
        fmt.Printf("data: %s\n", string(data))
        _, err = connection.WriteToUDP(data, addr)
        if err != nil {
            fmt.Println(err)
            return
        }
    }
}

ReadFromUDP() 函数允许您使用一个字节切片缓冲区从 UDP 连接读取数据。

执行 UDPserver.go 并用 UDPclient.go 连接它，将产生如下输出：

$ go run UDPserver.go 8001
-> Hello!
data: 156
-> Another random number please :)
data: 944
-> Leaving...
data: 491
-> STOP
Exiting UDP server!

客户端输出如下：

$ go run UDPclient.go localhost:8001
The UDP server is 127.0.0.1:8001
>> Hello!
Reply: 156
>> Another random number please :)
Reply: 944
>> Leaving...
Reply: 491
>> STOP
Exiting UDP client!