diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/config/endpoint_config.rs | 4 | ||||
-rw-r--r-- | src/endpoint/configured.rs | 9 |
2 files changed, 11 insertions, 2 deletions
diff --git a/src/config/endpoint_config.rs b/src/config/endpoint_config.rs index 69405dc..05d420f 100644 --- a/src/config/endpoint_config.rs +++ b/src/config/endpoint_config.rs @@ -50,6 +50,10 @@ pub struct Endpoint { #[getset(get = "pub")] network_mode: Option<String>, + + /// Duration length of timeout for connecting endpoint + #[getset(get = "pub")] + timeout: Option<u64>, } /// The type of an endpoint diff --git a/src/endpoint/configured.rs b/src/endpoint/configured.rs index f850ff6..8fdec64 100644 --- a/src/endpoint/configured.rs +++ b/src/endpoint/configured.rs @@ -84,8 +84,13 @@ impl Endpoint { Endpoint::check_api_version_compat(epc.required_docker_api_versions().as_ref(), &ep); let imgs_avail = Endpoint::check_images_available(epc.required_images().as_ref(), &ep); - let (versions_compat, api_versions_compat, imgs_avail) = - tokio::join!(versions_compat, api_versions_compat, imgs_avail); + let (versions_compat, api_versions_compat, imgs_avail) = { + let timeout = std::time::Duration::from_secs(epc.endpoint().timeout().unwrap_or(10)); + let versions_compat = tokio::time::timeout(timeout, versions_compat); + let api_versions_compat = tokio::time::timeout(timeout, api_versions_compat); + let imgs_avail = tokio::time::timeout(timeout, imgs_avail); + tokio::join!(versions_compat, api_versions_compat, imgs_avail) + }; let _ = versions_compat.with_context(|| { anyhow!( |