connect: Use klauspost/compress/gzip decompression (#3314)

* Fix missing parameter in makefile

* connect: Use klauspost/compress/gzip decompression

At the same time this make sures that all handlers and clients use the
same compressor pool. Otherwise every new handler and client maintains
its own pool.

Author: simonswine

Makefile

@@ -201,7 +201,7 @@ docker-image/pyroscope/build: frontend/build go/bin
 docker-image/pyroscope/push: GOOS=linux
 docker-image/pyroscope/push: GOARCH=amd64
 docker-image/pyroscope/push: frontend/build go/bin
-	$(call docker_buildx,--push)
+	$(call docker_buildx,--push,)
 
 define UPDATER_CONFIG_JSON
 {

pkg/api/connect/compression.go

@@ -0,0 +1,33 @@
+package connectapi
+
+import (
+	"io"
+
+	"connectrpc.com/connect"
+	"github.com/klauspost/compress/gzip"
+)
+
+const (
+	compressionGzip = "gzip"
+)
+
+var (
+	gzipPoolHandler = connect.WithCompression(
+		compressionGzip,
+		func() connect.Decompressor { return &gzip.Reader{} },
+		func() connect.Compressor { return gzip.NewWriter(io.Discard) },
+	)
+	gzipPoolClient = connect.WithAcceptCompression(
+		compressionGzip,
+		func() connect.Decompressor { return &gzip.Reader{} },
+		func() connect.Compressor { return gzip.NewWriter(io.Discard) },
+	)
+)
+
+func WithGzipHandler() connect.HandlerOption {
+	return gzipPoolHandler
+}
+
+func WithGzipClient() connect.ClientOption {
+	return gzipPoolClient
+}

pkg/api/connect/connect.go

@@ -7,11 +7,13 @@ import (
 func DefaultClientOptions() []connect.ClientOption {
 	return []connect.ClientOption{
 		connect.WithCodec(ProtoCodec),
+		WithGzipClient(),
 	}
 }
 
 func DefaultHandlerOptions() []connect.HandlerOption {
 	return []connect.HandlerOption{
 		connect.WithCodec(ProtoCodec),
+		WithGzipHandler(),
 	}
 }