Preserve typed vector write performance

marcboeker#276 made SetChunkValue
generic to avoid allocating when table UDFs write values. Routing every
write through the any-valued callback restored correctness locality but
boxed each generic value.

Install optional exact-type callbacks for each vector canonical Go type,
with setFn as the fallback for all other values, and keep numeric
conversion setters generic. The numeric optimization defaults to the
vector-installed setFn, so types it does not recognize retain the single
correctness dispatch in vector initialization.

Add benchmarks for both public table UDF write paths and a matrix
covering cross-numeric and representative canonical writes. These paths
now match or improve on main allocation counts.

Author: mlafeldt

data_chunk.go

@@ -96,7 +96,7 @@ func SetChunkValue[T any](chunk DataChunk, colIdx, rowIdx int, val T) error {
 		return getError(errAPI, err)
 	}
 
-	return chunk.columns[colIdx].SetValue(rowIdx, val)
+	return setVectorVal(&chunk.columns[colIdx], mapping.IdxT(rowIdx), val)
 }
 
 func inBounds[T any](s []T, idx int) bool {

vector.go

@@ -23,6 +23,8 @@ type vector struct {
 	getFn fnGetVectorValue
 	// A callback function to write to this vector.
 	setFn fnSetVectorValue
+	// An optional exact-type setter that avoids boxing generic writes.
+	setTypedFn any
 	// The child vectors of nested data types.
 	childVectors []vector
 	// structTemplate is a pre-allocated map[string]any with all struct keys
@@ -168,6 +170,7 @@ func initBool(vec *vector) {
 		}
 		return setBool(vec, rowIdx, val)
 	}
+	vec.setTypedFn = fnSetVectorValueTyped[bool](setBool[bool])
 	vec.Type = TYPE_BOOLEAN
 }
 
@@ -185,6 +188,7 @@ func initNumeric[T numericType](vec *vector, t Type) {
 		}
 		return setNumeric[any, T](vec, rowIdx, val)
 	}
+	vec.setTypedFn = fnSetVectorValueTyped[T](setNumeric[T, T])
 	vec.Type = t
 }
 
@@ -202,6 +206,7 @@ func (vec *vector) initTS(t Type) {
 		}
 		return setTS(vec, rowIdx, val)
 	}
+	vec.setTypedFn = fnSetVectorValueTyped[time.Time](setTS[time.Time])
 	vec.Type = t
 }
 
@@ -219,6 +224,7 @@ func (vec *vector) initDate() {
 		}
 		return setDate(vec, rowIdx, val)
 	}
+	vec.setTypedFn = fnSetVectorValueTyped[time.Time](setDate[time.Time])
 	vec.Type = TYPE_DATE
 }
 
@@ -236,6 +242,7 @@ func (vec *vector) initTime(t Type) {
 		}
 		return setTime(vec, rowIdx, val)
 	}
+	vec.setTypedFn = fnSetVectorValueTyped[time.Time](setTime[time.Time])
 	vec.Type = t
 }
 
@@ -253,6 +260,7 @@ func (vec *vector) initInterval() {
 		}
 		return setInterval(vec, rowIdx, val)
 	}
+	vec.setTypedFn = fnSetVectorValueTyped[Interval](setInterval[Interval])
 	vec.Type = TYPE_INTERVAL
 }
 
@@ -321,6 +329,11 @@ func (vec *vector) initBytes(t Type) {
 		}
 		return setBytes(vec, rowIdx, val)
 	}
+	if t == TYPE_VARCHAR {
+		vec.setTypedFn = fnSetVectorValueTyped[string](setBytes[string])
+	} else {
+		vec.setTypedFn = fnSetVectorValueTyped[[]byte](setBytes[[]byte])
+	}
 	vec.Type = t
 }
 
@@ -338,6 +351,7 @@ func (vec *vector) initBit() {
 		}
 		return setBit(vec, rowIdx, val)
 	}
+	vec.setTypedFn = fnSetVectorValueTyped[Bit](setBit[Bit])
 	vec.Type = TYPE_BIT
 }
 
@@ -355,6 +369,7 @@ func (vec *vector) initJSON() {
 		}
 		return setJSON(vec, rowIdx, val)
 	}
+	vec.setTypedFn = fnSetVectorValueTyped[string](setJSON[string])
 	vec.Type = TYPE_VARCHAR
 }
 
@@ -423,6 +438,7 @@ func (vec *vector) initEnum(logicalType mapping.LogicalType, colIdx int) error {
 
 	vec.Type = TYPE_ENUM
 	vec.internalType = t
+	vec.setTypedFn = fnSetVectorValueTyped[string](setEnum[string])
 	return nil
 }
 
@@ -451,6 +467,7 @@ func (vec *vector) initList(logicalType mapping.LogicalType, colIdx int) error {
 		}
 		return setList(vec, rowIdx, val)
 	}
+	vec.setTypedFn = fnSetVectorValueTyped[[]any](setList[[]any])
 	vec.Type = TYPE_LIST
 	return nil
 }
@@ -502,6 +519,7 @@ func (vec *vector) initStruct(logicalType mapping.LogicalType, colIdx int) error
 		}
 		return setStruct(vec, rowIdx, val)
 	}
+	vec.setTypedFn = fnSetVectorValueTyped[map[string]any](setStruct[map[string]any])
 	vec.Type = TYPE_STRUCT
 	return nil
 }
@@ -544,6 +562,7 @@ func (vec *vector) initMap(logicalType mapping.LogicalType, colIdx int) error {
 		}
 		return setMap(vec, rowIdx, val)
 	}
+	vec.setTypedFn = fnSetVectorValueTyped[OrderedMap](setMap[OrderedMap])
 	vec.Type = TYPE_MAP
 	return nil
 }
@@ -575,6 +594,7 @@ func (vec *vector) initArray(logicalType mapping.LogicalType, colIdx int) error
 		}
 		return setArray(vec, rowIdx, val)
 	}
+	vec.setTypedFn = fnSetVectorValueTyped[[]any](setArray[[]any])
 	vec.Type = TYPE_ARRAY
 	return nil
 }
@@ -621,6 +641,7 @@ func (vec *vector) initUnion(logicalType mapping.LogicalType, colIdx int) error
 		}
 		return setUnion(vec, rowIdx, val)
 	}
+	vec.setTypedFn = fnSetVectorValueTyped[Union](setUnion[Union])
 	vec.Type = TYPE_UNION
 	return nil
 }
@@ -640,6 +661,7 @@ func (vec *vector) initUUID() {
 		}
 		return setUUID(vec, rowIdx, val)
 	}
+	vec.setTypedFn = fnSetVectorValueTyped[UUID](setUUID[UUID])
 	vec.Type = TYPE_UUID
 }