Go語言如何實現TCP通信詳解

package main
 
import (
	"bufio"
	"fmt"
	"net"
	"os"
	"strings"
)
 
// TCP 服務端
func process(conn net.Conn) {
	// 函數執行完之後關閉連接
	defer conn.Close()
	// 輸出主函數傳遞的conn可以發現屬於*TCPConn類型, *TCPConn類型那麼就可以調用*TCPConn相關類型的方法, 其中可以調用read()方法讀取tcp連接中的數據
	fmt.Printf("服務端: %T\n", conn)
	for {
		var buf [128]byte
		// 將tcp連接讀取到的數據讀取到byte數組中, 返回讀取到的byte的數目
		n, err := conn.Read(buf[:])
		if err != nil {
			// 從客戶端讀取數據的過程中發生錯誤
			fmt.Println("read from client failed, err:", err)
			break
		}
		recvStr := string(buf[:n])
		fmt.Println("服務端收到客戶端發來的數據：", recvStr)
		// 由於是tcp連接所以雙方都可以發送數據, 下面接收服務端發送的數據這樣客戶端也可以收到對應的數據
		inputReader := bufio.NewReader(os.Stdin)
		s, _ := inputReader.ReadString('\n')
		t := strings.Trim(s, "\r\n")
		// 向當前建立的tcp連接發送數據, 客戶端就可以收到服務端發送的數據
		conn.Write([]byte(t))
	}
}
 
func main() {
	// 監聽當前的tcp連接
	listen, err := net.Listen("tcp", "127.0.0.1:20000")
	fmt.Printf("服務端: %T=====\n", listen)
	if err != nil {
		fmt.Println("listen failed, err:", err)
		return
	}
	for {
		conn, err := listen.Accept() // 建立連接
		fmt.Println("當前建立瞭tcp連接")
		if err != nil {
			fmt.Println("accept failed, err:", err)
			continue
		}
		// 對於每一個建立的tcp連接使用go關鍵字開啟一個goroutine處理
		go process(conn) 
	}
}

package main
 
import (
	"bufio"
	"fmt"
	"net"
	"os"
	"strings"
)
 
func main() {
	// 連接到服務端建立的tcp連接
	conn, err := net.Dial("tcp", "127.0.0.1:20000")
	// 輸出當前建Dial函數的返回值類型, 屬於*net.TCPConn類型
	fmt.Printf("客戶端: %T\n", conn)
	if err != nil {
		// 連接的時候出現錯誤
		fmt.Println("err :", err)
		return
	}
	// 當函數返回的時候關閉連接
	defer conn.Close() 
	// 獲取一個標準輸入的*Reader結構體指針類型的變量
	inputReader := bufio.NewReader(os.Stdin)
	for {
		// 調用*Reader結構體指針類型的讀取方法
		input, _ := inputReader.ReadString('\n') // 讀取用戶輸入
		// 去除掉\r \n符號
		inputInfo := strings.Trim(input, "\r\n")
		// 判斷輸入的是否是Q, 如果是Q則退出
		if strings.ToUpper(inputInfo) == "Q" { // 如果輸入q就退出
			return
		}
		_, err = conn.Write([]byte(inputInfo)) // 發送數據
		if err != nil {
			return
		}
		buf := [512]byte{}
		// 讀取服務端發送的數據
		n, err := conn.Read(buf[:])
		if err != nil {
			fmt.Println("recv failed, err:", err)
			return
		}
		fmt.Println("客戶端接收服務端發送的數據: ", string(buf[:n]))
	}
}