Basic list support

Giorgi · Oct 1, 2023 · 421983d · 421983d
1 parent b01befb
commit 421983d
Show file tree

Hide file tree

Showing 5 changed files with 467 additions and 52 deletions.
diff --git a/DuckDB.NET.Bindings/DuckDBNativeObjects.cs b/DuckDB.NET.Bindings/DuckDBNativeObjects.cs
@@ -51,7 +51,28 @@ public enum DuckDBType
     Varchar,
     // duckdb_blob
     Blob,
+    //decimal
     Decimal,
+    // duckdb_timestamp, in seconds
+    TimestampS,
+    // duckdb_timestamp, in milliseconds
+    TimestampMs,
+    // duckdb_timestamp, in nanoseconds
+    TimestampNs,
+    // enum type, only useful as logical type
+    Enum,
+    // list type, only useful as logical type
+    List,
+    // struct type, only useful as logical type
+    Struct,
+    // map type, only useful as logical type
+    Map,
+    // duckdb_hugeint
+    Uuid,
+    // union type, only useful as logical type
+    Union,
+    // duckdb_bit
+    Bit,
 }
 
 [StructLayout(LayoutKind.Sequential)]
@@ -117,6 +138,13 @@ public void Dispose()
     }
 }
 
+[StructLayout(LayoutKind.Sequential)]
+public struct DuckDBListEntry
+{
+    public ulong Offset { get; }
+    public ulong Length { get; }
+}
+
 [StructLayout(LayoutKind.Sequential)]
 public struct DuckDBHugeInt
 {

diff --git a/DuckDB.NET.Bindings/NativeMethods/NativeMethods.DataChunks.cs b/DuckDB.NET.Bindings/NativeMethods/NativeMethods.DataChunks.cs
@@ -26,13 +26,13 @@ public static class DataChunks
         public static extern unsafe ulong* DuckDBVectorGetValidity(IntPtr vector);
 
         [DllImport(DuckDbLibrary, CallingConvention = CallingConvention.Cdecl, EntryPoint = "duckdb_list_vector_get_child")]
-        public static extern long DuckDBListVectorGetChild(IntPtr vector);
+        public static extern IntPtr DuckDBListVectorGetChild(IntPtr vector);
 
         [DllImport(DuckDbLibrary, CallingConvention = CallingConvention.Cdecl, EntryPoint = "duckdb_list_vector_get_size")]
         public static extern long DuckDBListVectorGetSize(IntPtr vector);
 
         [DllImport(DuckDbLibrary, CallingConvention = CallingConvention.Cdecl, EntryPoint = "duckdb_struct_vector_get_child")]
-        public static extern long DuckDBStructVectorGetChild(IntPtr vector, long index);
+        public static extern IntPtr DuckDBStructVectorGetChild(IntPtr vector, long index);
 
         [DllImport(DuckDbLibrary, CallingConvention = CallingConvention.Cdecl, EntryPoint = "duckdb_destroy_data_chunk")]
         public static extern void DuckDBDestroyDataChunk(out IntPtr chunk);

diff --git a/DuckDB.NET.Bindings/NativeMethods/NativeMethods.LogicalType.cs b/DuckDB.NET.Bindings/NativeMethods/NativeMethods.LogicalType.cs
@@ -19,6 +19,9 @@ public static class LogicalType
         [DllImport(DuckDbLibrary, CallingConvention = CallingConvention.Cdecl, EntryPoint = "duckdb_decimal_internal_type")]
         public static extern DuckDBType DuckDBDecimalInternalType(DuckDBLogicalType type);
 
+        [DllImport(DuckDbLibrary, CallingConvention = CallingConvention.Cdecl, EntryPoint = "duckdb_list_type_child_type")]
+        public static extern DuckDBLogicalType DuckDBListTypeChildType(DuckDBLogicalType type);
+
         [DllImport(DuckDbLibrary, CallingConvention = CallingConvention.Cdecl, EntryPoint = "duckdb_destroy_logical_type")]
         public static extern void DuckDBDestroyLogicalType(out IntPtr type);
     }