diff --git a/hudi-client/hudi-client-common/src/main/java/org/apache/hudi/io/HoodieCreateHandle.java b/hudi-client/hudi-client-common/src/main/java/org/apache/hudi/io/HoodieCreateHandle.java index 99a43173e..b57a43991 100644 --- a/hudi-client/hudi-client-common/src/main/java/org/apache/hudi/io/HoodieCreateHandle.java +++ b/hudi-client/hudi-client-common/src/main/java/org/apache/hudi/io/HoodieCreateHandle.java @@ -73,6 +73,7 @@ public class HoodieCreateHandle extends taskContextSupplier); writeStatus.setFileId(fileId); writeStatus.setPartitionPath(partitionPath); + writeStatus.setStat(new HoodieWriteStat()); this.path = makeNewPath(partitionPath); @@ -200,7 +201,7 @@ public class HoodieCreateHandle extends * @throws IOException if error occurs */ protected void setupWriteStatus() throws IOException { - HoodieWriteStat stat = new HoodieWriteStat(); + HoodieWriteStat stat = writeStatus.getStat(); stat.setPartitionPath(writeStatus.getPartitionPath()); stat.setNumWrites(recordsWritten); stat.setNumDeletes(recordsDeleted); @@ -214,7 +215,6 @@ public class HoodieCreateHandle extends RuntimeStats runtimeStats = new RuntimeStats(); runtimeStats.setTotalCreateTime(timer.endTimer()); stat.setRuntimeStats(runtimeStats); - writeStatus.setStat(stat); } protected long computeTotalWriteBytes() throws IOException { diff --git a/hudi-client/hudi-flink-client/src/main/java/org/apache/hudi/io/storage/row/HoodieRowDataCreateHandle.java b/hudi-client/hudi-flink-client/src/main/java/org/apache/hudi/io/storage/row/HoodieRowDataCreateHandle.java index 3ceee8ba0..76fad8b1e 100644 --- a/hudi-client/hudi-flink-client/src/main/java/org/apache/hudi/io/storage/row/HoodieRowDataCreateHandle.java +++ b/hudi-client/hudi-flink-client/src/main/java/org/apache/hudi/io/storage/row/HoodieRowDataCreateHandle.java @@ -86,6 +86,7 @@ public class HoodieRowDataCreateHandle implements Serializable { writeConfig.getWriteStatusFailureFraction()); writeStatus.setPartitionPath(partitionPath); writeStatus.setFileId(fileId); + writeStatus.setStat(new HoodieWriteStat()); try { HoodiePartitionMetadata partitionMetadata = new HoodiePartitionMetadata( @@ -145,7 +146,7 @@ public class HoodieRowDataCreateHandle implements Serializable { */ public HoodieInternalWriteStatus close() throws IOException { fileWriter.close(); - HoodieWriteStat stat = new HoodieWriteStat(); + HoodieWriteStat stat = writeStatus.getStat(); stat.setPartitionPath(partitionPath); stat.setNumWrites(writeStatus.getTotalRecords()); stat.setNumDeletes(0); @@ -160,7 +161,6 @@ public class HoodieRowDataCreateHandle implements Serializable { HoodieWriteStat.RuntimeStats runtimeStats = new HoodieWriteStat.RuntimeStats(); runtimeStats.setTotalCreateTime(currTimer.endTimer()); stat.setRuntimeStats(runtimeStats); - writeStatus.setStat(stat); return writeStatus; } diff --git a/hudi-client/hudi-spark-client/src/main/java/org/apache/hudi/io/storage/row/HoodieRowCreateHandle.java b/hudi-client/hudi-spark-client/src/main/java/org/apache/hudi/io/storage/row/HoodieRowCreateHandle.java index 2aaa89a9e..85c2f8408 100644 --- a/hudi-client/hudi-spark-client/src/main/java/org/apache/hudi/io/storage/row/HoodieRowCreateHandle.java +++ b/hudi-client/hudi-spark-client/src/main/java/org/apache/hudi/io/storage/row/HoodieRowCreateHandle.java @@ -86,6 +86,7 @@ public class HoodieRowCreateHandle implements Serializable { writeConfig.getWriteStatusFailureFraction()); writeStatus.setPartitionPath(partitionPath); writeStatus.setFileId(fileId); + writeStatus.setStat(new HoodieWriteStat()); try { HoodiePartitionMetadata partitionMetadata = new HoodiePartitionMetadata( @@ -144,7 +145,7 @@ public class HoodieRowCreateHandle implements Serializable { */ public HoodieInternalWriteStatus close() throws IOException { fileWriter.close(); - HoodieWriteStat stat = new HoodieWriteStat(); + HoodieWriteStat stat = writeStatus.getStat(); stat.setPartitionPath(partitionPath); stat.setNumWrites(writeStatus.getTotalRecords()); stat.setNumDeletes(0); @@ -159,7 +160,6 @@ public class HoodieRowCreateHandle implements Serializable { HoodieWriteStat.RuntimeStats runtimeStats = new HoodieWriteStat.RuntimeStats(); runtimeStats.setTotalCreateTime(currTimer.endTimer()); stat.setRuntimeStats(runtimeStats); - writeStatus.setStat(stat); return writeStatus; }