ANY type creation

marcboeker · Sep 13, 2024 · cdf8723 · cdf8723
1 parent fbb2e4d
commit cdf8723
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/type_info.go b/type_info.go
@@ -72,7 +72,7 @@ type TypeInfo interface {
 // NewTypeInfo returns type information for DuckDB's primitive types.
 func NewTypeInfo(t Type) (TypeInfo, error) {
 	name, inMap := unsupportedTypeToStringMap[t]
-	if inMap {
+	if inMap && t != TYPE_ANY {
 		return nil, getError(errAPI, unsupportedTypeError(name))
 	}
 
@@ -183,7 +183,7 @@ func (info *typeInfo) logicalType() C.duckdb_logical_type {
 	case TYPE_BOOLEAN, TYPE_TINYINT, TYPE_SMALLINT, TYPE_INTEGER, TYPE_BIGINT, TYPE_UTINYINT, TYPE_USMALLINT,
 		TYPE_UINTEGER, TYPE_UBIGINT, TYPE_FLOAT, TYPE_DOUBLE, TYPE_TIMESTAMP, TYPE_TIMESTAMP_S, TYPE_TIMESTAMP_MS,
 		TYPE_TIMESTAMP_NS, TYPE_TIMESTAMP_TZ, TYPE_DATE, TYPE_TIME, TYPE_INTERVAL, TYPE_HUGEINT, TYPE_VARCHAR,
-		TYPE_BLOB, TYPE_UUID:
+		TYPE_BLOB, TYPE_UUID, TYPE_ANY:
 		return C.duckdb_create_logical_type(C.duckdb_type(info.Type))
 
 	case TYPE_DECIMAL:

diff --git a/type_info_test.go b/type_info_test.go
@@ -10,7 +10,7 @@ func TestTypeInfo(t *testing.T) {
 	var primitiveTypes []Type
 	for k := range typeToStringMap {
 		_, inMap := unsupportedTypeToStringMap[k]
-		if inMap {
+		if inMap && k != TYPE_ANY {
 			continue
 		}
 		switch k {
@@ -75,7 +75,9 @@ func TestErrTypeInfo(t *testing.T) {
 
 	var unsupportedTypes []Type
 	for k := range unsupportedTypeToStringMap {
-		unsupportedTypes = append(unsupportedTypes, k)
+		if k != TYPE_ANY {
+			unsupportedTypes = append(unsupportedTypes, k)
+		}
 	}
 
 	for _, unsupported := range unsupportedTypes {