fix(transport): Apply tls-connector for discovery when applicable (#334) (#338)

2020-05-05 13:19:04 -04:00
parent 59c4555f54
commit 99fbe22e7c
1 changed files with 9 additions and 1 deletions
@@ -1,3 +1,4 @@
+use super::super::service;
 use super::connection::Connection;
 use crate::transport::Endpoint;
 use std::{
@@ -52,8 +53,15 @@ impl Discover for ServiceList {
                let mut http = hyper::client::connect::HttpConnector::new();
                http.set_nodelay(endpoint.tcp_nodelay);
                http.set_keepalive(endpoint.tcp_keepalive);
+                http.enforce_http(false);

-                let fut = Connection::new(http, endpoint);
+                #[cfg(feature = "tls")]
+                let connector = service::connector(http, endpoint.tls.clone());
+
+                #[cfg(not(feature = "tls"))]
+                let connector = service::connector(http);
+
+                let fut = Connection::new(connector, endpoint);
                self.connecting = Some(Box::pin(fut));
            } else {
                return Poll::Pending;