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

apache · Dec 19, 2024 · 6c4287d · 6c4287d
1 parent 540d6a6
commit 6c4287d
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/core/src/main/java/org/apache/iceberg/TableMetadata.java b/core/src/main/java/org/apache/iceberg/TableMetadata.java
@@ -1299,11 +1299,8 @@ public Builder removeRef(String name) {
 
     private Builder resetMainBranch() {
       this.currentSnapshotId = -1;
-      SnapshotRef ref = refs.remove(SnapshotRef.MAIN_BRANCH);
-      if (ref != null) {
-        changes.add(new MetadataUpdate.RemoveSnapshotRef(SnapshotRef.MAIN_BRANCH));
-      }
-
+      refs.remove(SnapshotRef.MAIN_BRANCH);
+      setBranchSnapshot(-1, SnapshotRef.MAIN_BRANCH);
       return this;
     }