initial

Author: NEUpanning

velox/exec/Spiller.cpp

@@ -238,12 +238,28 @@ void SpillerBase::ensureSorted(SpillRun& run) {
             return container_->compareRows(left, right, compareFlags_) < 0;
           });
     } else {
-      PrefixSort::sort(
-          container_,
-          compareFlags_,
-          state_.prefixSortConfig().value(),
-          memory::spillMemoryPool(),
-          run.rows);
+      try {
+        PrefixSort::sort(
+            container_,
+            compareFlags_,
+            state_.prefixSortConfig().value(),
+            memory::spillMemoryPool(),
+            run.rows);
+      } catch (const VeloxRuntimeError& e) {
+        // If memory allocation failed, fallback to timsort.
+        if (e.errorCode() == error_code::kMemAllocError) {
+          LOG(WARNING) << "PrefixSort failed due to memory allocation error, "
+                       << "falling back to TimSort";
+          gfx::timsort(
+              run.rows.begin(),
+              run.rows.end(),
+              [&](const char* left, const char* right) {
+                return container_->compareRows(left, right, compareFlags_) < 0;
+              });
+        } else {
+          throw;
+        }
+      }
     }
 
     run.sorted = true;