Core: Reset to main as part of replace shouldn't remove main ref

apache · Dec 19, 2024 · 621b7af · 621b7af
1 parent 540d6a6
commit 621b7af
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/core/src/main/java/org/apache/iceberg/MetadataUpdate.java b/core/src/main/java/org/apache/iceberg/MetadataUpdate.java
@@ -338,7 +338,11 @@ public String name() {
 
     @Override
     public void applyTo(TableMetadata.Builder metadataBuilder) {
-      metadataBuilder.removeRef(refName);
+      if (refName.equals(SnapshotRef.MAIN_BRANCH)) {
+        metadataBuilder.resetMainBranch();
+      } else {
+        metadataBuilder.removeRef(refName);
+      }
     }
   }
 

diff --git a/core/src/main/java/org/apache/iceberg/TableMetadata.java b/core/src/main/java/org/apache/iceberg/TableMetadata.java
@@ -1297,7 +1297,7 @@ public Builder removeRef(String name) {
       return this;
     }
 
-    private Builder resetMainBranch() {
+    protected Builder resetMainBranch() {
       this.currentSnapshotId = -1;
       SnapshotRef ref = refs.remove(SnapshotRef.MAIN_BRANCH);
       if (ref != null) {