feat: added gzip auto-decoding

saskenuba · saskenuba · commit 2d974698ac0c · 2022-03-08T05:27:30.000-03:00
Signed-off-by: Martin &lt;martin@hotmail.com.br&gt;
diff --git a/Cargo.toml b/Cargo.toml
@@ -14,7 +14,7 @@ actix-web = { version = "4", default-features = false }
 
 futures-util = "0.3"
 log = "0.4"
-reqwest = { version = "0.11", features = ["tokio-rustls"] }
+reqwest = { version = "0.11", features = ["tokio-rustls", "gzip"] }
 thiserror = "1"
 url = "2"
 
diff --git a/src/builder.rs b/src/builder.rs
@@ -51,13 +51,21 @@ pub struct PrerenderBuilder {
     pub(crate) before_render_fn: Option<fn(&ServiceRequest, &mut HeaderMap)>,
 }
 
+fn default_client() -> Client {
+    Client::builder()
+        .gzip(true)
+        .timeout(std::time::Duration::new(25, 0))
+        .build()
+        .unwrap()
+}
+
 impl PrerenderBuilder {
     /// Creates a `Prerender` middleware that delegate requests to the web `prerender.io` service.
     pub fn use_prerender_io(self, token: String) -> Prerender {
         let inner = Inner {
             before_render_fn: self.before_render_fn,
             forward_headers: self.forward_headers,
-            inner_client: Client::default(),
+            inner_client: default_client(),
             prerender_service_url: middleware::prerender_url(),
             prerender_token: Some(token),
         };
@@ -72,7 +80,7 @@ impl PrerenderBuilder {
         let inner = Inner {
             before_render_fn: self.before_render_fn,
             forward_headers: self.forward_headers,
-            inner_client: Client::default(),
+            inner_client: default_client(),
             prerender_service_url,
             prerender_token: None,
         };
diff --git a/src/middleware.rs b/src/middleware.rs
@@ -3,6 +3,7 @@ use std::rc::Rc;
 use actix_service::Service;
 use actix_web::body::{EitherBody, MessageBody};
 use actix_web::dev::{ServiceRequest, ServiceResponse};
+use actix_web::http::header::ContentEncoding;
 use actix_web::http::uri::PathAndQuery;
 use actix_web::http::{header, Method};
 use actix_web::{Error, HttpResponse};
@@ -119,13 +120,13 @@ impl<S> PrerenderMiddleware<S> {
             prerender_headers.remove(header::HOST);
         }
 
-        prerender_headers.append(header::ACCEPT_ENCODING, "gzip".parse().unwrap());
+        prerender_headers.insert(header::ACCEPT_ENCODING, ContentEncoding::Gzip.to_header_value());
         if let Some(token) = &inner.prerender_token {
             prerender_headers.append("X-Prerender-Token", token.parse().unwrap());
         }
 
-        if let Some(function) = &inner.before_render_fn {
-            function(&req, &mut prerender_headers);
+        if let Some(before_render_fn) = &inner.before_render_fn {
+            before_render_fn(&req, &mut prerender_headers);
         }
 
         let url_to_request = Self::prepare_build_api_url(&inner.prerender_service_url, &req);
