[test](memory) find large memory allocation not catch exception

apache · May 22, 2024 · 3f4325f · 3f4325f
1 parent c511fe3
commit 3f4325f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/be/src/vec/common/allocator.cpp b/be/src/vec/common/allocator.cpp
@@ -75,6 +75,12 @@ void Allocator<clear_memory_, mmap_populate, use_mmap>::sys_memory_check(size_t
         }
     }
 
+    // Find large memory allocation not catch exception
+    if (size > 1000000 && !doris::enable_thread_catch_bad_alloc) {
+        LOG(INFO) << "There is a large allocation " << size << ", not catch exception. "
+                  << Status::InternalError("bad alloc");
+    }
+
     if (doris::GlobalMemoryArbitrator::is_exceed_hard_mem_limit(size)) {
         // Only thread attach query, and has not completely waited for thread_wait_gc_max_milliseconds,
         // will wait for gc, asynchronous cancel or throw bad::alloc.

diff --git a/regression-test/suites/correctness_p0/test_avg.groovy b/regression-test/suites/correctness_p0/test_avg.groovy
@@ -41,7 +41,7 @@ suite("test_avg") {
     qt_select """select c_bigint from ${tableName} order by c_bigint"""
     qt_sum """ SELECT SUM(c_bigint) FROM ${tableName} """
     qt_count """ SELECT COUNT(c_bigint) FROM ${tableName} """
-    qt_avg """ SELECT AVG(c_bigint) FROM ${tableName} """
+    qt_avg """ SELECT AVG(c_bigint1) FROM ${tableName} """
     sql""" DROP TABLE IF EXISTS ${tableName} """