status tool

rust/load/src/bin/chroma-load-status.rs

@@ -0,0 +1,35 @@
+//! Inspect chroma-load
+
+use clap::Parser;
+
+#[derive(Parser, Debug)]
+struct Args {
+    #[arg(long)]
+    host: String,
+}
+
+#[tokio::main]
+async fn main() {
+    let args = Args::parse();
+    let client = reqwest::Client::new();
+    match client
+        .get(&args.host)
+        .header(reqwest::header::ACCEPT, "application/json")
+        .send()
+        .await
+    {
+        Ok(resp) => {
+            if resp.status().is_success() {
+                let status = resp.json::<chroma_load::rest::Status>().await;
+                println!("{:#?}", status);
+            } else {
+                eprintln!(
+                    "Failed to get workload status on {}: {}",
+                    args.host,
+                    resp.status()
+                );
+            }
+        }
+        Err(e) => eprintln!("Failed to get workload status on {}: {}", args.host, e),
+    }
+}

rust/load/src/rest.rs

@@ -23,7 +23,7 @@ impl From<&dyn crate::DataSet> for Description {
 pub struct Status {
     pub running: Vec<WorkloadSummary>,
     pub data_sets: Vec<Description>,
-    pub workloads: Vec<Description>,
+    pub workloads: Vec<serde_json::Value>,
 }
 
 #[derive(Clone, Debug, serde::Deserialize, serde::Serialize)]