GO使用socket和channel實(shí)現(xiàn)簡單控制臺(tái)聊天室
使用socket和channel，實(shí)現(xiàn)簡單控制臺(tái)聊天室
這里使用socket和channel，演示在GO中如何編寫一個(gè)簡單網(wǎng)絡(luò)程序
功能分析
聊天室主要功能：用戶可以加入/離開聊天室；每個(gè)用戶發(fā)送的消息，廣播給所有人
聊天室分為客戶端和服務(wù)端，客戶端負(fù)責(zé)發(fā)送消息和打印服務(wù)器消息，服務(wù)器負(fù)責(zé)接收客戶端消息，并廣播給所有人
客戶端可以使用telnet程序
服務(wù)端是需要實(shí)現(xiàn)的。需要實(shí)現(xiàn)的功能，
- 如何保存多個(gè)客戶端的連接，管理連接的接入與斷開
- 如何接收和廣播客戶端消息
實(shí)現(xiàn)思路
通過功能分析，拆分為聊天室結(jié)構(gòu)體和客戶端結(jié)構(gòu)體
聊天室結(jié)構(gòu)體負(fù)責(zé)管理當(dāng)前接入的客戶端和廣播消息
客戶端結(jié)構(gòu)體負(fù)責(zé)管理socket連接和需要接收與發(fā)送的數(shù)據(jù)
客戶端連接/斷開時(shí)通知聊天室；客戶端發(fā)送的消息實(shí)際是轉(zhuǎn)發(fā)給聊天室，然后聊天室再廣播出去
完整代碼
package main
import (
"bufio"
"fmt"
"log"
"net"
)
type Client struct {
idstring
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()
}
}
分析
實(shí)現(xiàn)的關(guān)鍵是封裝了客戶端通信channel，無論是遠(yuǎn)程發(fā)送過來的消息還是聊天室廣播的消息，都通過這個(gè)channel傳遞，且這個(gè)channel是綁定客戶端的
參考鏈接中，直接使用channel來定義客戶端type client chan<- string，其實(shí)更能表達(dá)這一點(diǎn)
為了容易理解，這里將channel封裝為客戶端的一個(gè)通信管道，客戶端還可以有別的屬性，例如：id、連接和超時(shí)時(shí)間等
參考： Go 網(wǎng)絡(luò)編程示例
到此這篇關(guān)于GO實(shí)現(xiàn)簡單控制臺(tái)聊天室的文章就介紹到這了,更多相關(guān)go控制臺(tái)聊天室內(nèi)容請(qǐng)搜索本站以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持本站！
版權(quán)聲明：本站文章來源標(biāo)注為YINGSOO的內(nèi)容版權(quán)均為本站所有，歡迎引用、轉(zhuǎn)載，請(qǐng)保持原文完整并注明來源及原文鏈接。禁止復(fù)制或仿造本網(wǎng)站，禁止在非www.sddonglingsh.com所屬的服務(wù)器上建立鏡像，否則將依法追究法律責(zé)任。本站部分內(nèi)容來源于網(wǎng)友推薦、互聯(lián)網(wǎng)收集整理而來，僅供學(xué)習(xí)參考，不代表本站立場(chǎng)，如有內(nèi)容涉嫌侵權(quán)，請(qǐng)聯(lián)系alex-e#qq.com處理。
關(guān)注官方微信