use etcd::Client as EtcdClient;
use etcd::kv::KeyValueInfo;
use etcd::{kv, Response};
use tokio_core::reactor::Core;
pub fn get_worker_list(etcd_address: &str) -> Result<Vec<String>, String> {
let mut core = Core::new().unwrap();
let handle = core.handle();
let endpoints = &[etcd_address];
match EtcdClient::new(&handle, endpoints, None) {
Ok(etcd) => match core.run(kv::get(
&etcd,
"/datafusion/workers/",
kv::GetOptions::default(),
)) {
Ok(Response { ref data, .. }) => match data {
&KeyValueInfo { ref node, .. } => match &node.nodes {
&Some(ref workers) => {
Ok(workers.iter().flat_map(|w| w.value.clone()).collect())
}
_ => Ok(vec![]),
},
},
Err(e) => Err(format!("Etcd request failed ({:?}): {:?}", endpoints, e)),
},
Err(e) => Err(format!("Etcd connection failed ({:?}): {:?}", endpoints, e)),
}
}