From d1ef104aa3bd633e031a3b199fcb820ae6d75cc0 Mon Sep 17 00:00:00 2001
From: zhiqiang-hhhh <seuhezhiqiang@163.com>
Date: Fri, 14 Jun 2024 17:00:52 +0800
Subject: [PATCH 1/2] FIX

---
 be/src/vec/exec/scan/scanner_scheduler.cpp | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/be/src/vec/exec/scan/scanner_scheduler.cpp b/be/src/vec/exec/scan/scanner_scheduler.cpp
index 6c1f02530b2efe..6b828395a1128a 100644
--- a/be/src/vec/exec/scan/scanner_scheduler.cpp
+++ b/be/src/vec/exec/scan/scanner_scheduler.cpp
@@ -31,6 +31,7 @@
 #include "common/compiler_util.h" // IWYU pragma: keep
 #include "common/config.h"
 #include "common/logging.h"
+#include "common/status.h"
 #include "olap/tablet.h"
 #include "runtime/exec_env.h"
 #include "runtime/runtime_state.h"
@@ -315,7 +316,8 @@ void ScannerScheduler::_scanner_scan(std::shared_ptr<ScannerContext> ctx,
                 eos = true;
             },
             status);
-    if (status.is<doris::ErrorCode::MEM_ALLOC_FAILED>()) {
+
+    if (status.is<doris::ErrorCode::MEM_ALLOC_FAILED>() || status.is<doris::ErrorCode::MEM_LIMIT_EXCEEDED>()) {
         scan_task->set_status(status);
         eos = true;
     }

From 1d078db7ff5656e784d89d2f429c4af5d444e749 Mon Sep 17 00:00:00 2001
From: zhiqiang-hhhh <seuhezhiqiang@163.com>
Date: Fri, 14 Jun 2024 17:01:33 +0800
Subject: [PATCH 2/2] FORMAT

---
 be/src/vec/exec/scan/scanner_scheduler.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/be/src/vec/exec/scan/scanner_scheduler.cpp b/be/src/vec/exec/scan/scanner_scheduler.cpp
index 6b828395a1128a..63dd042d720252 100644
--- a/be/src/vec/exec/scan/scanner_scheduler.cpp
+++ b/be/src/vec/exec/scan/scanner_scheduler.cpp
@@ -317,7 +317,8 @@ void ScannerScheduler::_scanner_scan(std::shared_ptr<ScannerContext> ctx,
             },
             status);
 
-    if (status.is<doris::ErrorCode::MEM_ALLOC_FAILED>() || status.is<doris::ErrorCode::MEM_LIMIT_EXCEEDED>()) {
+    if (status.is<doris::ErrorCode::MEM_ALLOC_FAILED>() ||
+        status.is<doris::ErrorCode::MEM_LIMIT_EXCEEDED>()) {
         scan_task->set_status(status);
         eos = true;
     }