Go語言基礎——一個簡單的TCP程式設計

阿新 • • 發佈：2021-01-19

技術標籤：Go基礎 go

Go語言的TCP程式設計與很多其他程式語言的結構順序一樣，只是存在著語法的不同

TCP/IP協議

TCP/IP協議，即傳輸控制協議/網間協議，是一種面向連線的、可靠的、基於位元組流的傳輸層通訊協議，其傳輸的單位是報文段，因為是面向連線的協議，會存在黏包問題。

TCP程式設計的服務端

和Java程式設計中的一樣，Go語言的服務端也可以連線多個客戶端，執行的順序也大致一樣。（都遵循TCP/IP協議，肯定一樣啊，此處為廢話）

Go語言中的服務端的工作流程：
（1）監聽埠，等待被連線
（2）接收客戶端的連線請求
（3）.建立goroutine處理連結。(這裡算是與Java語言的不同之處了，Java中多是以一個執行緒去處理)

Goroutine是建立線上程之上的輕量級的抽象。它允許我們以非常低的代價在同一個地址空間中並行地執行多個函式或者方法。相比於執行緒，它的建立和銷燬的代價要小很多，並且它的排程是獨立於執行緒的。在golang中建立一個goroutine非常簡單，使用“go”關鍵字即可。

服務端程式碼

import (
	"bufio"
     "fmt"
	 "net"
)
//TCP server端
func process(conn net.Conn) {
	defer conn.Close() // 關閉連線
	for {
		reader := 
 bufio.NewReader(conn)
		var buf [128]byte
		n, err := reader.Read(buf[:]) // 讀取資料
		if err != nil {
		//如果有錯誤資訊，就列印錯誤資訊
			fmt.Println("read from client failed, err:", err)
			break
		}
		recvStr := string(buf[:n])
		fmt.Println("收到client端發來的資料：", recvStr)
		conn.Write([]byte(recvStr) 
) // 傳送資料
	}
}

func main() {
	fmt.Println("----------伺服器已經開啟----------")
	listen, err := net.Listen("tcp", "127.0.0.1:20000")
	if err != nil {
		fmt.Println("listen failed, err:", err)
		return
	}
	for {
		conn, err := listen.Accept() // 建立連線
		if err != nil {
			fmt.Println("accept failed, err:", err)
			continue
		}
		go process(conn) // 啟動一個goroutine處理連線
	}
}

TCP程式設計的客戶端

Go語言中的客戶端的工作流程：
（1）建立與伺服器的連線
（2）傳送訊息
（3）斷開連線

客戶端程式碼

package main
import (
	"bufio"
	"fmt"
	"net"
	"os"
	"strings"
)

//客戶端
func main() {
	conn,err :=net.Dial("tcp","127.0.0.1:20000")
	if err!=nil{
		fmt.Println("err:",err)
		return
	}
	defer conn.Close()//關閉連線
	inputReader :=bufio.NewReader(os.Stdin)
	for{
		input,_:=inputReader.ReadString('\n')//讀取使用者輸入
		inputInfo :=strings.Trim(input,"\r\n")
		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]))
	}
}