#[macro_use] extern crate log;
use kube::{
api::{RawApi, Informer, WatchEvent, ObjectList, Object},
client::APIClient,
config,
};
use k8s_openapi::api::core::v1::{
NodeSpec, NodeStatus,
Event, ListEventForAllNamespacesOptional,
};
type Node = Object<NodeSpec, NodeStatus>;
fn main() -> Result<(), failure::Error> {
std::env::set_var("RUST_LOG", "info,kube=trace");
env_logger::init();
let config = config::load_kube_config().expect("failed to load kubeconfig");
let client = APIClient::new(config);
let nodes = RawApi::v1Node();
let ni = Informer::raw(client.clone(), nodes)
.labels("role=worker")
.init()?;
loop {
ni.poll()?;
while let Some(event) = ni.pop() {
handle_nodes(&client, event)?;
}
}
}
fn handle_nodes(client: &APIClient, ev: WatchEvent<Node>) -> Result<(), failure::Error> {
match ev {
WatchEvent::Added(o) => {
info!("New Node: {}", o.spec.provider_id.unwrap());
},
WatchEvent::Modified(o) => {
if let Some(true) = o.spec.unschedulable {
let failed = o.status.unwrap().conditions.unwrap().into_iter().filter(|c| {
(c.status == "True" && c.type_ != "Ready") ||
(c.status == "False" && c.type_ == "Ready")
}).map(|c| c.message).collect::<Vec<_>>(); warn!("Unschedulable Node: {}, ({:?})", o.metadata.name, failed);
let sel = format!("involvedObject.kind=Node,involvedObject.name={}", o.metadata.name);
let opts = ListEventForAllNamespacesOptional {
field_selector: Some(&sel),
..Default::default()
};
let req = Event::list_event_for_all_namespaces(opts)?.0;
let res = client.request::<ObjectList<Event>>(req)?;
for e in res.items {
warn!("Node event: {:?}", e);
}
} else {
debug!("Normal node: {}", o.metadata.name);
}
},
WatchEvent::Deleted(o) => {
warn!("Deleted node: {} ({:?}) running {:?} with labels: {:?}",
o.metadata.name, o.spec.provider_id.unwrap(),
o.status.unwrap().conditions.unwrap(),
o.metadata.labels,
);
},
WatchEvent::Error(e) => {
warn!("Error event: {:?}", e);
}
}
Ok(())
}