来自websocket的Golang bufio在第一次阅读后中断

Question

我正在尝试从 websocket 流式传输 JSON 文本。但是，在初次阅读后，我注意到流似乎中断/断开连接。这是来自 Pleroma 服务器（想想：Mastodon）。我正在使用默认的 Golang websocket 库。package mainimport (    "bufio"    "fmt"    "log"    "golang.org/x/net/websocket")func main() {    origin := "https://poa.st/"    url := "wss://poa.st/api/v1/streaming/?stream=public"    ws, err := websocket.Dial(url, "", origin)    if err != nil {        log.Fatal(err)    }    s := bufio.NewScanner(ws)    for s.Scan() {        line := s.Text()        fmt.Println(line)    }}在初始 JSON 文本响应之后，for 循环中断。我希望它每隔几秒钟发送一条新消息。这可能是什么原因造成的？如果我可以使用 Gorilla websocket 库，我愿意切换到bufio.谢谢！

Answer 1

尽管 x/net/websocket 连接具有与 io.Reader 中的 Read 方法具有相同签名的 Read 方法，但该连接不像 io.Reader 那样工作。使用 bufio.Scanner 包裹时，连接将无法正常工作。

poa.st 端点发送消息流，其中每条消息都是一个 JSON 文档。使用以下代码使用Gorilla 包读取消息：

url := "wss://poa.st/api/v1/streaming/?stream=public"

ws, _, err := websocket.DefaultDialer.Dial(url, nil)

if err != nil {

    log.Fatal(err)

}

defer ws.Close()

for {

    _, p, err := ws.ReadMessage()

    if err != nil {

        log.Fatal(err)

    }

    // p is a []byte containing the JSON document.

    fmt.Printf("%s\n", p)

}

Gorilla 包有一个用于解码 JSON 消息的辅助方法。这是一个如何使用该方法的示例。

url := "wss://poa.st/api/v1/streaming/?stream=public"

ws, _, err := websocket.DefaultDialer.Dial(url, nil)

if err != nil {

    log.Fatal(err)

}

defer ws.Close()

for {

    // The JSON documents are objects containing two fields, 

    // the event type and the payload. The payload is a JSON

    // document itself.

    var e struct {

        Event   string

        Payload string

    }

    err := ws.ReadJSON(&e)

    if err != nil {

        log.Fatal(err)

    }

    // TODO: decode e.Payload based on e.Event

}