From e8442b7a88467aa68a87c0abfa3983131c7b8c75 Mon Sep 17 00:00:00 2001
From: Antonio Andelic <antonio@clickhouse.com>
Date: Thu, 1 Jun 2023 12:54:00 +0000
Subject: [PATCH] Add finalize calls

---
 src/Coordination/Changelog.cpp             | 9 ++++++++-
 src/Coordination/KeeperSnapshotManager.cpp | 6 ++++--
 2 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/src/Coordination/Changelog.cpp b/src/Coordination/Changelog.cpp
index bcce9982e04..97f2b437377 100644
--- a/src/Coordination/Changelog.cpp
+++ b/src/Coordination/Changelog.cpp
@@ -43,7 +43,8 @@ void moveFileBetweenDisks(DiskPtr disk_from, ChangelogFileDescriptionPtr descrip
     auto from_path = fs::path(description->path);
     auto tmp_changelog_name = from_path.parent_path() / (std::string{tmp_prefix} + from_path.filename().string());
     {
-        disk_to->writeFile(tmp_changelog_name);
+        auto buf = disk_to->writeFile(tmp_changelog_name);
+        buf->finalize();
     }
     disk_from->copyFile(from_path, *disk_to, path_to, {});
     disk_to->removeFile(tmp_changelog_name);
@@ -342,9 +343,15 @@ private:
         }
 
         if (log_file_settings.compress_logs)
+        {
             compressed_buffer.reset();
+        }
         else
+        {
+            chassert(file_buf);
+            file_buf->finalize();
             file_buf.reset();
+        }
     }
 
     WriteBuffer & getBuffer()
diff --git a/src/Coordination/KeeperSnapshotManager.cpp b/src/Coordination/KeeperSnapshotManager.cpp
index 4b9a34c07e9..8dd6d27c379 100644
--- a/src/Coordination/KeeperSnapshotManager.cpp
+++ b/src/Coordination/KeeperSnapshotManager.cpp
@@ -43,7 +43,8 @@ namespace
         auto from_path = fs::path(path_from);
         auto tmp_snapshot_name = from_path.parent_path() / (std::string{tmp_prefix} + from_path.filename().string());
         {
-            disk_to->writeFile(tmp_snapshot_name);
+            auto buf = disk_to->writeFile(tmp_snapshot_name);
+            buf->finalize();
         }
         disk_from->copyFile(from_path, *disk_to, path_to, {});
         disk_to->removeFile(tmp_snapshot_name);
@@ -784,7 +785,8 @@ SnapshotFileInfo KeeperSnapshotManager::serializeSnapshotToDisk(const KeeperStor
 
     auto disk = getLatestSnapshotDisk();
     {
-        disk->writeFile(tmp_snapshot_file_name);
+        auto buf = disk->writeFile(tmp_snapshot_file_name);
+        buf->finalize();
     }
 
     auto writer = disk->writeFile(snapshot_file_name);