Hike News
Hike News

Golang網路編程-day29-socket-TCP編程

Golang 本文总阅读量次

introduction

  • socket編程分為客戶端及服務器
  • golang中服務端處理流程：
    1. 監聽端口
    2. 接收客戶端的連接
    3. 創建goroutine，處理該連接
  • golang中客戶端處理流程：
    1. 建立與服務端的連接
    2. 進行數據受發
    3. 關閉連接

  • 一般socket基本的架構如圖所示
    socket

  • golang 將以上的服務皆封裝了，且性能非常高

服務端(server)

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
package main

import (
	"fmt"
	"net"
)

func main(){
	fmt.Println("start server....")
	// 1. 監聽端口
	listen,err := net.Listen("tcp","0.0.0.0:7777")
	if err != nil {
		fmt.Println("Build Listen Action Failed with Error:",err)
		return
	}
	defer listen.Close()
	for {
		// 2. 接收客戶端的連接
		conn,err := listen.Accept()
		if err != nil {
			fmt.Println("Accept Connet Failed with Error",err)
			continue
		}
		// 3. 創建goroutine，處理該連接
		go process(conn)
	}
}

func process(conn net.Conn){
	//退出連接時一定要關閉
	defer conn.Close()
	//建立512bytes的緩衝區
	buf := make([]byte,512)
	for {
		n,err :=  conn.Read(buf)
		if err != nil {
			fmt.Println("read bytes err:",err)
			return
		}
		fmt.Println(string(buf[:n]),"from:",conn.RemoteAddr())    //使用buf[:n]才不會用緩衝區未讀取的情況
		if string(buf[:n]) == "exit" {          
			return
	    }
	}
}
  • 可使用nc連接測試

客戶端(client)

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
package main

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

func send_data(terminal_reader *bufio.Reader,connect net.Conn) {
	for {
		str, err := terminal_reader.ReadString('\n')
		if err != nil {
			fmt.Println("Read string Error:", err)
		}
		str = strings.Trim(str, "\r\n")
		_, err = connect.Write([]byte(str))
		if err != nil {
			fmt.Println("Send Data Failed", err)
		} else {
			fmt.Println("Send data successs!!")
		}
		if str == "exit"{
			return
		}
	}
}


func main(){
	var forward_ip string = "127.0.0.1"
	var forward_port string = "8888"
	// 1. 建立與服務端的連接
	conn,err := net.Dial("tcp",fmt.Sprintf("%s:%s",forward_ip,forward_port))
	if err != nil {
		fmt.Println("Client Dial Error:",err)
	}
	// 3. 退出時務必關閉
	defer conn.Close()
	terminal_reader := bufio.NewReader(os.Stdin)
	// 2. 進行數據收發
	send_data(terminal_reader, conn)

}
Golang