From 3f20f1d6d8a6d53d42f78c5a2c424c66c01eb23b Mon Sep 17 00:00:00 2001
From: discord9 <discord9@163.com>
Date: Wed, 9 Sep 2026 17:01:25 +0800
Subject: [PATCH] feat(flow): expose task query context for admission checks

Signed-off-by: discord9 <discord9@163.com>
---
 src/flow/src/batching_mode/task.rs | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/flow/src/batching_mode/task.rs b/src/flow/src/batching_mode/task.rs
index 5ebf8797a0..a040848c8e 100644
--- a/src/flow/src/batching_mode/task.rs
+++ b/src/flow/src/batching_mode/task.rs
@@ -369,6 +369,11 @@ impl BatchingTask {
         self.state.read().unwrap().start_time_millis()
     }
 
+    /// Returns a stable clone of the task query context without exposing mutable task state.
+    pub fn query_context_snapshot(&self) -> QueryContextRef {
+        self.state.read().unwrap().query_ctx.clone()
+    }
+
     /// Collect flow-related extensions from the task's query context that should be
     /// forwarded to the frontend (e.g. scheduled time).
     fn frontend_extensions(&self) -> HashMap<String, String> {