Golang实现WebSocket聊天室教程

var upgrader = websocket.Upgrader{
    CheckOrigin: func(r *http.Request) bool { return true }, // 允许跨域
}

func handleConnections(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Fatal(err)
        return
    }
    defer conn.Close()

    // 连接成功后，等待接收消息
    for {
        var msg string
        err := conn.ReadJSON(&msg)
        if err != nil {
            log.Printf("读取消息错误: %v", err)
            break
        }
        // 后续将消息推送给所有用户
    }
}

type Client struct {
    conn *websocket.Conn
}

var clients = make(map[*Client]bool)
var broadcast = make(chan string)

func handleMessages() {
    for {
        msg := <-broadcast
        for client := range clients {
            err := client.conn.WriteJSON(msg)
            if err != nil {
                client.conn.Close()
                delete(clients, client)
            }
        }
    }
}

const ws = new WebSocket("ws://localhost:8080/ws");
ws.onmessage = function(event) {
    const chatBox = document.getElementById("chat");
    const msg = document.createElement("div");
    msg.textContent = event.data;
    chatBox.appendChild(msg);
};

function send() {
    const input = document.getElementById("input");
    ws.send(input.value);
    input.value = "";
}

func main() {
    http.HandleFunc("/ws", handleConnections)
    http.Handle("/", http.FileServer(http.Dir("./static")))
    log.Println("服务器运行在 :8080")
    log.Fatal(http.ListenAndServe(":8080", nil))
}