support grpc gzip compression

centrifugal · Oct 7, 2023 · c253bb1 · c253bb1
1 parent 0c5c2ae
commit c253bb1
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/internal/proxy/grpc.go b/internal/proxy/grpc.go
@@ -7,6 +7,8 @@ import (
 	"net/url"
 	"strings"
 
+	"google.golang.org/grpc/encoding/gzip"
+
 	"github.com/centrifugal/centrifugo/v5/internal/middleware"
 	"github.com/centrifugal/centrifugo/v5/internal/proxyproto"
 
@@ -64,6 +66,9 @@ func getDialOpts(p Proxy) ([]grpc.DialOption, error) {
 	} else {
 		dialOpts = append(dialOpts, grpc.WithTransportCredentials(insecure.NewCredentials()))
 	}
+	if p.GrpcCompression {
+		dialOpts = append(dialOpts, grpc.WithDefaultCallOptions(grpc.UseCompressor(gzip.Name)))
+	}
 
 	if p.testGrpcDialer != nil {
 		dialOpts = append(dialOpts, grpc.WithContextDialer(p.testGrpcDialer))

diff --git a/internal/proxy/proxy.go b/internal/proxy/proxy.go
@@ -42,6 +42,8 @@ type Proxy struct {
 	GrpcCredentialsKey string `mapstructure:"grpc_credentials_key" json:"grpc_credentials_key,omitempty"`
 	// GrpcCredentialsValue is a custom value for GrpcCredentialsKey.
 	GrpcCredentialsValue string `mapstructure:"grpc_credentials_value" json:"grpc_credentials_value,omitempty"`
+	// GrpcCompression enables compression for outgoing calls (gzip).
+	GrpcCompression bool `mapstructure:"grpc_compression" json:"grpc_compression,omitempty"`
 
 	testGrpcDialer func(context.Context, string) (net.Conn, error)
 }

diff --git a/main.go b/main.go
@@ -81,6 +81,7 @@ import (
 	"golang.org/x/crypto/acme/autocert"
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/credentials"
+	_ "google.golang.org/grpc/encoding/gzip"
 	"google.golang.org/grpc/keepalive"
 	"google.golang.org/grpc/reflection"
 )
@@ -271,6 +272,7 @@ var defaults = map[string]any{
 	"proxy_binary_encoding":         false,
 	"proxy_include_connection_meta": false,
 	"proxy_grpc_cert_file":          "",
+	"proxy_grpc_compression":        false,
 
 	"tarantool_mode":     "standalone",
 	"tarantool_address":  "tcp://127.0.0.1:3301",
@@ -1746,6 +1748,7 @@ func proxyMapConfig() (*client.ProxyMap, bool) {
 	p.GrpcCertFile = v.GetString("proxy_grpc_cert_file")
 	p.GrpcCredentialsKey = v.GetString("proxy_grpc_credentials_key")
 	p.GrpcCredentialsValue = v.GetString("proxy_grpc_credentials_value")
+	p.GrpcCompression = v.GetBool("proxy_grpc_compression")
 
 	connectEndpoint := v.GetString("proxy_connect_endpoint")
 	connectTimeout := GetDuration("proxy_connect_timeout")