1 files changed, 96 insertions, 0 deletions
diff --git a/core/ops_metrics.rs b/core/ops_metrics.rs
new file mode 100644
index 000000000..9dd5e2edf
--- /dev/null
+++ b/core/ops_metrics.rs
@@ -0,0 +1,96 @@
+// Copyright 2018-2021 the Deno authors. All rights reserved. MIT license.
+use crate::serde::Serialize;
+use crate::OpId;
+
+// TODO(@AaronO): split into AggregateMetrics & PerOpMetrics
+#[derive(Clone, Default, Debug, Serialize)]
+#[serde(rename_all = "camelCase")]
+pub struct OpMetrics {
+  pub ops_dispatched: u64,
+  pub ops_dispatched_sync: u64,
+  pub ops_dispatched_async: u64,
+  pub ops_dispatched_async_unref: u64,
+  pub ops_completed: u64,
+  pub ops_completed_sync: u64,
+  pub ops_completed_async: u64,
+  pub ops_completed_async_unref: u64,
+  pub bytes_sent_control: u64,
+  pub bytes_sent_data: u64,
+  pub bytes_received: u64,
+}
+
+// TODO(@AaronO): track errors
+#[derive(Default, Debug)]
+pub struct OpsTracker {
+  pub ops: Vec<OpMetrics>,
+}
+
+impl OpsTracker {
+  pub fn per_op(&self) -> Vec<OpMetrics> {
+    self.ops.clone()
+  }
+
+  pub fn aggregate(&self) -> OpMetrics {
+    let mut sum = OpMetrics::default();
+
+    for metrics in self.ops.iter() {
+      sum.ops_dispatched += metrics.ops_dispatched;
+      sum.ops_dispatched_sync += metrics.ops_dispatched_sync;
+      sum.ops_dispatched_async += metrics.ops_dispatched_async;
+      sum.ops_dispatched_async_unref += metrics.ops_dispatched_async_unref;
+      sum.ops_completed += metrics.ops_completed;
+      sum.ops_completed_sync += metrics.ops_completed_sync;
+      sum.ops_completed_async += metrics.ops_completed_async;
+      sum.ops_completed_async_unref += metrics.ops_completed_async_unref;
+      sum.bytes_sent_control += metrics.bytes_sent_control;
+      sum.bytes_sent_data += metrics.bytes_sent_data;
+      sum.bytes_received += metrics.bytes_received;
+    }
+
+    sum
+  }
+
+  fn ensure_capacity(&mut self, op_id: OpId) {
+    if op_id >= self.ops.len() {
+      let delta_len = 1 + op_id - self.ops.len();
+      self.ops.extend(vec![OpMetrics::default(); delta_len])
+    }
+  }
+
+  fn metrics_mut(&mut self, id: OpId) -> &mut OpMetrics {
+    self.ensure_capacity(id);
+    self.ops.get_mut(id).unwrap()
+  }
+
+  pub fn track_sync(&mut self, id: OpId) {
+    let metrics = self.metrics_mut(id);
+    metrics.ops_dispatched += 1;
+    metrics.ops_completed += 1;
+    metrics.ops_dispatched_sync += 1;
+    metrics.ops_completed_sync += 1;
+  }
+
+  pub fn track_async(&mut self, id: OpId) {
+    let metrics = self.metrics_mut(id);
+    metrics.ops_dispatched += 1;
+    metrics.ops_dispatched_async += 1;
+  }
+
+  pub fn track_async_completed(&mut self, id: OpId) {
+    let metrics = self.metrics_mut(id);
+    metrics.ops_completed += 1;
+    metrics.ops_completed_async += 1;
+  }
+
+  pub fn track_unref(&mut self, id: OpId) {
+    let metrics = self.metrics_mut(id);
+    metrics.ops_dispatched += 1;
+    metrics.ops_dispatched_async_unref += 1;
+  }
+
+  pub fn track_unref_completed(&mut self, id: OpId) {
+    let metrics = self.metrics_mut(id);
+    metrics.ops_completed += 1;
+    metrics.ops_completed_async_unref += 1;
+  }
+}