Go語言通過TCP協(xié)議實現(xiàn)聊天室功能
更新時間:2024年04月23日 10:40:20 作者:谷隱凡二
這篇文章主要為大家詳細(xì)介紹了Go語言中如何通過TCP協(xié)議實現(xiàn)聊天室功能,文中的示例代碼講解詳細(xì),感興趣的小伙伴可以跟隨小編一起學(xué)習(xí)一下
1、服務(wù)端
package main import ( "fmt" "net" "sync" ) type ChatServer struct { clients map[string]net.Conn clientsMux sync.Mutex } func NewChatServer() *ChatServer { return &ChatServer{ clients: make(map[string]net.Conn), } } func (s *ChatServer) broadcast(message string, sender string) { s.clientsMux.Lock() defer s.clientsMux.Unlock() for clientAddr, conn := range s.clients { if clientAddr != sender { fmt.Fprintf(conn, "[%s]: %s\n", sender, message) } } } func (s *ChatServer) handleConnection(conn net.Conn) { defer conn.Close() clientAddr := conn.RemoteAddr().String() fmt.Printf("New connection from %s\n", clientAddr) s.clientsMux.Lock() s.clients[clientAddr] = conn s.clientsMux.Unlock() for { buffer := make([]byte, 1024) n, err := conn.Read(buffer) if err != nil { fmt.Printf("Error reading from %s: %v\n", clientAddr, err) break } message := string(buffer[:n]) fmt.Printf("[%s]: %s\n", clientAddr, message) s.broadcast(message, clientAddr) } } func main() { server := NewChatServer() listener, err := net.Listen("tcp", ":8080") if err != nil { fmt.Println("Error:", err) return } defer listener.Close() fmt.Println("TCP chat server listening on :8080") for { conn, err := listener.Accept() if err != nil { fmt.Println("Error accepting connection:", err) continue } go server.handleConnection(conn) } }
2、客戶端
package main import ( "bufio" "fmt" "net" "os" ) func readMessages(conn net.Conn) { reader := bufio.NewReader(conn) for { message, err := reader.ReadString('\n') if err != nil { fmt.Println("Error reading message:", err) break } fmt.Print(message) } } func main() { conn, err := net.Dial("tcp", "127.0.0.1:8080") if err != nil { fmt.Println("Error connecting to server:", err) return } defer conn.Close() go readMessages(conn) for { scanner := bufio.NewScanner(os.Stdin) scanner.Scan() message := scanner.Text() _, err := conn.Write([]byte(message + "\n")) if err != nil { fmt.Println("Error sending message:", err) break } } }
3、實現(xiàn)效果
到此這篇關(guān)于Go語言通過TCP協(xié)議實現(xiàn)聊天室功能的文章就介紹到這了,更多相關(guān)Go TCP實現(xiàn)聊天室內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
Golang標(biāo)準(zhǔn)庫time包日常用法小結(jié)
本文主要介紹了Golang標(biāo)準(zhǔn)庫time包日常用法小結(jié),可以通過它們來獲取當(dāng)前時間、創(chuàng)建指定時間、解析時間字符串、控制時間間隔等操作,感興趣的可以了解一下2023-11-11使用Golang實現(xiàn)對網(wǎng)絡(luò)數(shù)據(jù)包的捕獲與分析
在網(wǎng)絡(luò)通信中,網(wǎng)絡(luò)數(shù)據(jù)包是信息傳遞的基本單位,抓包是一種監(jiān)控和分析網(wǎng)絡(luò)流量的方法,用于獲取網(wǎng)絡(luò)數(shù)據(jù)包并對其進(jìn)行分析,本文將介紹如何使用Golang實現(xiàn)抓包功能,包括網(wǎng)絡(luò)數(shù)據(jù)包捕獲和數(shù)據(jù)包分析,需要的朋友可以參考下2023-11-11詳解Golang中創(chuàng)建error的方式總結(jié)與應(yīng)用場景
Golang中創(chuàng)建error的方式包括errors.New、fmt.Errorf、自定義實現(xiàn)了error接口的類型等,本文主要為大家介紹了這些方式的具體應(yīng)用場景,需要的可以參考一下2023-07-07