在Go中写入客户端UDP套接字



我正在寻找一个很好的解决方案,用于使用Go语言的UDP套接字进行客户端/服务器通信。

我在互联网上找到的例子向我展示了如何将数据发送到服务器,但它们并没有教我如何将数据发回客户端。

为了演示,我的程序执行以下操作:

我的客户端程序在4444端口上创建一个套接字,如下所示:

con, err := net.Dial("udp", "127.0.0.1:4444")

我向服务器发送了一个字符串和本地地址,这样它就可以打印字符串并发送OK消息。我用gob做这个:

enc := gob.NewEncoder(con)
enc.Encode(Data{"test", con.LocalAddr().String()})

我的数据结构如下:

type Data struct{
    Msg string
    Addr string
}

我的服务器监听4444端口并正确解码Gob,但我如何才能将OK消息发回?我使用客户端地址(在server.go文件上):

con, err := net.Dial("udp", data.Addr)

然后,我得到一个错误代码:

write udp 127.0.0.1:35290: connection refused

当客户端尝试连接到服务器的端口4444时,客户端会创建一个带有随机数的端口(在本例中为35290),以便它们可以通信。我知道我不应该将客户端的地址传递给服务器,但连接RemoteAddress()不起作用。如果能找到客户的地址,我们将不胜感激。

Obs.:我知道有ReadFromUDP,所以我可以读取包。我应该阅读它,发现客户的地址,并将数据发送给Gob,以便它可以解码吗?

检查以下通过UDP进行客户端/服务器通信的示例。sendResponse例程用于将响应发送回客户端。

udpclient.go

package main
import (
    "fmt"
    "net"
    "bufio"
)
func main() {
    p :=  make([]byte, 2048)
    conn, err := net.Dial("udp", "127.0.0.1:1234")
    if err != nil {
        fmt.Printf("Some error %v", err)
        return
    }
    fmt.Fprintf(conn, "Hi UDP Server, How are you doing?")
    _, err = bufio.NewReader(conn).Read(p)
    if err == nil {
        fmt.Printf("%sn", p)
    } else {
        fmt.Printf("Some error %vn", err)
    }
    conn.Close()
}

udpserver.go

package main
import (
    "fmt" 
    "net"  
)

func sendResponse(conn *net.UDPConn, addr *net.UDPAddr) {
    _,err := conn.WriteToUDP([]byte("From server: Hello I got your message "), addr)
    if err != nil {
        fmt.Printf("Couldn't send response %v", err)
    }
}

func main() {
    p := make([]byte, 2048)
    addr := net.UDPAddr{
        Port: 1234,
        IP: net.ParseIP("127.0.0.1"),
    }
    ser, err := net.ListenUDP("udp", &addr)
    if err != nil {
        fmt.Printf("Some error %vn", err)
        return
    }
    for {
        _,remoteaddr,err := ser.ReadFromUDP(p)
        fmt.Printf("Read a message from %v %s n", remoteaddr, p)
        if err !=  nil {
            fmt.Printf("Some error  %v", err)
            continue
        }
        go sendResponse(ser, remoteaddr)
    }
}

hello_echo.go

 package main                                                                                                          
 import (                                                                                                              
     "bufio"                                                                                                           
     "fmt"                                                                                                             
     "net"                                                                                                             
     "time"                                                                                                            
 )                                                                                                                     
 const proto, addr = "udp", ":8888"                                                                                    
 func main() {  
     go func() {                                                                                                       
         conn, _ := net.ListenPacket(proto, addr)                                                                      
         buf := make([]byte, 1024)                                                                                     
         n, dst, _ := conn.ReadFrom(buf)                                                                               
         fmt.Println("serv recv", string(buf[:n]))                                                                     
         conn.WriteTo(buf, dst)                                                                                        
     }()        
     time.Sleep(1 * time.Second)   
     conn, _ := net.Dial(proto, addr)                                                                                  
     conn.Write([]byte("hellon"))                                                                                     
     buf, _, _ := bufio.NewReader(conn).ReadLine()                                                                     
     fmt.Println("clnt recv", string(buf))                                                                             
 }                                                                                                                     

要点在这里:

https://gist.github.com/winlinvip/e8665ba888e2fd489ccd5a449fadfa73

服务器.go

/*
Usage:
    go run server.go
See https://gist.github.com/winlinvip/e8665ba888e2fd489ccd5a449fadfa73
See https://stackoverflow.com/a/70576851/17679565
See https://github.com/ossrs/srs/issues/2843
 */
package main
import (
    "fmt"
    "net"
    "os"
    "strconv"
)
func main() {
    serverPort := 8000
    if len(os.Args) > 1 {
        if v,err := strconv.Atoi(os.Args[1]); err != nil {
            fmt.Printf("Invalid port %v, err %v", os.Args[1], err)
            os.Exit(-1)
        } else {
            serverPort = v
        }
    }
    addr := net.UDPAddr{
        Port: serverPort,
        IP:   net.ParseIP("0.0.0.0"),
    }
    server, err := net.ListenUDP("udp", &addr)
    if err != nil {
        fmt.Printf("Listen err %vn", err)
        os.Exit(-1)
    }
    fmt.Printf("Listen at %vn", addr.String())
    for {
        p := make([]byte, 1024)
        nn, raddr, err := server.ReadFromUDP(p)
        if err != nil {
            fmt.Printf("Read err  %v", err)
            continue
        }
        msg := p[:nn]
        fmt.Printf("Received %v %sn", raddr, msg)
        go func(conn *net.UDPConn, raddr *net.UDPAddr, msg []byte) {
            _, err := conn.WriteToUDP([]byte(fmt.Sprintf("Pong: %s", msg)), raddr)
            if err != nil {
                fmt.Printf("Response err %v", err)
            }
        }(server, raddr, msg)
    }
}

客户端.go

/*
Usage:
    go run client.go
    go run client.go 101.201.77.240
See https://gist.github.com/winlinvip/e8665ba888e2fd489ccd5a449fadfa73
See https://stackoverflow.com/a/70576851/17679565
See https://github.com/ossrs/srs/issues/2843
*/
package main
import (
    "fmt"
    "net"
    "os"
    "strings"
)
func main() {
    serverEP := "127.0.0.1"
    if len(os.Args) > 1 {
        serverEP = os.Args[1]
    }
    if !strings.Contains(serverEP, ":") {
        serverEP = fmt.Sprintf("%v:8000", serverEP)
    }
    conn, err := net.Dial("udp", serverEP)
    if err != nil {
        fmt.Printf("Dial err %v", err)
        os.Exit(-1)
    }
    defer conn.Close()
    msg := "Hello, UDP server"
    fmt.Printf("Ping: %vn", msg)
    if _, err = conn.Write([]byte(msg)); err != nil {
        fmt.Printf("Write err %v", err)
        os.Exit(-1)
    }
    p := make([]byte, 1024)
    nn, err := conn.Read(p)
    if err != nil {
        fmt.Printf("Read err %vn", err)
        os.Exit(-1)
    }
    fmt.Printf("%vn", string(p[:nn]))
}

最新更新