this repo has no description
0
fork

Configure Feed

Select the types of activity you want to include in your feed.

switch to use of websocket.Upgrader

+12 -2
+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
··· 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