首頁 > 軟體

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

2022-07-25 10:01:12

⼀、使用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:%vn", 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:%vn", 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:%vn", 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:%vn", err)
            break
        }
 
        data = strings.TrimSpace(data)
        //傳輸資料到伺服器端
        _, err = conn.Write([]byte(data))
        if err != nil {
            fmt.Printf("write failed, err:%vn", err)
            break
        }
    }
}

執⾏⼀下:

使用者端:

伺服器端:

⼆、使用Golang建立⼀個HTTP連線

  • A.HTTP協定是基於TCP協定之上的⽂本協定。
  • B.每行⽂本使⽤rn結尾,當連續兩個rn時,表示整個封包結束。
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:%vn", err)
        return
    }
 
    data := "GET / HTTP/1.1rn"
    data += "HOST: www.baidu.comrn"
    data += "connection: closern"
    data += "rnrn"
 
    //寫入資料
    _, err = io.WriteString(conn, data)
    if err != nil {
        fmt.Printf("wirte string failed, err:%vn", 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:%vn", err)
        return
    }
 
    for {
        var data [1024]byte
        //讀取UDP資料
        count, addr, err := listen.ReadFromUDP(data[:])
        if err != nil {
            fmt.Printf("read udp failed, err:%vn", err)
            continue
        }
 
        fmt.Printf("data:%s addr:%v count:%dn", string(data[0:count]), addr, count)
        //返回資料
        _, err = listen.WriteToUDP([]byte("hello client"), addr)
        if err != nil {
            fmt.Printf("write udp failed, err:%vn", 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("%sn", data)
}

以上就是這篇文章的全部內容了,希望本文的內容對大家的學習或者工作具有一定的參考學習價值,謝謝大家對it145.com的支援。如果你想了解更多相關內容請檢視下面相關連結


IT145.com E-mail:sddin#qq.com