GO使用socket和channel實現簡單控制臺聊天室
使用socket和channel,實現簡單控制臺聊天室
這裡使用socket和channel,演示在GO中如何編寫一個簡單網絡程序
功能分析
聊天室主要功能:用戶可以加入/離開聊天室;每個用戶發送的消息,廣播給所有人
聊天室分為客戶端和服務端,客戶端負責發送消息和打印服務器消息,服務器負責接收客戶端消息,並廣播給所有人
客戶端可以使用telnet程序
服務端是需要實現的。需要實現的功能,
- 如何保存多個客戶端的連接,管理連接的接入與斷開
- 如何接收和廣播客戶端消息
實現思路
通過功能分析,拆分為聊天室結構體和客戶端結構體
聊天室結構體負責管理當前接入的客戶端和廣播消息
客戶端結構體負責管理socket連接和需要接收與發送的數據
客戶端連接/斷開時通知聊天室;客戶端發送的消息實際是轉發給聊天室,然後聊天室再廣播出去
完整代碼
package main import ( "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控制臺聊天室內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet!