a246104d04
* MM-21012: Revamp websocket implementation We replace the old gorilla/websocket implementation with the gobwas/ws library. The gorilla library was in maintenance mode and had a high level API due to which we cannot use that for situations where a large number of concurrent connections needs to be supported. The ws library is a very low-level library that allows us to work with raw net.Conns. We make several improvements: - We completely remove the reader goroutines, and instead replace them with a manual epoll implementation which sends off messages to be read when it receives any data on the connection. This lets us scale to a much larger number of connections. - The reader buffer is eliminated, because we directly read from the connection now. https://mattermost.atlassian.net/browse/MM-21012 ```release-notes Improved the websocket implementation by using epoll manually to read from a websocket. As a result, the number of goroutines is expected to go down by half. ``` * fix tests * fix shadowing errors * final changes * windows support! * Remove pointer to waitgroup * Fix edge case * Trigger CI * Trigger CI Co-authored-by: Mattermod <mattermod@users.noreply.github.com>
58 строки
1.2 KiB
Go
58 строки
1.2 KiB
Go
/*
|
|
Package wsutil provides utilities for working with WebSocket protocol.
|
|
|
|
Overview:
|
|
|
|
// Read masked text message from peer and check utf8 encoding.
|
|
header, err := ws.ReadHeader(conn)
|
|
if err != nil {
|
|
// handle err
|
|
}
|
|
|
|
// Prepare to read payload.
|
|
r := io.LimitReader(conn, header.Length)
|
|
r = wsutil.NewCipherReader(r, header.Mask)
|
|
r = wsutil.NewUTF8Reader(r)
|
|
|
|
payload, err := ioutil.ReadAll(r)
|
|
if err != nil {
|
|
// handle err
|
|
}
|
|
|
|
You could get the same behavior using just `wsutil.Reader`:
|
|
|
|
r := wsutil.Reader{
|
|
Source: conn,
|
|
CheckUTF8: true,
|
|
}
|
|
|
|
payload, err := ioutil.ReadAll(r)
|
|
if err != nil {
|
|
// handle err
|
|
}
|
|
|
|
Or even simplest:
|
|
|
|
payload, err := wsutil.ReadClientText(conn)
|
|
if err != nil {
|
|
// handle err
|
|
}
|
|
|
|
Package is also exports tools for buffered writing:
|
|
|
|
// Create buffered writer, that will buffer output bytes and send them as
|
|
// 128-length fragments (with exception on large writes, see the doc).
|
|
writer := wsutil.NewWriterSize(conn, ws.StateServerSide, ws.OpText, 128)
|
|
|
|
_, err := io.CopyN(writer, rand.Reader, 100)
|
|
if err == nil {
|
|
err = writer.Flush()
|
|
}
|
|
if err != nil {
|
|
// handle error
|
|
}
|
|
|
|
For more utils and helpers see the documentation.
|
|
*/
|
|
package wsutil
|