使用socket和channel，實現簡單控制臺聊天室
這里使用socket和channel，演示在GO中如何編寫一個簡單網絡程序
功能分析
聊天室主要功能：用戶可以加入/離開聊天室；每個用戶發送的消息，廣播給所有人
聊天室分為客戶端和服務端，客戶端負責發送消息和打印服務器消息，服務器負責接收客戶端消息，并廣播給所有人
客戶端可以使用telnet程序
服務端是需要實現的。需要實現的功能，
- 如何保存多個客戶端的連接，管理連接的接入與斷開
- 如何接收和廣播客戶端消息
實現思路
通過功能分析，拆分為聊天室結構體和客戶端結構體
聊天室結構體負責管理當前接入的客戶端和廣播消息
客戶端結構體負責管理socket連接和需要接收與發送的數據
客戶端連接/斷開時通知聊天室；客戶端發送的消息實際是轉發給聊天室，然后聊天室再廣播出去
完整代碼
|
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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
package mainimport ( "bufio" "fmt" "log" "net")type Client struct { id string conn *net.Conn message chan string}type Hub struct { clients map[*Client]bool entering chan *Client leaving chan *Client messages chan string}func main() { hub := &Hub{ clients: make(map[*Client]bool), entering: make(chan *Client), leaving: make(chan *Client), messages: make(chan string), } listener, err := net.Listen("tcp", ":8000") if err != nil { log.Fatal(err) } go hub.broadcaster() for { conn, err := listener.Accept() if err != nil { log.Println(err) continue } go hub.handleConn(conn) }}func (hub *Hub) broadcaster() { for { select { case msg := <-hub.messages: for cli := range hub.clients { cli.message <- msg } case cli := <-hub.entering: hub.clients[cli] = true case cli := <-hub.leaving: delete(hub.clients, cli) } }}func (hub *Hub) handleConn(conn net.Conn) { defer conn.Close() ch := make(chan string) who := conn.RemoteAddr().String() client := &Client{who, &conn, ch} go hub.writeLoop(client) ch <- "welcome " + client.id hub.messages <- client.id + " join chat" hub.entering <- client hub.readLoop(client) hub.messages <- client.id + " has left" hub.leaving <- client}func (hub *Hub) writeLoop(client *Client) { for msg := range client.message { fmt.Fprintf(*client.conn, "%s\n", msg) }}func (hub *Hub) readLoop(client *Client) { input := bufio.NewScanner(*client.conn) for input.Scan() { hub.messages <- client.id + ": " + input.Text() }} |
分析
實現的關鍵是封裝了客戶端通信channel，無論是遠程發送過來的消息還是聊天室廣播的消息，都通過這個channel傳遞，且這個channel是綁定客戶端的
參考鏈接中，直接使用channel來定義客戶端type client chan<- string，其實更能表達這一點
為了容易理解，這里將channel封裝為客戶端的一個通信管道，客戶端還可以有別的屬性，例如：id、連接和超時時間等
參考： Go 網絡編程示例
到此這篇關于GO實現簡單控制臺聊天室的文章就介紹到這了,更多相關go控制臺聊天室內容請搜索服務器之家以前的文章或繼續瀏覽下面的相關文章希望大家以后多多支持服務器之家！
原文鏈接：https://www.cnblogs.com/snowsteps/p/15711223.html
