From b07807809a64a89db8e3b450b49cb14b9bea5089 Mon Sep 17 00:00:00 2001 From: bryan newbold Date: Mon, 14 Apr 2025 19:15:11 -0700 Subject: [PATCH] switch to use of websocket.Upgrader --- cmd/relay/relay/broadcast.go | 7 ++++++- cmd/relay/testing/producer.go | 7 ++++++- 2 files changed, 12 insertions(+), 2 deletions(-) diff --git a/cmd/relay/relay/broadcast.go b/cmd/relay/relay/broadcast.go index ab2e02ec..74f671ae 100644 --- a/cmd/relay/relay/broadcast.go +++ b/cmd/relay/relay/broadcast.go @@ -54,13 +54,18 @@ func (r *Relay) cleanupConsumer(id uint64) { delete(r.consumers, id) } +var wsUpgrader = websocket.Upgrader{ + ReadBufferSize: 10 << 10, + WriteBufferSize: 10 << 10, +} + // Main HTTP request handler for clients connecting to the firehose (com.atproto.sync.subscribeRepos) func (r *Relay) HandleSubscribeRepos(resp http.ResponseWriter, req *http.Request, since *int64, realIP string) error { ctx, cancel := context.WithCancel(req.Context()) defer cancel() - conn, err := websocket.Upgrade(resp, req, resp.Header(), 10<<10, 10<<10) + conn, err := wsUpgrader.Upgrade(resp, req, resp.Header()) if err != nil { return fmt.Errorf("upgrading websocket: %w", err) } diff --git a/cmd/relay/testing/producer.go b/cmd/relay/testing/producer.go index 1bd89681..d3e7d1ef 100644 --- a/cmd/relay/testing/producer.go +++ b/cmd/relay/testing/producer.go @@ -36,12 +36,17 @@ func NewProducer() *Producer { return &p } +var wsUpgrader = websocket.Upgrader{ + ReadBufferSize: 1024, + WriteBufferSize: 1024, +} + func (p *Producer) handleSubscribeRepos(resp http.ResponseWriter, req *http.Request) { ctx, cancel := context.WithCancel(req.Context()) defer cancel() - conn, err := websocket.Upgrade(resp, req, nil, 1024, 1024) + conn, err := wsUpgrader.Upgrade(resp, req, nil) if err != nil { slog.Error("websocket upgrade", "err", err) return -- 2.51.2