From 69fd36847dcde41073ad3f0ca085b5bf0863ea8c Mon Sep 17 00:00:00 2001
From: Alexey Antipovsky <alexey.vladimirovich@mariadb.com>
Date: Sun, 4 Sep 2022 23:44:19 +0000
Subject: [PATCH] [MCOL-5213] Fix a rare IO error

---
 utils/rowgroup/rowstorage.cpp | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/utils/rowgroup/rowstorage.cpp b/utils/rowgroup/rowstorage.cpp
index c326dab9f..e73790746 100644
--- a/utils/rowgroup/rowstorage.cpp
+++ b/utils/rowgroup/rowstorage.cpp
@@ -1556,9 +1556,10 @@ bool RowAggStorage::getTargetRow(const Row& row, uint64_t hash, Row& rowOut)
                                        !fEnabledDiskAggregation, fCompressor.get()));
     if (fExtKeys)
     {
-      fKeysStorage = new RowGroupStorage(fTmpDir, fKeysRowGroup, fMaxRows, fMM->getResourceManaged(),
+      fRealKeysStorage.reset(new RowGroupStorage(fTmpDir, fKeysRowGroup, fMaxRows, fMM->getResourceManaged(),
                                          fMM->getSessionLimit(), !fEnabledDiskAggregation,
-                                         !fEnabledDiskAggregation, fCompressor.get());
+                                         !fEnabledDiskAggregation, fCompressor.get()));
+      fKeysStorage = fRealKeysStorage.get();
     }
     else
     {