Golang实现WebSocket通信教程

package main

import (
    "log"
    "net/http"
    "github.com/gorilla/websocket"
)

var upgrader = websocket.Upgrader{
    CheckOrigin: func(r *http.Request) bool {
        return true // 允许跨域连接，生产环境应做更严格校验
    },
}

func wsHandler(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Printf("升级失败: %v", err)
        return
    }
    defer conn.Close()

    // 循环读取消息
    for {
        msgType, message, err := conn.ReadMessage()
        if err != nil {
            log.Printf("读取失败: %v", err)
            break
        }

        log.Printf("收到消息: %s", message)

        // 回显消息
        if err := conn.WriteMessage(msgType, message); err != nil {
            log.Printf("发送失败: %v", err)
            break
        }
    }
}

func main() {
    http.HandleFunc("/ws", wsHandler)
    log.Println("服务器启动，地址: :8080")
    log.Fatal(http.ListenAndServe(":8080", nil))
}

const ws = new WebSocket("ws://localhost:8080/ws");
ws.onopen = () => {
    ws.send("你好，Golang WebSocket！");
};
ws.onmessage = (event) => {
    console.log("收到:", event.data);
};
ws.onclose = () => {
    console.log("连接关闭");
};