feat: first draft for gzip compresison

Author: Miaxos

CHANGELOG.md

@@ -6,6 +6,9 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
 ## [Unreleased]
+### Feat
+
+- GZIP Compression
 
 ## [0.3.6]
 

Cargo.toml

@@ -11,10 +11,13 @@ keywords = ["async_graphql", "async", "graphql", "apollo", "studio"]
 categories = ["network-programming", "asynchronous"]
 edition = "2018"
 
+[features]
+default = []
+compression = ["libflate"]
+
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 [dependencies]
 protobuf = { version = "2.18.1", features = ["bytes"] }
-libflate = "1"
 async-graphql = {version = "2.9.*", features = ["tracing"] }
 tracing = "0.1.*"
 tracing-futures = { version = "0.2.5", default-features = false, features = ["tokio", "futures-03", "std"] }
@@ -27,3 +30,6 @@ serde_json = "1.0.*"                     # A JSON serialization file format
 sha2 = "0.9.*"                # Pure Rust implementation of the SHA-2 hash function family including SHA-224, SHA-256, SHA-384, a…
 anyhow = "1.0.*"                # Flexible concrete Error type built on std::error::Error
 uuid = { version = "0.8.*", features = ["v4"] }                  # A library to generate and parse UUIDs.
+
+# Non-feature optional dependencies
+libflate = { version = "1", optional = true }

src/lib.rs

@@ -203,13 +203,40 @@ impl ApolloTracing {
                         }
                     };
 
-                    let result = client
-                        .post(REPORTING_URL)
-                        .body(msg)
-                        .header("content-type", "application/protobuf")
-                        .header("X-Api-Key", &authorization_token)
-                        .send()
-                        .await;
+                    let result = if cfg!(features = "compression") {
+                        let mut encoder = libflate::gzip::Encoder::new(Vec::new()).unwrap();
+                        let mut msg = std::io::Cursor::new(msg);
+                        match std::io::copy(&mut msg, &mut encoder) {
+                            Ok(_) => {}
+                            Err(e) => {
+                                error!(target: TARGET_LOG, message = "An issue happened while GZIP compression", err = ?e);
+                                continue;
+                            }
+                        };
+                        match encoder.finish().into_result() {
+                            Ok(msg) => {
+                                client
+                                    .post(REPORTING_URL)
+                                    .body(msg)
+                                    .header("content-type", "application/protobuf")
+                                    .header("X-Api-Key", &authorization_token)
+                                    .send()
+                                    .await
+                            }
+                            Err(e) => {
+                                error!(target: TARGET_LOG, message = "An issue happened while GZIP compression", err = ?e);
+                                continue;
+                            }
+                        }
+                    } else {
+                        client
+                            .post(REPORTING_URL)
+                            .body(msg)
+                            .header("content-type", "application/protobuf")
+                            .header("X-Api-Key", &authorization_token)
+                            .send()
+                            .await
+                    };
 
                     match result {
                         Ok(data) => {