feat: helper methods on WalOp

influxdata · Oct 23, 2024 · b0d5a9f · b0d5a9f
1 parent f1bd868
commit b0d5a9f
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/influxdb3_wal/src/lib.rs b/influxdb3_wal/src/lib.rs
@@ -211,6 +211,22 @@ pub enum WalOp {
     Catalog(CatalogBatch),
 }
 
+impl WalOp {
+    pub fn as_write(&self) -> Option<&WriteBatch> {
+        match self {
+            WalOp::Write(w) => Some(w),
+            WalOp::Catalog(_) => None,
+        }
+    }
+
+    pub fn as_catalog(&self) -> Option<&CatalogBatch> {
+        match self {
+            WalOp::Write(_) => None,
+            WalOp::Catalog(c) => Some(c),
+        }
+    }
+}
+
 #[derive(Debug, Clone, Eq, PartialEq, Serialize, Deserialize)]
 pub struct CatalogBatch {
     pub database_id: DbId,