Merge branch 'dfinity:dynamic_route' into dynamic_route

Author: nikolay-komarevskiy

ic-agent/Cargo.toml

@@ -74,8 +74,14 @@ hyper-rustls = { version = "0.27", default-features = false, features = [
     "http1",
     "http2",
 ], optional = true }
-tokio = { version = "1.24.2", features = ["time"] }
+tokio = { version = "1.24.2", features = ["macros", "time"] }
 tower = { version = "0.4.13", optional = true }
+async-trait = "^0.1.0"
+tracing = "^0.1.0"
+arc-swap = "^1.0.0"
+simple_moving_average = "^1.0.0"
+tracing-subscriber = "^0.2.0"
+tokio-util = { version = "^0.7.0", features = ["rt"] }
 rustls-webpki = "0.102"
 
 [target.'cfg(target_family = "wasm")'.dependencies]

ic-agent/src/agent/builder.rs

@@ -16,6 +16,38 @@ impl AgentBuilder {
         Agent::new(self.config)
     }
 
+    #[cfg(all(feature = "reqwest", not(target_family = "wasm")))]
+    /// Set the dynamic transport layer for the [`Agent`], performing continuos discovery of the API boundary nodes and routing traffic via them based on the latencies.
+    pub async fn with_discovery_transport(self, client: reqwest::Client) -> Self {
+        use crate::agent::http_transport::{
+            dynamic_routing::{
+                dynamic_route_provider::{DynamicRouteProviderBuilder, IC0_SEED_DOMAIN},
+                node::Node,
+                snapshot::latency_based_routing::LatencyRoutingSnapshot,
+            },
+            route_provider::RouteProvider,
+            ReqwestTransport,
+        };
+
+        // TODO: This is a temporary solution to get the seed node.
+        let seed = Node::new(IC0_SEED_DOMAIN).unwrap();
+
+        let route_provider = DynamicRouteProviderBuilder::new(
+            LatencyRoutingSnapshot::new(),
+            vec![seed],
+            client.clone(),
+        )
+        .build()
+        .await;
+
+        let route_provider = Arc::new(route_provider) as Arc<dyn RouteProvider>;
+
+        let transport = ReqwestTransport::create_with_client_route(route_provider, client)
+            .expect("failed to create transport");
+
+        self.with_transport(transport)
+    }
+
     /// Set the URL of the [Agent].
     #[cfg(feature = "reqwest")]
     pub fn with_url<S: Into<String>>(self, url: S) -> Self {