add setTypeId to sparse UnionVector

xtdb · May 21, 2024 · 8573abb · 8573abb
1 parent d45468e
commit 8573abb
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/java/vector/src/main/codegen/templates/UnionVector.java b/java/vector/src/main/codegen/templates/UnionVector.java
@@ -827,10 +827,14 @@ public void setSafe(int index, Nullable${name}Holder holder) {
     </#list>
 
     public void setType(int index, MinorType type) {
+      setTypeId(index, (byte) type.ordinal());
+    }
+
+    public void setTypeId(int index, byte typeId) {
       while (index >= getTypeBufferValueCapacity()) {
         reallocTypeBuffer();
       }
-      typeBuffer.setByte(index * TYPE_WIDTH , (byte) type.ordinal());
+      typeBuffer.setByte(index * TYPE_WIDTH , typeId);
     }
 
     private int getTypeBufferValueCapacity() {