首頁 > 軟體

go語言中的udp協定及TCP通訊實現範例

2022-04-14 16:00:49

udp協定

server端

package main
import (
	"fmt"
	"net"
)
func main() {
	// 建立監聽
	socket, err := net.ListenUDP("udp4", &net.UDPAddr{
		IP:   net.IPv4(0, 0, 0, 0),
		Port: 8080,
	})
	if err != nil {
		fmt.Println("監聽失敗!", err)
		return
	}
	defer socket.Close()
	for {
		// 讀取使用者端傳來的資料
		data := make([]byte, 4096)
		read, remoteAddr, err := socket.ReadFromUDP(data)
		if err != nil {
			fmt.Println("讀取資料失敗!", err)
			continue
		}
		fmt.Println(read, remoteAddr)
		fmt.Printf("%snn", data)
		//傳送資料,告訴使用者端已收到
		senddata := []byte("hello client!")
		_, err = socket.WriteToUDP(senddata, remoteAddr)
		if err != nil {
			return
			fmt.Println("傳送資料失敗!", err)
		}
	}
}

client端

package main
import (
	"fmt"
	"net"
)
func main() {
	// 建立連線
	socket, err := net.DialUDP("udp4", nil, &net.UDPAddr{
		IP:   net.IPv4(0, 0, 0, 0),
		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)
}

TCP通訊(基本)

伺服器端

package main
import (
	"fmt"
	"net"
)
func main() {
	//監聽
	listener, err := net.Listen("tcp", "127.0.0.1:8000")
	if err != nil {
		fmt.Println("err = ", err)
		return
	}
	defer listener.Close()
	//阻塞等待使用者連結
	conn, err := listener.Accept()
	if err != nil {
		fmt.Println("err = ", err)
		return
	}
	//接收使用者的請求
	buf := make([]byte, 1024) //1024大小的緩衝區
	n, err1 := conn.Read(buf)
	if err1 != nil {
		fmt.Println("err1 = ", err1)
		return
	}
	fmt.Println("buf = ", string(buf[:n]))
	defer conn.Close() //關閉當前使用者連結
}

使用者端

package main
import (
	"fmt"
	"net"
)
func main() {
	//主動連線伺服器
	conn, err := net.Dial("tcp", "127.0.0.1:8000")
	if err != nil {
		fmt.Println("err = ", err)
		return
	}
	defer conn.Close()
	//傳送資料
	conn.Write([]byte("are u ok?"))
}

TCP通訊並行伺服器

伺服器端

package main
import (
	"fmt"
	"net"
	"strings"
)
//處理使用者請求
func HandleConn(conn net.Conn) {
	//函數呼叫完畢,自動關閉conn
	defer conn.Close()
	//獲取使用者端的網路地址資訊
	addr := conn.RemoteAddr().String()
	fmt.Println(addr, " conncet sucessful")
	buf := make([]byte, 2048)
	for {
		//讀取使用者資料
		n, err := conn.Read(buf)
		if err != nil {
			fmt.Println("err = ", err)
			return
		}
		fmt.Printf("[%s]: %sn", addr, string(buf[:n]))
		fmt.Println("len = ", len(string(buf[:n])))
		//if "exit" == string(buf[:n-1]) { //nc測試
		if "exit" == string(buf[:n-2]) { //自己寫的使用者端測試, 傳送時,多了2個字元, "rn"
			fmt.Println(addr, " exit")
			return
		}
		//把資料轉換為大寫,再給使用者傳送
		conn.Write([]byte(strings.ToUpper(string(buf[:n]))))
	}

}
func main() {
	//監聽
	listener, err := net.Listen("tcp", "127.0.0.1:8000")
	if err != nil {
		fmt.Println("err = ", err)
		return
	}
	defer listener.Close()
	//接收多個使用者
	for {
		conn, err := listener.Accept()
		if err != nil {
			fmt.Println("err = ", err)
			return
		}
		//處理使用者請求, 新建一個協程
		go HandleConn(conn)
	}
}

使用者端

package main
import (
	"fmt"
	"net"
	"os"
)
func main() {
	//主動連線伺服器
	conn, err := net.Dial("tcp", "127.0.0.1:8000")
	if err != nil {
		fmt.Println("net.Dial err = ", err)
		return
	}
	//main呼叫完畢,關閉連線
	defer conn.Close()
	go func() {
		//從鍵盤輸入內容,給伺服器傳送內容
		str := make([]byte, 1024)
		for {
			n, err := os.Stdin.Read(str) //從鍵盤讀取內容, 放在str
			if err != nil {
				fmt.Println("os.Stdin. err = ", err)
				return
			}
			//把輸入的內容給伺服器傳送
			conn.Write(str[:n])
		}
	}()
	//接收伺服器回覆的資料
	//切片緩衝
	buf := make([]byte, 1024)
	for {
		n, err := conn.Read(buf) //接收伺服器的請求
		if err != nil {
			fmt.Println("conn.Read err = ", err)
			return
		}
		fmt.Println(string(buf[:n])) //列印接收到的內容, 轉換為字串再列印
	}
}

以上就是go語言中的udp協定及TCP通訊實現範例的詳細內容,更多關於go語言udp協定TCP通訊的資料請關注it145.com其它相關文章!


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