From d035fcbb3cf7aa33bafdd902a5e6a76f549e7ee1 Mon Sep 17 00:00:00 2001
From: hiscat <46845236+MyLanPangzi@users.noreply.github.com>
Date: Wed, 7 Apr 2021 14:13:31 +0800
Subject: [PATCH] [HUDI-1767] Add setter to HoodieKey and HoodieRecordLocation
 to have better SE/DE performance for Flink (#2779)

---
 .../org/apache/hudi/common/model/HoodieKey.java   | 14 ++++++++++++--
 .../hudi/common/model/HoodieRecordLocation.java   | 15 +++++++++++++--
 2 files changed, 25 insertions(+), 4 deletions(-)

diff --git a/hudi-common/src/main/java/org/apache/hudi/common/model/HoodieKey.java b/hudi-common/src/main/java/org/apache/hudi/common/model/HoodieKey.java
index c40bdc45c..903020409 100644
--- a/hudi-common/src/main/java/org/apache/hudi/common/model/HoodieKey.java
+++ b/hudi-common/src/main/java/org/apache/hudi/common/model/HoodieKey.java
@@ -29,19 +29,29 @@ import java.util.Objects;
  */
 public class HoodieKey implements Serializable {
 
-  private final String recordKey;
+  private String recordKey;
+  private String partitionPath;
 
-  private final String partitionPath;
+  public HoodieKey() {
+  }
 
   public HoodieKey(String recordKey, String partitionPath) {
     this.recordKey = recordKey;
     this.partitionPath = partitionPath;
   }
 
+  public void setRecordKey(String recordKey) {
+    this.recordKey = recordKey;
+  }
+
   public String getRecordKey() {
     return recordKey;
   }
 
+  public void setPartitionPath(String partitionPath) {
+    this.partitionPath = partitionPath;
+  }
+
   public String getPartitionPath() {
     return partitionPath;
   }
diff --git a/hudi-common/src/main/java/org/apache/hudi/common/model/HoodieRecordLocation.java b/hudi-common/src/main/java/org/apache/hudi/common/model/HoodieRecordLocation.java
index 690db8837..1692cfbaa 100644
--- a/hudi-common/src/main/java/org/apache/hudi/common/model/HoodieRecordLocation.java
+++ b/hudi-common/src/main/java/org/apache/hudi/common/model/HoodieRecordLocation.java
@@ -26,8 +26,11 @@ import java.util.Objects;
  */
 public class HoodieRecordLocation implements Serializable {
 
-  private final String instantTime;
-  private final String fileId;
+  private String instantTime;
+  private String fileId;
+
+  public HoodieRecordLocation() {
+  }
 
   public HoodieRecordLocation(String instantTime, String fileId) {
     this.instantTime = instantTime;
@@ -64,7 +67,15 @@ public class HoodieRecordLocation implements Serializable {
     return instantTime;
   }
 
+  public void setInstantTime(String instantTime) {
+    this.instantTime = instantTime;
+  }
+
   public String getFileId() {
     return fileId;
   }
+
+  public void setFileId(String fileId) {
+    this.fileId = fileId;
+  }
 }