学习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()完成，并且需要指定对方的地址信息。

 1package main
 2
 3import (
 4        "flag"
 5        "fmt"
 6        "net"
 7        "os"
 8)
 9
10var host = flag.String("host", "", "Specify the hostname")
11var port = flag.Int("port", 12345, "Specify the port")
12
13func main() {
14        flag.Parse()
15
16        var udpAddr = net.UDPAddr{
17                IP:   net.ParseIP(*host),
18                Port: *port,
19        }
20
21        conn, err := net.ListenUDP("udp", &udpAddr)
22        if err != nil {
23                fmt.Println("Error listening:", err)
24                os.Exit(1)
25        }
26        defer conn.Close()
27
28        var data = make([]byte, 1024)
29        for {
30                n, remoteAddr, err := conn.ReadFromUDP(data)      // conn.Read() also works here
31                if err != nil {
32                        fmt.Println("Error read UDP:", err.Error())
33                }
34
35                if n <= 0 {
36                        continue
37                }
38
39                fmt.Printf("[%v]:", remoteAddr)
40                fmt.Println(data[:n])
41
42                _, err = conn.WriteToUDP(data, remoteAddr)
43                //_, err = conn.Write(data)                      // conn.Write() doesn't work here
44                if err != nil {
45                        fmt.Println("Error write UDP:", err.Error())
46                }
47        }
48}

UDP client

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

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

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

 1package main
 2
 3import (
 4        "bufio"
 5        "flag"
 6        "fmt"
 7        "net"
 8        "os"
 9)
10
11var host = flag.String("host", "127.0.0.1", "Specify the target host IP")
12var port = flag.Int("port", 12345, "Specifiy the target host port")
13
14func main() {
15        flag.Parse()
16
17        var targetAddr = net.UDPAddr{
18                IP:   net.ParseIP(*host),
19                Port: *port,
20        }
21
22        conn, err := net.DialUDP("udp", nil, &targetAddr)
23        if err != nil {
24                fmt.Println("Error DialUDP,", err.Error())
25                os.Exit(1)
26        }
27        defer conn.Close()
28
29        reader := bufio.NewReader(os.Stdin)
30        for {
31                fmt.Print("> ")
32                data, _, err := reader.ReadLine()
33                if err != nil {
34                        continue
35                }
36
37                if string(data) == "quit" {
38                        fmt.Println("Quit now ...")
39                        os.Exit(0)
40                }
41
42                fmt.Println("Writing: ", data)
43                //conn.WriteToUDP(data, &targetAddr)      // conn.WriteToUDP() doesn't work here
44                n, err := conn.Write(data)
45                fmt.Printf("Written %d bytes", n)
46                if err != nil {
47                        fmt.Println("Error WriteToUDP: ", err.Error())
48                }
49
50                //n, _, err = conn.ReadFromUDP(data)     // conn.ReadFromUDP() works here
51                n, err = conn.Read(data)
52                if err != nil {
53                        fmt.Println("Error ReadFromUDP: ", err.Error())
54                        continue
55                }
56                fmt.Println(string(data))
57        }
58}

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