fix(prover): grpc server support gzip

Galxe · Dec 26, 2024 · c516805 · c516805
1 parent a45b8f8
commit c516805
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 2 deletions.
diff --git a/prover-service/Cargo.lock b/prover-service/Cargo.lock
diff --git a/prover-service/Cargo.toml b/prover-service/Cargo.toml
@@ -11,7 +11,7 @@ path = "src/bin/server.rs"
 [dependencies]
 sp1-sdk = "3.0.0"
 alloy-sol-types = "0.8.12"
-tonic = "0.10"
+tonic = { version = "0.10", features = ["gzip"] }
 tonic-health = "0.10"
 prost = "0.12"
 tokio = { version = "1.0", features = ["macros", "rt-multi-thread"] }

diff --git a/prover-service/src/bin/server.rs b/prover-service/src/bin/server.rs
@@ -2,6 +2,7 @@ use clap::Parser;
 use prover_service::proto::prover::prover_service_server::ProverServiceServer;
 use prover_service::service::ProverServiceImpl;
 use std::net::SocketAddr;
+use tonic::codec::CompressionEncoding;
 use tonic::transport::Server;
 use tonic_health::server::health_reporter;
 use warp::Filter;
@@ -76,11 +77,16 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
             .await;
     });
 
+    // Start prover service
+    let prover_service = ProverServiceServer::new(service)
+        .accept_compressed(CompressionEncoding::Gzip)
+        .send_compressed(CompressionEncoding::Gzip);
+
     // Serve requests
     log::info!("Health Server and GRPC Prover Server listening on {}", addr);
     Server::builder()
         .add_service(health_service)
-        .add_service(ProverServiceServer::new(service))
+        .add_service(prover_service)
         .serve(addr)
         .await?;
     Ok(())