Hike News
Hike News

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

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)

}