summaryrefslogtreecommitdiff
path: root/src/catalog.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/catalog.rs')
-rw-r--r--src/catalog.rs17
1 files changed, 16 insertions, 1 deletions
diff --git a/src/catalog.rs b/src/catalog.rs
index fe82c5a..87c7775 100644
--- a/src/catalog.rs
+++ b/src/catalog.rs
@@ -13,7 +13,7 @@ use futures::future::BoxFuture;
use futures::stream::futures_unordered::FuturesUnordered;
use futures::{FutureExt, StreamExt, TryFutureExt};
use log::*;
-use serde::{Deserialize, Serialize};
+use serde::{Deserialize, Deserializer, Serialize};
use tokio::select;
use tokio::sync::watch;
@@ -26,6 +26,7 @@ use crate::{Consul, WithIndex};
pub struct Node {
pub node: String,
pub address: String,
+ #[serde(default, deserialize_with = "deserialize_null_default")]
pub meta: HashMap<String, String>,
}
@@ -37,6 +38,8 @@ pub struct Service {
pub address: String,
pub port: u16,
pub tags: Vec<String>,
+ #[serde(default, deserialize_with = "deserialize_null_default")]
+ pub meta: HashMap<String, String>,
}
/// Full node info, as specified in response to "retrieve map of services for a node" API call in
@@ -45,6 +48,7 @@ pub struct Service {
#[serde(rename_all = "PascalCase")]
pub struct CatalogNode {
pub node: Node,
+ #[serde(default, deserialize_with = "deserialize_null_default")]
pub services: HashMap<String, Service>,
}
@@ -59,11 +63,14 @@ pub type ServiceList = HashMap<String, Vec<String>>;
pub struct ServiceNode {
pub node: String,
pub address: String,
+ #[serde(default, deserialize_with = "deserialize_null_default")]
pub node_meta: HashMap<String, String>,
pub service_name: String,
pub service_tags: Vec<String>,
pub service_address: String,
pub service_port: u16,
+ #[serde(default, deserialize_with = "deserialize_null_default")]
+ pub service_meta: HashMap<String, String>,
}
/// Node serving a service with health info,
@@ -273,3 +280,11 @@ fn retry_to_time(retries: usize, max_time: Duration) -> Duration {
Duration::from_secs_f64(2.0f64 * 1.5f64.powf(retries as f64)),
)
}
+
+fn deserialize_null_default<'de, D, T>(deserializer: D) -> Result<T, D::Error>
+where
+ T: Default + Deserialize<'de>,
+ D: Deserializer<'de>,
+{
+ Option::deserialize(deserializer).map(Option::unwrap_or_default)
+}