fix: Internode communication should be compressed too

sorock/src/service/raft/communicator/mod.rs

@@ -6,6 +6,7 @@ mod stream;
 use heartbeat_multiplex::*;
 use std::sync::Arc;
 use tokio::task::AbortHandle;
+use tonic::codec::CompressionEncoding;
 
 pub struct HandleDrop(AbortHandle);
 impl Drop for HandleDrop {
@@ -31,6 +32,8 @@ impl RaftConnection {
 
             let chan = endpoint.connect_lazy();
             raft::RaftClient::new(chan)
+                .send_compressed(CompressionEncoding::Zstd)
+                .accept_compressed(CompressionEncoding::Zstd)
         };
 
         let heartbeat_buffer = Arc::new(HeartbeatBuffer::new());

tests/sorock-tests/Cargo.toml

@@ -7,6 +7,7 @@ edition = "2021"
 
 [dependencies]
 anyhow.workspace = true
+tonic.workspace = true
 
 env = { path = "../env" } 
 sorock = { workspace = true }

tests/sorock-tests/src/lib.rs

@@ -1,6 +1,7 @@
 use anyhow::{ensure, Result};
 use env::Env;
 use sorock::service::raft::client::*;
+use tonic::codegen::CompressionEncoding;
 
 pub struct Builder {
     with_persistency: bool,
@@ -66,6 +67,8 @@ impl Cluster {
     pub fn admin(&self, id: u8) -> RaftClient {
         let conn = self.env.get_connection(id);
         RaftClient::new(conn)
+            .accept_compressed(CompressionEncoding::Zstd)
+            .send_compressed(CompressionEncoding::Zstd)
     }
 
     /// Request node `to` to add a node `id`.