From 61a2511ed0ac550693a20d2088aae59557a8f89c Mon Sep 17 00:00:00 2001
From: Yang Guo <jimmy_guoyang@outlook.com>
Date: Fri, 10 Apr 2026 16:16:28 +0800
Subject: [PATCH] fix: add lakeSource check in batch mode

---
 .../fluss/flink/source/enumerator/FlinkSourceEnumerator.java  | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/fluss-flink/fluss-flink-common/src/main/java/org/apache/fluss/flink/source/enumerator/FlinkSourceEnumerator.java b/fluss-flink/fluss-flink-common/src/main/java/org/apache/fluss/flink/source/enumerator/FlinkSourceEnumerator.java
index b537a81224..3ad234aa01 100644
--- a/fluss-flink/fluss-flink-common/src/main/java/org/apache/fluss/flink/source/enumerator/FlinkSourceEnumerator.java
+++ b/fluss-flink/fluss-flink-common/src/main/java/org/apache/fluss/flink/source/enumerator/FlinkSourceEnumerator.java
@@ -354,6 +354,10 @@ public void start() {
 
     private void startInBatchMode() {
         if (lakeEnabled) {
+            if (lakeSource == null) {
+                throw new IllegalStateException(
+                        "The 'lakeSource' is null in batch mode. It should be set if lake is enabled.");
+            }
             context.callAsync(
                     () -> {
                         List<SourceSplitBase> splits = generateHybridLakeFlussSplits();