Go語言通過TCP協(xié)議實現(xiàn)聊天室功能
更新時間:2024年04月23日 10:40:20 作者:谷隱凡二
這篇文章主要為大家詳細介紹了Go語言中如何通過TCP協(xié)議實現(xiàn)聊天室功能,文中的示例代碼講解詳細,感興趣的小伙伴可以跟隨小編一起學習一下
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實現(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ù)包并對其進行分析,本文將介紹如何使用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

