go簡易udp客戶端和服務器

go簡易udp socket客戶端和服務器

1.Socket編程

以前使用Socket編程時，一般是如下步驟

1. 建立socket，socket
2. 綁定socket，bind
3. 監聽，listen
4. 接受連接，accept
5. 接受/發送，recv/send

Go語言對其進行了抽象和封裝，剛開始接觸有可能不太適應（譬如我第一天用的時候覺得API好難找……建議參考文檔），後來發現用起來很爽

簡單來說，客戶端省去了很多！客戶端只需要調用net.Dial()即可，服務器我這裏還需要摸索一下，但是也是很簡單了，不過流程感覺沒簡化- -

廢話不多說，直接上代碼

2.Server端

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

func checkError(err error){
    if  err != nil {
        fmt.Println("Error: %s", err.Error())
        os.Exit(1)  
    }
}

func recvUDPMsg(conn *net.UDPConn){
    var buf [20]byte

    n, raddr, err := conn.ReadFromUDP(buf[0:])
    if err != nil {
        return  
    }

    fmt.Println("msg is ", string(buf[0:n]))

    //WriteToUDP
    //func (c *UDPConn) WriteToUDP(b []byte, addr *UDPAddr) (int, error) 
    _, err = conn.WriteToUDP([]byte("nice to see u"), raddr)
    checkError(err)
}

func main() {
    udp_addr, err := net.ResolveUDPAddr("udp", ":11110")
    checkError(err) 

    conn, err := net.ListenUDP("udp", udp_addr)
    defer conn.Close()
    checkError(err) 

    //go recvUDPMsg(conn)
    recvUDPMsg(conn)
}

流程

1. 先通過net.ResolveUDPAddr創建監聽地址
2. net.ListenUDP創建監聽鏈接
3. 然後通過conn.ReadFromUDP和conn.WriteToUDP收發UDP報文

3.Client端

package main

import (
    "os"
    "fmt"
    "net"
//  "io"
)

func main() {
    conn, err := net.Dial("udp", "127.0.0.1:11110")
    defer conn.Close()
    if err != nil {
        os.Exit(1)  
    }

    conn.Write([]byte("Hello world!"))  

    fmt.Println("send msg")

    var msg [20]byte
    conn.Read(msg[0:])

    fmt.Println("msg is", string(msg[0:10]))
}

客戶端非常簡單，先通過net.Dial(“udp”, “127.0.0.1:11110”)，建立發送報文至本機11110端口的socket，然後使用conn.Write和conn.Read收發包，當然conn.ReadFromUDP和conn.WriteToUDP也是可以的