this repo has no description
switch to use of websocket.Upgrader
+12
-2
+6
-1
cmd/relay/relay/broadcast.go
+6
-1
cmd/relay/relay/broadcast.go
···
54
54
delete(r.consumers, id)
55
55
}
56
56
57
+
var wsUpgrader = websocket.Upgrader{
58
+
ReadBufferSize: 10 << 10,
59
+
WriteBufferSize: 10 << 10,
60
+
}
61
+
57
62
// Main HTTP request handler for clients connecting to the firehose (com.atproto.sync.subscribeRepos)
58
63
func (r *Relay) HandleSubscribeRepos(resp http.ResponseWriter, req *http.Request, since *int64, realIP string) error {
59
64
60
65
ctx, cancel := context.WithCancel(req.Context())
61
66
defer cancel()
62
67
63
-
conn, err := websocket.Upgrade(resp, req, resp.Header(), 10<<10, 10<<10)
68
+
conn, err := wsUpgrader.Upgrade(resp, req, resp.Header())
64
69
if err != nil {
65
70
return fmt.Errorf("upgrading websocket: %w", err)
66
71
}
+6
-1
cmd/relay/testing/producer.go
+6
-1
cmd/relay/testing/producer.go
···
36
36
return &p
37
37
}
38
38
39
+
var wsUpgrader = websocket.Upgrader{
40
+
ReadBufferSize: 1024,
41
+
WriteBufferSize: 1024,
42
+
}
43
+
39
44
func (p *Producer) handleSubscribeRepos(resp http.ResponseWriter, req *http.Request) {
40
45
41
46
ctx, cancel := context.WithCancel(req.Context())
42
47
defer cancel()
43
48
44
-
conn, err := websocket.Upgrade(resp, req, nil, 1024, 1024)
49
+
conn, err := wsUpgrader.Upgrade(resp, req, nil)
45
50
if err != nil {
46
51
slog.Error("websocket upgrade", "err", err)
47
52
return