forked from Deuxfleurs/garage
69 lines
1.6 KiB
Rust
69 lines
1.6 KiB
Rust
use std::collections::HashMap;
|
|
use std::net::{IpAddr, SocketAddr};
|
|
|
|
use hyper::client::Client;
|
|
use hyper::StatusCode;
|
|
use hyper::{Body, Method, Request};
|
|
use serde::Deserialize;
|
|
|
|
use netapp::NodeID;
|
|
|
|
use garage_util::error::Error;
|
|
|
|
#[derive(Deserialize, Clone, Debug)]
|
|
struct ConsulEntry {
|
|
#[serde(alias = "Address")]
|
|
address: String,
|
|
#[serde(alias = "ServicePort")]
|
|
service_port: u16,
|
|
#[serde(alias = "NodeMeta")]
|
|
node_meta: HashMap<String, String>,
|
|
}
|
|
|
|
pub async fn get_consul_nodes(
|
|
consul_host: &str,
|
|
consul_service_name: &str,
|
|
) -> Result<Vec<(NodeID, SocketAddr)>, Error> {
|
|
let url = format!(
|
|
"http://{}/v1/catalog/service/{}",
|
|
consul_host, consul_service_name
|
|
);
|
|
let req = Request::builder()
|
|
.uri(url)
|
|
.method(Method::GET)
|
|
.body(Body::default())?;
|
|
|
|
let client = Client::new();
|
|
|
|
let resp = client.request(req).await?;
|
|
if resp.status() != StatusCode::OK {
|
|
return Err(Error::Message(format!("HTTP error {}", resp.status())));
|
|
}
|
|
|
|
let body = hyper::body::to_bytes(resp.into_body()).await?;
|
|
let entries = serde_json::from_slice::<Vec<ConsulEntry>>(body.as_ref())?;
|
|
|
|
let mut ret = vec![];
|
|
for ent in entries {
|
|
let ip = ent.address.parse::<IpAddr>().ok();
|
|
let pubkey = ent
|
|
.node_meta
|
|
.get("pubkey")
|
|
.map(|k| hex::decode(&k).ok())
|
|
.flatten()
|
|
.map(|k| NodeID::from_slice(&k[..]))
|
|
.flatten();
|
|
if let (Some(ip), Some(pubkey)) = (ip, pubkey) {
|
|
ret.push((pubkey, SocketAddr::new(ip, ent.service_port)));
|
|
} else {
|
|
warn!(
|
|
"Could not process node spec from Consul: {:?} (invalid IP or public key)",
|
|
ent
|
|
);
|
|
}
|
|
}
|
|
debug!("Got nodes from Consul: {:?}", ret);
|
|
|
|
Ok(ret)
|
|
}
|