From 2caf003db01369ac084315627628d09897069e6c Mon Sep 17 00:00:00 2001
From: Ning Sun <sunng@protonmail.com>
Date: Sat, 4 Jan 2025 11:49:01 +0800
Subject: [PATCH] fix: brings back functions required by downstream projects
 (#5283)

---
 src/cmd/src/datanode.rs | 5 +++++
 src/cmd/src/flownode.rs | 5 +++++
 2 files changed, 10 insertions(+)

diff --git a/src/cmd/src/datanode.rs b/src/cmd/src/datanode.rs
index be2aedf57e..3737b9266a 100644
--- a/src/cmd/src/datanode.rs
+++ b/src/cmd/src/datanode.rs
@@ -62,6 +62,11 @@ impl Instance {
     pub fn datanode(&self) -> &Datanode {
         &self.datanode
     }
+
+    /// allow customizing datanode for downstream projects
+    pub fn datanode_mut(&mut self) -> &mut Datanode {
+        &mut self.datanode
+    }
 }
 
 #[async_trait]
diff --git a/src/cmd/src/flownode.rs b/src/cmd/src/flownode.rs
index b399bf37f7..e629edf251 100644
--- a/src/cmd/src/flownode.rs
+++ b/src/cmd/src/flownode.rs
@@ -66,6 +66,11 @@ impl Instance {
     pub fn flownode(&self) -> &FlownodeInstance {
         &self.flownode
     }
+
+    /// allow customizing flownode for downstream projects
+    pub fn flownode_mut(&mut self) -> &mut FlownodeInstance {
+        &mut self.flownode
+    }
 }
 
 #[async_trait::async_trait]