var (
	// 存儲(chǔ)所有在線的WebSocket連接（核心：全員廣播的基礎(chǔ)）
	clients = make(map[*websocket.Conn]bool)
	// 存儲(chǔ)連接對(duì)應(yīng)的用戶名（區(qū)分誰發(fā)的消息）
	clientNames = make(map[*websocket.Conn]string)
	// 互斥鎖：多個(gè)協(xié)程同時(shí)修改map，必須加鎖，否則程序崩潰
	clientsMutex sync.Mutex
	// HTTP 升級(jí)為 WebSocket 的工具
	upgrader = websocket.Upgrader{
		CheckOrigin: func(r *http.Request) bool {
			return true // 允許跨域（瀏覽器訪問必備）
		},
	}
)

func main() {
	// 注冊(cè)路由：訪問 /ws 就觸發(fā)聊天處理邏輯
	http.HandleFunc("/ws", wsHandler)

	fmt.Println("WebSocket 服務(wù)啟動(dòng)：ws://127.0.0.1:8081/ws")
	// 核心：?jiǎn)?dòng)HTTP服務(wù)，自動(dòng)為每個(gè)客戶端開goroutine（不阻塞）
	if err := http.ListenAndServe(":8081", nil); err != nil {
		fmt.Println("服務(wù)啟動(dòng)失敗：", err)
	}
}

// 把普通HTTP請(qǐng)求，升級(jí)為長(zhǎng)連接WebSocket
conn, err := upgrader.Upgrade(w, r, nil)
// 函數(shù)結(jié)束自動(dòng)關(guān)閉連接（用戶斷開時(shí)觸發(fā)）
defer conn.Close()

// 服務(wù)端主動(dòng)發(fā)消息：歡迎語
conn.WriteMessage(websocket.TextMessage, []byte("歡迎連接！請(qǐng)輸入你的名字："))
// 讀取用戶輸入的名字
_, msg, err := conn.ReadMessage()
name := string(msg)

// 把當(dāng)前用戶加入在線列表
registerClient(conn, name)
// 廣播：XXX 進(jìn)入聊天室（全員可見）
broadcast("系統(tǒng): " + name + " 進(jìn)入了聊天室")

// defer：函數(shù)最后執(zhí)行（用戶關(guān)閉網(wǎng)頁時(shí)觸發(fā)）
defer func() {
	unregisterClient(conn) // 從在線列表刪除
	broadcast("系統(tǒng): " + name + " 離開了聊天室") // 廣播離開消息
}()

// 死循環(huán)：持續(xù)監(jiān)聽用戶發(fā)的消息
for {
	_, msg, err := conn.ReadMessage()
	if err != nil {
		break // 出錯(cuò)/斷開，退出循環(huán)
	}
	// 拼接消息：用戶名: 內(nèi)容
	chatMsg := fmt.Sprintf("%s: %s", name, string(msg))
	// 廣播給所有人
	broadcast(chatMsg)
}

// 注冊(cè)用戶：加鎖 → 修改 → 解鎖
func registerClient(conn *websocket.Conn, name string) {
	clientsMutex.Lock()   // 加鎖
	defer clientsMutex.Unlock() // 自動(dòng)解鎖
	clients[conn] = true
	clientNames[conn] = name
}

// 注銷用戶：從map刪除
func unregisterClient(conn *websocket.Conn) {
	clientsMutex.Lock()
	defer clientsMutex.Unlock()
	delete(clients, conn)
	delete(clientNames, conn)
}

func broadcast(message string) {
	clientsMutex.Lock()
	defer clientsMutex.Unlock()

	// 遍歷所有在線用戶
	for client := range clients {
		// 給每個(gè)用戶發(fā)送消息
		client.WriteMessage(websocket.TextMessage, []byte(message))
	}
}

package main

import (
    "fmt"
    "net/http"
    "sync"

    "github.com/gorilla/websocket" // go get github.com/gorilla/websocket
)

