Use a header that won't get rewritten by a reverse-proxy

bluesky-social · Sep 22, 2024 · 3a85ad0 · 3a85ad0
1 parent 836b317
commit 3a85ad0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/cmd/jetstream/server.go b/cmd/jetstream/server.go
@@ -311,8 +311,8 @@ func (s *Server) HandleSubscribe(c echo.Context) error {
 	}
 
 	// Check if the user wants zstd compression
-	acceptEncoding := c.Request().Header.Get("Accept-Encoding")
-	compress := strings.Contains(acceptEncoding, "zstd")
+	socketEncoding := c.Request().Header.Get("Socket-Encoding")
+	compress := strings.Contains(socketEncoding, "zstd")
 
 	var cursor *int64
 	var err error

diff --git a/pkg/client/client.go b/pkg/client/client.go
@@ -64,7 +64,7 @@ func NewClient(config *ClientConfig, logger *slog.Logger, scheduler Scheduler) (
 	}
 
 	if config.Compress {
-		c.config.ExtraHeaders["Accept-Encoding"] = "zstd"
+		c.config.ExtraHeaders["Socket-Encoding"] = "zstd"
 		dec, err := zstd.NewReader(nil, zstd.WithDecoderDicts(models.ZSTDDictionary))
 		if err != nil {
 			return nil, fmt.Errorf("failed to create zstd decoder: %w", err)