学习golang进行UDP client/server通讯的过程中发现Read/ReadFromUDP/Write/WriteToUDP的使用有些需要注意之处，这里记录一下。

代码实验

UDP server

UDP服务器端在调用”net.ListenUDP()“后创建”net.UDPConn”，read/write操作是通过这个UDPConn来完成的。因为listen的时候只指定了本地绑定的地址，它只能被动的接收来自客户端的消息，因此这个UDPConn在golang中为’unconnected’类型的。

这种类型的UDPConn的读操作可以接受Read()及ReadFromUDP()。区别是Read()无法知道远程连接的地址信息而ReadFromUDP()可以，所以如果后续需要跟远程进行双向通讯需要使用ReadFromUDP()。

这种类型的UDPConn在进行写操作时必须使用WriteToUDP()完成，并且需要指定对方的地址信息。

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48

package main

import (
        "flag"
        "fmt"
        "net"
        "os"
)

var host = flag.String("host", "", "Specify the hostname")
var port = flag.Int("port", 12345, "Specify the port")

func main() {
        flag.Parse()

        var udpAddr = net.UDPAddr{
                IP:   net.ParseIP(*host),
                Port: *port,
        }

        conn, err := net.ListenUDP("udp", &udpAddr)
        if err != nil {
                fmt.Println("Error listening:", err)
                os.Exit(1)
        }
        defer conn.Close()

        var data = make([]byte, 1024)
        for {
                n, remoteAddr, err := conn.ReadFromUDP(data)      // conn.Read() also works here
                if err != nil {
                        fmt.Println("Error read UDP:", err.Error())
                }

                if n <= 0 {
                        continue
                }

                fmt.Printf("[%v]:", remoteAddr)
                fmt.Println(data[:n])

                _, err = conn.WriteToUDP(data, remoteAddr)
                //_, err = conn.Write(data)                      // conn.Write() doesn't work here
                if err != nil {
                        fmt.Println("Error write UDP:", err.Error())
                }
        }
}

UDP client

UDP客户端的连接通过”net.DialUDP()“来创建。因为这个连接创建时指定了远程服务器地址，因此这种连接在golang中称为’connected’类型的连接。

这种类型的连接进行读操作可以使用Read()及ReadFromUDP()，区别在前面已经说过了，主要区别就是ReadFromUDP()会返回远程端的地址信息。

这种类型的连接进行写操作只能用Write()，如果使用WriteToUDP()则无法完成数据发送。

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58

package main

import (
        "bufio"
        "flag"
        "fmt"
        "net"
        "os"
)

var host = flag.String("host", "127.0.0.1", "Specify the target host IP")
var port = flag.Int("port", 12345, "Specifiy the target host port")

func main() {
        flag.Parse()

        var targetAddr = net.UDPAddr{
                IP:   net.ParseIP(*host),
                Port: *port,
        }

        conn, err := net.DialUDP("udp", nil, &targetAddr)
        if err != nil {
                fmt.Println("Error DialUDP,", err.Error())
                os.Exit(1)
        }
        defer conn.Close()

        reader := bufio.NewReader(os.Stdin)
        for {
                fmt.Print("> ")
                data, _, err := reader.ReadLine()
                if err != nil {
                        continue
                }

                if string(data) == "quit" {
                        fmt.Println("Quit now ...")
                        os.Exit(0)
                }

                fmt.Println("Writing: ", data)
                //conn.WriteToUDP(data, &targetAddr)      // conn.WriteToUDP() doesn't work here
                n, err := conn.Write(data)
                fmt.Printf("Written %d bytes", n)
                if err != nil {
                        fmt.Println("Error WriteToUDP: ", err.Error())
                }

                //n, _, err = conn.ReadFromUDP(data)     // conn.ReadFromUDP() works here
                n, err = conn.Read(data)
                if err != nil {
                        fmt.Println("Error ReadFromUDP: ", err.Error())
                        continue
                }
                fmt.Println(string(data))
        }
}

connected vs unconnected

unconnected:

  • 不知道远程地址的socket
  • server调用listenUDP()创建的UDPConn为unconnected状态

connected:

  • 知道远程地址的socket
  • client调用DialUDP()创建的UDPConn为connectd状态
socket状态 Read方法 Write方法
connected ReadFromUDP()
Read()		 Write()
unconnected ReadFromUDP()
Read()		 WriteToUDP()

其他注意点：

  • 1. net.ListenUDP()的第一个参数必须为小写’udp’
  • 2. ReadFromUDP()中的参数[]byte长度必须大于0，否则这个函数会在调用后迅速返回，最终会导致CPU使用率飙升。参见语言编程陷阱，陷阱2