Simplified Endpoint::new initialization

hyperium · Aug 20, 2024 · de5cec8 · de5cec8
1 parent 55bb186
commit de5cec8
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 15 deletions.
diff --git a/tonic/src/transport/channel/endpoint.rs b/tonic/src/transport/channel/endpoint.rs
@@ -50,21 +50,8 @@ impl Endpoint {
     {
         let me = dst.try_into().map_err(|e| Error::from_source(e.into()))?;
         #[cfg(feature = "tls")]
-        if let Some(tls_config) = me
-            .uri
-            .scheme()
-            .map(|s| s.as_str() == http::uri::Scheme::HTTPS.as_str())
-            .unwrap_or(false)
-            .then(|| {
-                let config = ClientTlsConfig::new();
-                #[cfg(feature = "tls-native-roots")]
-                let config = config.with_native_roots();
-                #[cfg(feature = "tls-webpki-roots")]
-                let config = config.with_webpki_roots();
-                config
-            })
-        {
-            return me.tls_config(tls_config);
+        if me.uri.scheme() == Some(&http::uri::Scheme::HTTPS) {
+            return me.tls_config(ClientTlsConfig::new().with_enabled_roots());
         }
 
         Ok(me)

diff --git a/tonic/src/transport/channel/tls.rs b/tonic/src/transport/channel/tls.rs
@@ -81,6 +81,16 @@ impl ClientTlsConfig {
         }
     }
 
+    /// Activates all TLS roots enabled through `tls-*-roots` feature flags
+    pub fn with_enabled_roots(self) -> Self {
+        let config = ClientTlsConfig::new();
+        #[cfg(feature = "tls-native-roots")]
+        let config = config.with_native_roots();
+        #[cfg(feature = "tls-webpki-roots")]
+        let config = config.with_webpki_roots();
+        config
+    }
+
     pub(crate) fn into_tls_connector(self, uri: &Uri) -> Result<TlsConnector, crate::Error> {
         let domain = match &self.domain {
             Some(domain) => domain,