如何从golang的websocket服务器向客户端主动发送消息

Z时代
2024-01-10
分类：问答

我是golang和websocket的新手。

我正在尝试编写一个websocket服务器，一旦握手完成，该服务器可以将消息主动发送到客户端。

但是我的服务器只会在收到来自客户端的请求后才将消息发送给客户端。

有谁知道如何实现此功能，或者在哪里可以找到相关的答案？

非常感谢。

源代码如下：

package main
import (
    "log"
    "net/http"
)
func handler(w http.ResponseWriter, req *http.Request) {
    w.Header().Set("Content-Type", "text/plain")
    w.Write([]byte("Hi, the handshake is completed.\n"))
    w.Write([]byte("Let's start to talk something.\n"))
}
func main() {
    http.HandleFunc("/", handler)
    log.Printf("Start to listen on 443.")
    err := http.ListenAndServeTLS(":443", "server.crt", "server.key", nil)
    log.Fatal(err)
}

回答：

尝试打包websocket。

这是Godoc的一个简单示例：

var upgrader = websocket.Upgrader{
    ReadBufferSize:  1024,
    WriteBufferSize: 1024,
}
func handler(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println(err)
        return
    }
    defer conn.Close()
    msg := []byte("Let's start to talk something.")
    err = conn.WriteMessage(websocket.TextMessage, msg)
    if err != nil {
        log.Println(err)
    }
    // do other stuff...
}