Fix pageutils will truncate the file if write failed

dbms/src/Storages/Page/PageUtil.h

@@ -152,7 +152,9 @@ void ftruncateFile(T & file, off_t length)
         DB::throwFromErrno(fmt::format("Cannot truncate file: {}. ", file->getFileName()), ErrorCodes::CANNOT_FTRUNCATE);
 }
 
-
+// TODO: split current api into V2 and V3.
+// Too many args in this function.
+// Also split read
 template <typename T>
 void writeFile(
     T & file,
@@ -161,6 +163,7 @@ void writeFile(
     size_t to_write,
     const WriteLimiterPtr & write_limiter = nullptr,
     const bool background = false,
+    const bool truncate_if_failed = true,
     [[maybe_unused]] bool enable_failpoint = false)
 {
     if (write_limiter)
@@ -192,15 +195,21 @@ void writeFile(
             {
                 ProfileEvents::increment(ProfileEvents::PSMWriteFailed);
                 auto saved_errno = errno;
-                // If error occurs, apply `ftruncate` try to truncate the broken bytes we have written.
-                // Note that the result of this ftruncate is ignored, there is nothing we can do to
-                // handle ftruncate error. The errno may change after ftruncate called.
-                int truncate_res = ::ftruncate(file->getFd(), offset);
+
+                int truncate_res = 0;
+                // If write failed in V3, Don't do truncate
+                if (truncate_if_failed)
+                {
+                    // If error occurs, apply `ftruncate` try to truncate the broken bytes we have written.
+                    // Note that the result of this ftruncate is ignored, there is nothing we can do to
+                    // handle ftruncate error. The errno may change after ftruncate called.
+                    truncate_res = ::ftruncate(file->getFd(), offset);
+                }
 
                 DB::throwFromErrno(fmt::format("Cannot write to file {},[truncate_res = {}],[errno_after_truncate = {}],"
                                                "[bytes_written={},to_write={},offset = {}]",
                                                file->getFileName(),
-                                               truncate_res,
+                                               truncate_if_failed ? DB::toString(truncate_res) : "no need truncate",
                                                strerror(errno),
                                                bytes_written,
                                                to_write,

dbms/src/Storages/Page/V2/tests/gtest_page_util.cpp

@@ -41,7 +41,7 @@ TEST(PageUtils_test, ReadWriteFile)
         buff_write[i] = i % 0xFF;
     }
     WritableFilePtr file_for_write = std::make_shared<PosixWritableFile>(FileName, true, -1, 0666);
-    PageUtil::writeFile(file_for_write, 0, buff_write, buff_size, /*write_limiter*/ nullptr, /*background*/ false, /*enable_failpoint*/ true);
+    PageUtil::writeFile(file_for_write, 0, buff_write, buff_size, /*write_limiter*/ nullptr, /*background*/ false, /*truncate_if_failed*/ true, /*enable_failpoint*/ true);
     PageUtil::syncFile(file_for_write);
     file_for_write->close();
 
@@ -78,7 +78,7 @@ TEST(PageUtils_test, BigReadWriteFile)
             buff_write[i] = i % 0xFF;
         }
 
-        PageUtil::writeFile(file_for_write, 0, buff_write, buff_size, nullptr, /*background*/ false, /*enable_failpoint*/ false);
+        PageUtil::writeFile(file_for_write, 0, buff_write, buff_size, nullptr, /*background*/ false, /*truncate_if_failed*/ true, /*enable_failpoint*/ false);
         PageUtil::syncFile(file_for_write);
         file_for_write->close();
 

dbms/src/Storages/Page/V3/BlobFile.cpp

@@ -91,9 +91,9 @@ void BlobFile::write(char * buffer, size_t offset, size_t size, const WriteLimit
               });
 
 #ifndef NDEBUG
-    PageUtil::writeFile(wrfile, offset, buffer, size, write_limiter, background, true);
+    PageUtil::writeFile(wrfile, offset, buffer, size, write_limiter, background, /*truncate_if_failed*/ false, /*enable_failpoint*/ true);
 #else
-    PageUtil::writeFile(wrfile, offset, buffer, size, write_limiter, background, false);
+    PageUtil::writeFile(wrfile, offset, buffer, size, write_limiter, background, /*truncate_if_failed*/ false, /*enable_failpoint*/ false);
 #endif
     PageUtil::syncFile(wrfile);
 

dbms/src/Storages/Page/V3/LogFile/LogWriter.cpp

@@ -69,7 +69,14 @@ size_t LogWriter::writtenBytes() const
 
 void LogWriter::flush(const WriteLimiterPtr & write_limiter, const bool background)
 {
-    PageUtil::writeFile(log_file, written_bytes, write_buffer.buffer().begin(), write_buffer.offset(), write_limiter, /*background*/ background, /*enable_failpoint*/ false);
+    PageUtil::writeFile(log_file,
+                        written_bytes,
+                        write_buffer.buffer().begin(),
+                        write_buffer.offset(),
+                        write_limiter,
+                        /*background*/ background,
+                        /*truncate_if_failed*/ false,
+                        /*enable_failpoint*/ false);
     log_file->fsync();
     written_bytes += write_buffer.offset();