From c5c104ce72238e25ac950a41a46974ff8651316a Mon Sep 17 00:00:00 2001
From: jeadie <jack@spice.ai>
Date: Mon, 3 Jun 2024 20:57:13 +1000
Subject: [PATCH] add support for DuckDB arrays when using Arrow's
 FixedSizeList

---
 src/vtab/arrow.rs        | 4 ++--
 src/vtab/logical_type.rs | 9 +++++++++
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/src/vtab/arrow.rs b/src/vtab/arrow.rs
index 3d6c23c2..bde98c5d 100644
--- a/src/vtab/arrow.rs
+++ b/src/vtab/arrow.rs
@@ -196,8 +196,8 @@ pub fn to_duckdb_logical_type(data_type: &DataType) -> Result<LogicalType, Box<d
         Ok(LogicalType::list(&to_duckdb_logical_type(child.data_type())?))
     } else if let DataType::LargeList(child) = data_type {
         Ok(LogicalType::list(&to_duckdb_logical_type(child.data_type())?))
-    } else if let DataType::FixedSizeList(child, _) = data_type {
-        Ok(LogicalType::list(&to_duckdb_logical_type(child.data_type())?))
+    } else if let DataType::FixedSizeList(child, array_size) = data_type {
+        Ok(LogicalType::array(&to_duckdb_logical_type(child.data_type())?, *array_size as u64))
     } else {
         Err(
             format!("Unsupported data type: {data_type}, please file an issue https://github.com/wangfenjin/duckdb-rs")
diff --git a/src/vtab/logical_type.rs b/src/vtab/logical_type.rs
index 76a17182..1ee2543a 100644
--- a/src/vtab/logical_type.rs
+++ b/src/vtab/logical_type.rs
@@ -182,6 +182,15 @@ impl LogicalType {
         }
     }
 
+    /// Creates an array type from its child type.
+    pub fn array(child_type: &LogicalType, array_size: u64) -> Self {
+        unsafe {
+            Self {
+                ptr: duckdb_create_array_type(child_type.ptr, array_size),
+            }
+        }
+    }
+
     /// Creates a decimal type from its `width` and `scale`.
     pub fn decimal(width: u8, scale: u8) -> Self {
         unsafe {