Update paimon-core/src/main/java/org/apache/paimon/utils/BulkFormatMa…

…pping.java Co-authored-by: tsreaper <[email protected]>
apache · Dec 9, 2024 · a906414 · a906414
1 parent 3f102a3
commit a906414
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/paimon-core/src/main/java/org/apache/paimon/utils/BulkFormatMapping.java b/paimon-core/src/main/java/org/apache/paimon/utils/BulkFormatMapping.java
@@ -207,8 +207,8 @@ static Pair<int[], RowType> trimKeyFields(
                     if (positionMap.containsKey(id)) {
                         map[i] = positionMap.get(id);
                     } else {
-                        trimmedFields.add(keyField ? f : field);
-                        map[i] = positionMap.computeIfAbsent(id, k -> index.getAndIncrement());
+                        map[i] = positionMap.put(id, trimmedFields.size());
+                        trimmedFields.add(f);
                     }
                 } else {
                     throw new RuntimeException("Can't find field with id: " + id + " in fields.");