From 30429a7f6c9bf4eef09cb2cf98e26ba3aefbfd7f Mon Sep 17 00:00:00 2001
From: Alexey Antipovsky <alexey.vladimirovich@mariadb.com>
Date: Fri, 1 Oct 2021 16:10:46 +0300
Subject: [PATCH] Fix excessive memory consumption at the last stage of
 aggregation

---
 utils/rowgroup/rowstorage.cpp | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/utils/rowgroup/rowstorage.cpp b/utils/rowgroup/rowstorage.cpp
index 4c296ca6e..28152122d 100644
--- a/utils/rowgroup/rowstorage.cpp
+++ b/utils/rowgroup/rowstorage.cpp
@@ -684,7 +684,13 @@ class RowGroupStorage
               logging::ERR_AGGREGATION_TOO_BIG);
         }
 
-        fLRU->add(rgid);
+        if (fMM->getFree() < memSz * 2)
+        {
+          saveRG(rgid);
+          fRGDatas[rgid].reset();
+        }
+        else
+          fLRU->add(rgid);
       }
       else
       {