Go語言實現UDP協議及TCP通訊

Posted on by

⼀、使用Golang創建⼀一個TCP連接

1.服務端處理理流程

  • a.監聽端口
  • b.接受客戶端的鏈接
  • c.創建Goroutine，處理這個鏈接(⼀個服務端要鏈接多個客戶端，所以使用Goroutine⾮常簡單)

題外話:要是用Java、C#服務這邊每⼀個請求都開⼀個線程處理的話，頂多⼏千個，但是Goroutine就⾮常簡單。

package main
 
import (
    "fmt"
    "net"
)
 
func main() {
    //1.建立監聽端口
    listen, err := net.Listen("tcp", "0.0.0.0:20000")
    if err != nil {
        fmt.Println("listen failed, err:", err)
        return
    }
 
    fmt.Println("listen Start...:")
 
    for {
        //2.接收客戶端的鏈接
        conn, err := listen.Accept()
        if err != nil {
            fmt.Printf("accept failed, err:%v\n", err)
            continue
        }
        //3.開啟一個Goroutine，處理鏈接
        go process(conn)
    }
}
 
//處理請求，類型就是net.Conn
func process(conn net.Conn) {
 
    //處理結束後關閉鏈接
    defer conn.Close()
    for {
        var buf [128]byte
        n, err := conn.Read(buf[:])
        if err != nil {
            fmt.Printf("read from conn failed, err:%v", err)
            break
        }
        fmt.Printf("recv from client, content:%v\n", string(buf[:n]))
    }
 
}

2.客戶端處理流程

  • a.和服務端建立一個鏈接
  • b.進行數據的收發
  • c.關閉鏈接
package main
 
import (
    "bufio"
    "fmt"
    "net"
    "os"
    "strings"
)
 
func main() {
    //1.建立一個鏈接（Dial撥號）
    conn, err := net.Dial("tcp", "0.0.0.0:20000")
    if err != nil {
        fmt.Printf("dial failed, err:%v\n", err)
        return
    }
 
    fmt.Println("Conn Established...:")
 
    //讀入輸入的信息
    reader := bufio.NewReader(os.Stdin)
    for {
        data, err := reader.ReadString('\n')
        if err != nil {
            fmt.Printf("read from console failed, err:%v\n", err)
            break
        }
 
        data = strings.TrimSpace(data)
        //傳輸數據到服務端
        _, err = conn.Write([]byte(data))
        if err != nil {
            fmt.Printf("write failed, err:%v\n", err)
            break
        }
    }
}

執⾏⼀下:

客戶端:

服務端：

⼆、使用Golang創建⼀個HTTP連接

  • A.HTTP協議是基於TCP協議之上的⽂本協議。
  • B.每行⽂本使⽤\r\n結尾，當連續兩個\r\n時，表示整個數據包結束。
package main
 
import (
    "fmt"
    "io"
    "net"
)
 
func main() {
    conn, err := net.Dial("tcp", "www.baidu.com:80")
    if err != nil {
        fmt.Printf("dial failed, err:%v\n", err)
        return
    }
 
    data := "GET / HTTP/1.1\r\n"
    data += "HOST: www.baidu.com\r\n"
    data += "connection: close\r\n"
    data += "\r\n\r\n"
 
    //寫入數據
    _, err = io.WriteString(conn, data)
    if err != nil {
        fmt.Printf("wirte string failed, err:%v\n", err)
        return
    }
 
    var buf [1024]byte
    for {
        //讀取返回的數據
        n, err := conn.Read(buf[:])
        if err != nil || n == 0 {
            break
        }
 
        fmt.Println(string(buf[:n]))
    }
}

三、使⽤用Golang創建⼀個UDP連接

1.服務流程，相比於TCP連接需要Accept一個鏈接，UDP不需要進⾏這步操作，服務端代碼如下:

package main
 
import (
    "fmt"
    "net"
)
 
func main() {
    //建立一個UDP的監聽，這裡使用的是ListenUDP，並且地址是一個結構體
    listen, err := net.ListenUDP("udp", &net.UDPAddr{
        IP:   net.IPv4(0, 0, 0, 0),
        Port: 8080,
    })
    if err != nil {
        fmt.Printf("listen failed, err:%v\n", err)
        return
    }
 
    for {
        var data [1024]byte
        //讀取UDP數據
        count, addr, err := listen.ReadFromUDP(data[:])
        if err != nil {
            fmt.Printf("read udp failed, err:%v\n", err)
            continue
        }
 
        fmt.Printf("data:%s addr:%v count:%d\n", string(data[0:count]), addr, count)
        //返回數據
        _, err = listen.WriteToUDP([]byte("hello client"), addr)
        if err != nil {
            fmt.Printf("write udp failed, err:%v\n", err)
            continue
        }
    }
}

2.客戶端流程

package main
 
import (
    "fmt"
    "net"
)
 
func main() {
    // 創建連接
    socket, err := net.DialUDP("udp4", nil, &net.UDPAddr{
        IP:   net.IPv4(127, 0, 0, 1),
        Port: 8080,
    })
    if err != nil {
        fmt.Println("連接失敗!", err)
        return
    }
    defer socket.Close()
    // 發送數據
    senddata := []byte("hello server!")
    _, err = socket.Write(senddata)
    if err != nil {
        fmt.Println("發送數據失敗!", err)
        return
    }
    // 接收數據
    data := make([]byte, 4096)
    read, remoteAddr, err := socket.ReadFromUDP(data)
    if err != nil {
        fmt.Println("讀取數據失敗!", err)
        return
    }
    fmt.Println(read, remoteAddr)
    fmt.Printf("%s\n", data)
}

以上就是這篇文章的全部內容瞭，希望本文的內容對大傢的學習或者工作具有一定的參考學習價值，謝謝大傢對WalkonNet的支持。如果你想瞭解更多相關內容請查看下面相關鏈接

推薦閱讀：