Adding tonic with gzip feature

blockworks-foundation · Apr 22, 2024 · 950559b · 950559b
1 parent 58b05ce
commit 950559b
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 1 deletion.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -31,6 +31,7 @@ bincode = "1.3.3"
 
 csv = "1.3.0"
 
+tonic = { version="0.10.2", features=["gzip"] }
 tonic-health = "0.10.2"
 
 [dev-dependencies]

diff --git a/src/yellowstone_grpc_util.rs b/src/yellowstone_grpc_util.rs
@@ -5,6 +5,7 @@ use yellowstone_grpc_proto::geyser::geyser_client::GeyserClient;
 use yellowstone_grpc_proto::geyser::SubscribeRequest;
 use yellowstone_grpc_proto::prost::bytes::Bytes;
 use yellowstone_grpc_proto::tonic;
+use tonic::codec::CompressionEncoding;
 use yellowstone_grpc_proto::tonic::metadata::errors::InvalidMetadataValue;
 use yellowstone_grpc_proto::tonic::metadata::AsciiMetadataValue;
 use yellowstone_grpc_proto::tonic::service::Interceptor;
@@ -87,7 +88,9 @@ where
     let client = GeyserGrpcClient::new(
         HealthClient::with_interceptor(channel.clone(), interceptor.clone()),
         GeyserClient::with_interceptor(channel, interceptor)
-            .max_decoding_message_size(GeyserGrpcClient::max_decoding_message_size()),
+            .max_decoding_message_size(GeyserGrpcClient::max_decoding_message_size())
+            .accept_compressed(CompressionEncoding::Gzip)
+            .send_compressed(CompressionEncoding::Gzip),
     );
     Ok(client)
 }