写一个golang的websocket

以下是一个简单的 Golang WebSocket 服务器示例代码：

package main

import (
    "fmt"
    "log"
    "net/http"

    "github.com/gorilla/websocket"
)

var upgrader = websocket.Upgrader{
    ReadBufferSize:  1024,
    WriteBufferSize: 1024,
}

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

    for {
        messageType, p, err := conn.ReadMessage()
        if err != nil {
            log.Println(err)
            return
        }

        message := string(p)
        fmt.Printf("Received message: %s\n", message)

        if err := conn.WriteMessage(messageType, p); err != nil {
            log.Println(err)
            return
        }
    }
}

func main() {
    http.HandleFunc("/ws", handleWebSocket)
    log.Fatal(http.ListenAndServe(":8080", nil))
}

这个示例代码使用 Gorilla WebSocket 库来实现 WebSocket 服务器。在 `handleWebSocket` 函数中，我们使用 `upgrader` 对象将 HTTP 连接升级为 WebSocket 连接。然后我们可以读取和写入消息。在这个示例中，我们简单地将收到的消息发送回客户端。

要运行这个示例，可以将上述代码保存到名为 `main.go` 的文件中，然后使用以下命令运行服务器：

go run main.go

然后，您可以使用 WebSocket 客户端连接到 `ws://localhost:8080/ws`。