feat: add pausing and unpausing states

Signed-off-by: Rustin170506 <[email protected]>

Author: 0xPoe

tokio-console/src/main.rs

@@ -1,6 +1,6 @@
 use color_eyre::{eyre::eyre, Help, SectionExt};
 use console_api::tasks::TaskDetails;
-use state::State;
+use state::{State, Temporality};
 
 use futures::stream::StreamExt;
 use ratatui::{
@@ -95,8 +95,10 @@ async fn main() -> color_eyre::Result<()> {
                 if input::is_space(&input) {
                     if state.is_paused() {
                         conn.resume().await;
+                        state.start_unpausing();
                     } else {
                         conn.pause().await;
+                        state.start_pausing();
                     }
                 }
 
@@ -153,8 +155,17 @@ async fn main() -> color_eyre::Result<()> {
                 .split(f.size());
 
             let mut header_text = conn.render(&view.styles);
-            if state.is_paused() {
-                header_text.push_span(Span::styled(" PAUSED", view.styles.fg(Color::Red)));
+            match state.temporality() {
+                Temporality::Paused => {
+                    header_text.push_span(Span::styled(" PAUSED", view.styles.fg(Color::Red)));
+                }
+                Temporality::Pausing => {
+                    header_text.push_span(Span::styled(" PAUSING", view.styles.fg(Color::Yellow)));
+                }
+                Temporality::Unpausing => {
+                    header_text.push_span(Span::styled(" UNPAUSING", view.styles.fg(Color::Green)));
+                }
+                Temporality::Live => {}
             }
             let dropped_async_ops_state = state.async_ops_state().dropped_events();
             let dropped_tasks_state = state.tasks_state().dropped_events();

tokio-console/src/state/mod.rs

@@ -34,7 +34,7 @@ pub(crate) type DetailsRef = Rc<RefCell<Option<Details>>>;
 pub(crate) struct State {
     metas: HashMap<u64, Metadata>,
     last_updated_at: Option<SystemTime>,
-    temporality: proto::instrument::Temporality,
+    temporality: Temporality,
     tasks_state: TasksState,
     resources_state: ResourcesState,
     async_ops_state: AsyncOpsState,
@@ -71,6 +71,29 @@ pub(crate) enum FieldValue {
     Debug(String),
 }
 
+#[derive(Debug)]
+pub(crate) enum Temporality {
+    Unpausing,
+    Live,
+    Pausing,
+    Paused,
+}
+
+impl Default for Temporality {
+    fn default() -> Self {
+        Self::Live
+    }
+}
+
+impl From<proto::instrument::Temporality> for Temporality {
+    fn from(pb: proto::instrument::Temporality) -> Self {
+        match pb {
+            proto::instrument::Temporality::Live => Self::Live,
+            proto::instrument::Temporality::Paused => Self::Paused,
+        }
+    }
+}
+
 #[derive(Debug, Eq, PartialEq)]
 pub(crate) struct Attribute {
     field: Field,
@@ -230,13 +253,26 @@ impl State {
     }
 
     // temporality methods
+    pub(crate) fn temporality(&self) -> &Temporality {
+        &self.temporality
+    }
+
+    pub(crate) fn start_unpausing(&mut self) {
+        self.temporality = Temporality::Unpausing;
+    }
+
+    pub(crate) fn start_pausing(&mut self) {
+        self.temporality = Temporality::Pausing;
+    }
 
     pub(crate) fn update_state(&mut self, state: proto::instrument::State) {
-        self.temporality = proto::instrument::Temporality::try_from(state.temporality).unwrap();
+        self.temporality = proto::instrument::Temporality::try_from(state.temporality)
+            .expect("invalid temporality")
+            .into();
     }
 
     pub(crate) fn is_paused(&self) -> bool {
-        matches!(self.temporality, proto::instrument::Temporality::Paused)
+        matches!(self.temporality, Temporality::Paused | Temporality::Pausing)
     }
 }