Added YellowZone check after preallocation in gracejoin

Additional check that after allocating memory we did not go beyond the yellow zone. Otherwise deallocate and switch to spilling
commit_hash:c240687c121d99a4b794f5da2993e984854d6bc8

1 files changed, 5 insertions, 0 deletions

diff --git a/yql/essentials/minikql/comp_nodes/mkql_grace_join_imp.cpp b/yql/essentials/minikql/comp_nodes/mkql_grace_join_imp.cpp
index 475e9a012e5..5163d779423 100644
--- a/yql/essentials/minikql/comp_nodes/mkql_grace_join_imp.cpp
+++ b/yql/essentials/minikql/comp_nodes/mkql_grace_join_imp.cpp
@@ -359,10 +359,15 @@ bool TTable::TryToPreallocateMemoryForJoin(TTable & t1, TTable & t2, EJoinKind /
                 tableForPreallocation.NumberOfKeyStringColumns != 0, tableForPreallocation.NumberOfKeyIColumns != 0);
         const auto slotSize = ComputeNumberOfSlots(tableForPreallocation.TableBucketsStats[bucket].TuplesNum);
 
+        bool wasException = false;
         try {
             bucketForPreallocation.JoinSlots.reserve(nSlots*slotSize);
             bucketForPreallocationStats.BloomFilter.Reserve(bucketForPreallocationStats.TuplesNum);
         } catch (TMemoryLimitExceededException) {
+            wasException = true;
+        }
+
+        if (wasException || TlsAllocState->IsMemoryYellowZoneEnabled()) {
             for (ui64 i = 0; i < bucket; ++i) {
                 auto& b1 = t1.TableBuckets[i];
                 b1.JoinSlots.resize(0);