generalized interface

pendulum-project · Dec 8, 2023 · 6de1c86 · 6de1c86
1 parent 60eb995
commit 6de1c86
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 7 deletions.
diff --git a/ntp-proto/src/nts_record.rs b/ntp-proto/src/nts_record.rs
@@ -1181,7 +1181,7 @@ impl KeyExchangeClient {
     pub fn new(
         server_name: String,
         tls_config: rustls::ClientConfig,
-        denied_servers: Vec<String>,
+        denied_servers: impl IntoIterator<Item = String>,
     ) -> Result<Self, KeyExchangeError> {
         let mut client = Self::new_without_tls_write(server_name, tls_config)?;
 

diff --git a/ntpd/src/daemon/keyexchange.rs b/ntpd/src/daemon/keyexchange.rs
@@ -56,7 +56,7 @@ pub(crate) async fn key_exchange_client_with_denied_servers(
     server_name: String,
     port: u16,
     extra_certificates: &[Certificate],
-    denied_servers: Vec<String>,
+    denied_servers: impl IntoIterator<Item = String>,
 ) -> Result<KeyExchangeResult, KeyExchangeError> {
     let socket = tokio::net::TcpStream::connect((server_name.as_str(), port)).await?;
     let config = build_client_config(extra_certificates)?;
@@ -218,7 +218,7 @@ where
         io: IO,
         server_name: String,
         config: rustls::ClientConfig,
-        denied_servers: Vec<String>,
+        denied_servers: impl IntoIterator<Item = String>,
     ) -> Result<Self, KeyExchangeError> {
         Ok(Self {
             inner: Some(BoundKeyExchangeClientData {

diff --git a/ntpd/src/daemon/spawn/nts_pool.rs b/ntpd/src/daemon/spawn/nts_pool.rs
@@ -96,10 +96,7 @@ impl NtsPoolSpawner {
                     self.config.addr.server_name.clone(),
                     self.config.addr.port,
                     &self.config.certificate_authorities,
-                    self.current_peers
-                        .iter()
-                        .map(|peer| peer.remote.clone())
-                        .collect(),
+                    self.current_peers.iter().map(|peer| peer.remote.clone()),
                 )
                 .await
                 {