Inventory: Implement display for CockroachStatus

This is used by omdb, reconfigurator-cli

Author: smklein

nexus/types/src/inventory.rs

@@ -618,7 +618,9 @@ impl IdOrdItem for SledAgent {
     id_upcast!();
 }
 
-#[derive(Clone, Default, Debug, PartialEq, Eq, Deserialize, Serialize)]
+#[derive(
+    Clone, Default, Debug, Hash, PartialEq, Eq, Deserialize, Serialize,
+)]
 pub struct CockroachStatus {
     pub ranges_underreplicated: Option<u64>,
     pub liveness_live_nodes: Option<u64>,

nexus/types/src/inventory/display.rs

@@ -26,6 +26,7 @@ use nexus_sled_agent_shared::inventory::{
 use omicron_uuid_kinds::{
     DatasetUuid, OmicronZoneUuid, PhysicalDiskUuid, ZpoolUuid,
 };
+use std::collections::HashMap;
 use strum::IntoEnumIterator;
 use tabled::Tabled;
 use tufaceous_artifact::ArtifactHash;
@@ -42,6 +43,7 @@ pub struct CollectionDisplay<'a> {
     include_sleds: bool,
     include_orphaned_datasets: bool,
     include_clickhouse_keeper_membership: bool,
+    include_cockroach_status: bool,
     long_string_formatter: LongStringFormatter,
 }
 
@@ -54,6 +56,7 @@ impl<'a> CollectionDisplay<'a> {
             include_sleds: true,
             include_orphaned_datasets: true,
             include_clickhouse_keeper_membership: true,
+            include_cockroach_status: true,
             long_string_formatter: LongStringFormatter::new(),
         }
     }
@@ -94,6 +97,15 @@ impl<'a> CollectionDisplay<'a> {
         self
     }
 
+    /// Control display of Cockroach cluster information (defaults to true).
+    pub fn include_cockroach_status(
+        &mut self,
+        include_cockroach_status: bool,
+    ) -> &mut Self {
+        self.include_cockroach_status = include_cockroach_status;
+        self
+    }
+
     /// Show long strings (defaults to false).
     pub fn show_long_strings(&mut self, show_long_strings: bool) -> &mut Self {
         self.long_string_formatter.show_long_strings = show_long_strings;
@@ -111,6 +123,7 @@ impl<'a> CollectionDisplay<'a> {
             .include_clickhouse_keeper_membership(
                 filter.include_keeper_membership(),
             )
+            .include_cockroach_status(filter.include_cockroach_status())
     }
 }
 
@@ -134,6 +147,9 @@ impl fmt::Display for CollectionDisplay<'_> {
         if self.include_clickhouse_keeper_membership {
             display_keeper_membership(&self.collection, f)?;
         }
+        if self.include_cockroach_status {
+            display_cockroach_status(&self.collection, f)?;
+        }
 
         if nerrors > 0 {
             writeln!(
@@ -199,6 +215,14 @@ impl CollectionDisplayCliFilter {
             Self::OrphanedDatasets => false,
         }
     }
+
+    fn include_cockroach_status(&self) -> bool {
+        match self {
+            Self::All => true,
+            Self::Sp { .. } => false,
+            Self::OrphanedDatasets => false,
+        }
+    }
 }
 
 /// Which SPs within a collection to display.
@@ -1002,6 +1026,56 @@ fn display_keeper_membership(
     Ok(())
 }
 
+fn display_cockroach_status(
+    collection: &Collection,
+    f: &mut dyn fmt::Write,
+) -> fmt::Result {
+    writeln!(f, "\nCOCKROACH STATUS")?;
+
+    // Under normal conditions, cockroach nodes will report the same data. For
+    // brevity, we will map "status" -> "nodes reporting that status", to avoid
+    // emitting the same information repeatedly for each node.
+    let mut status_to_node = HashMap::new();
+
+    for (node, status) in &collection.cockroach_status {
+        status_to_node
+            .entry(status)
+            .and_modify(|nodes: &mut Vec<_>| nodes.push(node))
+            .or_insert(vec![node]);
+    }
+
+    for (status, nodes) in &status_to_node {
+        writeln!(
+            f,
+            "\n  status from nodes: {}",
+            nodes.iter().map(|n| n.to_string()).join(", ")
+        )?;
+
+        writeln!(
+            f,
+            " \n    ranges underreplicated: {}",
+            status
+                .ranges_underreplicated
+                .map(|r| r.to_string())
+                .unwrap_or_else(|| "-".to_string())
+        )?;
+        writeln!(
+            f,
+            " \n    live nodes: {}",
+            status
+                .liveness_live_nodes
+                .map(|r| r.to_string())
+                .unwrap_or_else(|| "-".to_string())
+        )?;
+    }
+    if status_to_node.is_empty() {
+        writeln!(f, "    no cockroach status retrieved")?;
+    }
+    writeln!(f, "")?;
+
+    Ok(())
+}
+
 #[derive(Debug)]
 struct LongStringFormatter {
     show_long_strings: bool,