diff options
Diffstat (limited to 'src/catalog.rs')
-rw-r--r-- | src/catalog.rs | 25 |
1 files changed, 14 insertions, 11 deletions
diff --git a/src/catalog.rs b/src/catalog.rs index cad7f5b..042f745 100644 --- a/src/catalog.rs +++ b/src/catalog.rs @@ -22,7 +22,7 @@ use crate::{Consul, WithIndex}; /// Node summary, as specified in response to "list nodes" API calls in /// <https://developer.hashicorp.com/consul/api-docs/catalog#list-nodes> -#[derive(Serialize, Deserialize, Debug, Clone)] +#[derive(Serialize, Deserialize, Debug, Clone, PartialEq, Eq)] #[serde(rename_all = "PascalCase")] pub struct Node { pub node: String, @@ -31,7 +31,7 @@ pub struct Node { } /// One of the services returned in a CatalogNode -#[derive(Serialize, Deserialize, Debug, Clone)] +#[derive(Serialize, Deserialize, Debug, Clone, PartialEq, Eq)] #[serde(rename_all = "PascalCase")] pub struct Service { pub service: String, @@ -42,7 +42,7 @@ pub struct Service { /// Full node info, as specified in response to "retrieve map of services for a node" API call in /// <https://developer.hashicorp.com/consul/api-docs/catalog#retrieve-map-of-services-for-a-node> -#[derive(Serialize, Deserialize, Debug)] +#[derive(Serialize, Deserialize, Debug, PartialEq, Eq)] #[serde(rename_all = "PascalCase")] pub struct CatalogNode { pub node: Node, @@ -55,7 +55,7 @@ pub type ServiceList = HashMap<String, Vec<String>>; /// Node serving a service, as specified in response to "list nodes for a service" API call in /// <https://developer.hashicorp.com/consul/api-docs/catalog#list-nodes-for-service> -#[derive(Serialize, Deserialize, Debug)] +#[derive(Serialize, Deserialize, Debug, PartialEq, Eq)] #[serde(rename_all = "PascalCase")] pub struct ServiceNode { pub node: String, @@ -70,7 +70,7 @@ pub struct ServiceNode { /// Node serving a service with health info, /// as specified in response to "list service instances for a service" health API call in /// <https://developer.hashicorp.com/consul/api-docs/health#list-service-instances-for-service> -#[derive(Serialize, Deserialize, Debug, Clone)] +#[derive(Serialize, Deserialize, Debug, Clone, PartialEq, Eq)] #[serde(rename_all = "PascalCase")] pub struct HealthServiceNode { pub node: Node, @@ -79,7 +79,7 @@ pub struct HealthServiceNode { } /// A health check as returned in HealthServiceNode -#[derive(Serialize, Deserialize, Debug, Clone)] +#[derive(Serialize, Deserialize, Debug, Clone, PartialEq, Eq)] #[serde(rename_all = "PascalCase")] pub struct HealthCheck { pub node: String, @@ -244,7 +244,14 @@ async fn do_watch_all_service_health( match watch_res { Ok(nodes) => { let index = nodes.index(); - services.insert(service.clone(), nodes.into_inner().into()); + + let nodes = nodes.into_inner(); + if services.get(&service).as_ref().map(|n| &n[..]) != Some(&nodes[..]) { + services.insert(service.clone(), nodes.into()); + if tx.send(services.clone()).is_err() { + break; + } + } let consul = &consul; service_watchers.push(Box::pin(async move { @@ -252,10 +259,6 @@ async fn do_watch_all_service_health( .map_err(|e| (1, e)); (service, res) })); - - if tx.send(services.clone()).is_err() { - break; - } } Err((err_count, e)) => { warn!("Error getting service {}: {} ({} consecutive errors)", service, e, err_count); |