Merge pull request #72386 from ClickHouse/chesema-fix-wb-finalize

fix finalize/cancel write buffers
This commit is contained in:
Sema Checherinda 2024-11-25 17:26:23 +00:00 committed by GitHub
commit 7173c71142
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 1 deletions

View File

@ -74,6 +74,7 @@ void StorageObjectStorageSink::finalizeBuffers()
catch (...)
{
/// Stop ParallelFormattingOutputFormat correctly.
cancelBuffers();
releaseBuffers();
throw;
}

View File

@ -561,7 +561,7 @@ void StorageMemory::restoreDataImpl(const BackupPtr & backup, const String & dat
temp_data_file.emplace(temporary_disk);
auto out = std::make_unique<WriteBufferFromFile>(temp_data_file->getAbsolutePath());
copyData(*in, *out);
out.reset();
out->finalize();
in = createReadBufferFromFileBase(temp_data_file->getAbsolutePath(), {});
}
std::unique_ptr<ReadBufferFromFileBase> in_from_file{static_cast<ReadBufferFromFileBase *>(in.release())};