Add example for Pod attach

examples/Cargo.toml

@@ -101,6 +101,10 @@ path = "multi_watcher.rs"
 name = "pod_api"
 path = "pod_api.rs"
 
+[[example]]
+name = "pod_attach"
+path = "pod_attach.rs"
+
 [[example]]
 name = "proxy"
 path = "proxy.rs"

examples/pod_attach.rs

@@ -0,0 +1,75 @@
+#[macro_use]
+extern crate log;
+
+use futures::{StreamExt, TryStreamExt};
+use k8s_openapi::api::core::v1::Pod;
+
+use kube::{
+    api::{Api, AttachParams, DeleteParams, ListParams, Meta, PostParams, WatchEvent},
+    Client,
+};
+
+#[tokio::main]
+async fn main() -> anyhow::Result<()> {
+    std::env::set_var("RUST_LOG", "info,kube=debug");
+    env_logger::init();
+    let client = Client::try_default().await?;
+    let namespace = std::env::var("NAMESPACE").unwrap_or_else(|_| "default".into());
+
+    info!("Creating a Pod that outputs numbers for 15s");
+    let p: Pod = serde_json::from_value(serde_json::json!({
+        "apiVersion": "v1",
+        "kind": "Pod",
+        "metadata": { "name": "example" },
+        "spec": {
+            "containers": [{
+                "name": "example",
+                "image": "alpine",
+                "command": ["sh", "-c", "for i in `seq 15`; do echo \"$i\"; sleep 1; done"]
+            }],
+        }
+    }))?;
+
+    let pods: Api<Pod> = Api::namespaced(client, &namespace);
+    // Stop on error including a pod already exists or is still being deleted.
+    pods.create(&PostParams::default(), &p).await?;
+
+    // Wait until the pod is running, otherwise we get 500 error.
+    let lp = ListParams::default().fields("metadata.name=example").timeout(10);
+    let mut stream = pods.watch(&lp, "0").await?.boxed();
+    while let Some(status) = stream.try_next().await? {
+        match status {
+            WatchEvent::Added(o) => {
+                info!("Added {}", Meta::name(&o));
+            }
+            WatchEvent::Modified(o) => {
+                let s = o.status.as_ref().expect("status exists on pod");
+                if s.phase.clone().unwrap_or_default() == "Running" {
+                    info!("Ready to attach to {}", Meta::name(&o));
+                    break;
+                }
+            }
+            _ => {}
+        }
+    }
+
+    // Type inference doesn't work :(
+    let ap = AttachParams {
+        container: None,
+        stdout: Some(tokio::io::stdout()),
+        stdin: Some(tokio::io::empty()),
+        stderr: None,
+        tty: false,
+    };
+    // Attach to see numbers printed on stdout.
+    pods.attach("example", ap).await?;
+
+    // Delete it
+    pods.delete("example", &DeleteParams::default())
+        .await?
+        .map_left(|pdel| {
+            assert_eq!(Meta::name(&pdel), "example");
+        });
+
+    Ok(())
+}