support setting gzip level (tikv#577)

Signed-off-by: zkkxu <[email protected]>

Author: zkkxu

.github/workflows/ci.yml

@@ -83,6 +83,7 @@ jobs:
     - run: cargo build
     - run: cargo test --all
     - run: RUSTFLAGS="-Z sanitizer=address" cargo test --all --target x86_64-unknown-linux-gnu
+    - run: cargo test --features "nightly"
 
   Mac:
     name: Mac
@@ -98,6 +99,7 @@ jobs:
     - run: cargo build --no-default-features --features "prost-codec"
     - run: cargo build
     - run: cargo test --all
+    - run: cargo test --features "nightly"
 
   Mac-openssl:
     name: Mac-openssl
@@ -124,6 +126,7 @@ jobs:
     - run: cargo xtask submodule
     - run: cargo build
     - run: cargo test --all
+    - run: cargo test --features "nightly"
 
   Pre-Release:
     name: Pre-Release

Cargo.toml

@@ -45,6 +45,7 @@ default = ["protobuf-codec", "boringssl"]
 _secure = []
 protobuf-codec = ["protobuf"]
 prost-codec = ["prost", "bytes"]
+nightly = []
 boringssl = ["grpcio-sys/boringssl", "_secure"]
 openssl = ["_secure", "grpcio-sys/openssl"]
 openssl-vendored = ["_secure", "grpcio-sys/openssl-vendored"]

grpc-sys/bindings/bindings.rs

@@ -76,6 +76,9 @@ pub const GRPC_ARG_SERVICE_CONFIG_DISABLE_RESOLUTION: &[u8; 39usize] =
 pub const GRPC_ARG_LB_POLICY_NAME: &[u8; 20usize] = b"grpc.lb_policy_name\0";
 pub const GRPC_ARG_SOCKET_MUTATOR: &[u8; 20usize] = b"grpc.socket_mutator\0";
 pub const GRPC_ARG_SOCKET_FACTORY: &[u8; 20usize] = b"grpc.socket_factory\0";
+pub const GRPC_ARG_GZIP_COMPRESSION_LEVEL: &[u8; 28usize] = b"grpc.gzip_compression_level\0";
+pub const GRPC_ARG_MIN_MESSAGE_SIZE_TO_COMPRESS: &[u8; 39usize] =
+    b"grpc.gprc_min_message_size_to_compress\0";
 pub const GRPC_ARG_MAX_CHANNEL_TRACE_EVENT_MEMORY_PER_NODE: &[u8; 45usize] =
     b"grpc.max_channel_trace_event_memory_per_node\0";
 pub const GRPC_ARG_ENABLE_CHANNELZ: &[u8; 21usize] = b"grpc.enable_channelz\0";

grpc-sys/grpc

@@ -314,6 +314,29 @@ impl ChannelBuilder {
         self
     }
 
+    /// Set default gzip compression level.
+    #[cfg(feature = "nightly")]
+    pub fn default_gzip_compression_level(mut self, level: usize) -> ChannelBuilder {
+        self.options.insert(
+            Cow::Borrowed(grpcio_sys::GRPC_ARG_GZIP_COMPRESSION_LEVEL),
+            Options::Integer(level as i32),
+        );
+        self
+    }
+
+    /// Set default grpc min message size to compression.
+    #[cfg(feature = "nightly")]
+    pub fn default_grpc_min_message_size_to_compress(
+        mut self,
+        lower_bound: usize,
+    ) -> ChannelBuilder {
+        self.options.insert(
+            Cow::Borrowed(grpcio_sys::GRPC_ARG_MIN_MESSAGE_SIZE_TO_COMPRESS),
+            Options::Integer(lower_bound as i32),
+        );
+        self
+    }
+
     /// Set default compression level for the channel.
     pub fn default_compression_level(mut self, level: CompressionLevel) -> ChannelBuilder {
         self.options.insert(
@@ -733,3 +756,26 @@ impl Channel {
         &self.cq
     }
 }
+
+#[cfg(test)]
+#[cfg(feature = "nightly")]
+mod tests {
+    use crate::env::Environment;
+    use crate::ChannelBuilder;
+    use std::sync::Arc;
+
+    #[test]
+    #[cfg(feature = "nightly")]
+    fn test_grpc_min_message_size_to_compress() {
+        let env = Arc::new(Environment::new(1));
+        let cb = ChannelBuilder::new(env);
+        cb.default_grpc_min_message_size_to_compress(1);
+    }
+    #[test]
+    #[cfg(feature = "nightly")]
+    fn test_gzip_compression_level() {
+        let env = Arc::new(Environment::new(1));
+        let cb = ChannelBuilder::new(env);
+        cb.default_gzip_compression_level(1);
+    }
+}