var (
    // clients 存儲(chǔ)所有在線的 WebSocket 連接
    clients = make(map[*websocket.Conn]bool)
    // clientNames 存儲(chǔ)連接對(duì)應(yīng)的用戶名
    clientNames = make(map[*websocket.Conn]string)
    // clientsMutex 保護(hù) clients 和 clientNames 的并發(fā)讀寫
    clientsMutex sync.Mutex
    // upgrader 用于將 HTTP 請(qǐng)求升級(jí)為 WebSocket 請(qǐng)求
    upgrader = websocket.Upgrader{
       CheckOrigin: func(r *http.Request) bool {
          return true // 允許跨域
       },
    }
)

func main() {
    // 注冊(cè) WebSocket 路由
    http.HandleFunc("/ws", wsHandler)

    fmt.Println("WebSocket 服務(wù)啟動(dòng)：ws://127.0.0.1:8081/ws")
    if err := http.ListenAndServe(":8081", nil); err != nil {
       fmt.Println("服務(wù)啟動(dòng)失敗：", err)
    }
}

// wsHandler 處理 WebSocket 連接請(qǐng)求
func wsHandler(w http.ResponseWriter, r *http.Request) {
    // 1. 升級(jí) HTTP 連接為 WebSocket 連接
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
       fmt.Println("升級(jí)失?。?, err)
       return
    }
    // 確保連接最終被關(guān)閉
    defer conn.Close()

    // 2. 詢問并讀取用戶名
    if err := conn.WriteMessage(websocket.TextMessage, []byte("歡迎連接！請(qǐng)輸入你的名字：")); err != nil {
       return
    }
    _, msg, err := conn.ReadMessage()
    if err != nil {
       return
    }
    name := string(msg)

    // 3. 注冊(cè)新用戶并廣播進(jìn)入消息
    registerClient(conn, name)
    broadcast("系統(tǒng): " + name + " 進(jìn)入了聊天室")
    fmt.Printf("用戶 [%s] 已加入\n", name)

    // 4. 設(shè)置斷開連接時(shí)的清理操作
    defer func() {
       unregisterClient(conn)
       broadcast("系統(tǒng): " + name + " 離開了聊天室")
       fmt.Printf("用戶 [%s] 已斷開\n", name)
    }()

    // 5. 循環(huán)讀取消息并廣播
    for {
       _, msg, err := conn.ReadMessage()
       if err != nil {
          break // 讀取錯(cuò)誤或連接關(guān)閉
       }

       // 構(gòu)造并廣播消息
       chatMsg := fmt.Sprintf("%s: %s", name, string(msg))
       fmt.Println(chatMsg) // 服務(wù)端日志
       broadcast(chatMsg)
    }
}

// registerClient 線程安全地注冊(cè)新連接
func registerClient(conn *websocket.Conn, name string) {
    clientsMutex.Lock()
    defer clientsMutex.Unlock()

    clients[conn] = true
    clientNames[conn] = name
}

// unregisterClient 線程安全地注銷連接
func unregisterClient(conn *websocket.Conn) {
    clientsMutex.Lock()
    defer clientsMutex.Unlock()

    delete(clients, conn)
    delete(clientNames, conn)
}

// broadcast 向所有在線客戶端廣播消息
func broadcast(message string) {
    clientsMutex.Lock()
    defer clientsMutex.Unlock()

    msgBytes := []byte(message)
    for client := range clients {
       // 注意：如果某個(gè)客戶端網(wǎng)絡(luò)阻塞，這里可能會(huì)阻塞較長(zhǎng)時(shí)間
       // 實(shí)際生產(chǎn)中應(yīng)將發(fā)送邏輯放入獨(dú)立的 goroutine 或使用 channel
       err := client.WriteMessage(websocket.TextMessage, msgBytes)
       if err != nil {
          fmt.Printf("廣播消息失敗: %v\n", err)
          client.Close()
          delete(clients, client)
          delete(clientNames, client)
       }
    }
}