GolangWebSocket客户端与服务器实例

package main

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

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

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

    for {
        // 读取消息
        messageType, message, err := conn.ReadMessage()
        if err != nil {
            log.Print("读取失败:", err)
            break
        }

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

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

func main() {
    http.HandleFunc("/ws", echoHandler)
    log.Print("服务器启动在 :8080")
    log.Fatal(http.ListenAndServe(":8080", nil))
}

package main

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

func main() {
    u := url.URL{Scheme: "ws", Host: "localhost:8080", Path: "/ws"}
    log.Printf("连接到 %s", u.String())

    conn, _, err := websocket.DefaultDialer.Dial(u.String(), nil)
    if err != nil {
        log.Fatal("连接失败:", err)
    }
    defer conn.Close()

    // 发送消息
    for i := 1; i <= 5; i++ {
        msg := "Hello, WebSocket " + string(rune('0'+i))
        if err := conn.WriteMessage(websocket.TextMessage, []byte(msg)); err != nil {
            log.Println("发送失败:", err)
            return
        }

        // 读取回显
        _, reply, err := conn.ReadMessage()
        if err != nil {
            log.Println("接收失败:", err)
            return
        }
        log.Printf("回显: %s", reply)

        time.Sleep(time.Second)
    }
}

go mod init websocket-demo
go get github.com/gorilla/websocket