KAFKA-17181 Replace fileChannel.write with Utils.writeFully to prevent partial write (apache#16707)

mingyen066 · web-flow · commit 65786fecbf6b · 2024-08-20T09:26:43.000+08:00
Reviewers: Chia-Ping Tsai &lt;chia7712@gmail.com&gt;
diff --git a/storage/src/main/java/org/apache/kafka/storage/internals/log/ProducerStateManager.java b/storage/src/main/java/org/apache/kafka/storage/internals/log/ProducerStateManager.java
@@ -24,6 +24,7 @@
 import org.apache.kafka.common.utils.Crc32C;
 import org.apache.kafka.common.utils.LogContext;
 import org.apache.kafka.common.utils.Time;
+import org.apache.kafka.common.utils.Utils;
 import org.apache.kafka.server.log.remote.metadata.storage.generated.ProducerSnapshot;
 
 import org.slf4j.Logger;
@@ -684,7 +685,7 @@ public static void writeSnapshot(File file, Map<Long, ProducerStateEntry> entrie
         ByteUtils.writeUnsignedInt(buffer, CRC_OFFSET, crc);
 
         try (FileChannel fileChannel = FileChannel.open(file.toPath(), StandardOpenOption.CREATE, StandardOpenOption.WRITE)) {
-            fileChannel.write(buffer);
+            Utils.writeFully(fileChannel, buffer);
             if (sync) {
                 fileChannel.force(true);
             }
