impl arrow_assoc for postgres for vec<bool>

Author: Amar1729

connectorx/src/destinations/arrow2/arrow_assoc.rs

@@ -62,12 +62,12 @@ impl_arrow_assoc!(f64, ArrowDataType::Float64, MutablePrimitiveArray<f64>);
 impl_arrow_assoc!(bool, ArrowDataType::Boolean, MutableBooleanArray);
 
 macro_rules! impl_arrow_assoc_vec {
-    ($T:ty, $AT:expr) => {
+    ($T:ty, $PT:ty, $AT:expr) => {
         impl ArrowAssoc for Vec<$T> {
-            type Builder = MutableListArray<i64, MutablePrimitiveArray<$T>>;
+            type Builder = MutableListArray<i64, $PT>;
 
             fn builder(nrows: usize) -> Self::Builder {
-                MutableListArray::<i64, MutablePrimitiveArray<$T>>::with_capacity(nrows)
+                MutableListArray::<i64, $PT>::with_capacity(nrows)
             }
 
             #[inline]
@@ -86,10 +86,10 @@ macro_rules! impl_arrow_assoc_vec {
         }
 
         impl ArrowAssoc for Option<Vec<$T>> {
-            type Builder = MutableListArray<i64, MutablePrimitiveArray<$T>>;
+            type Builder = MutableListArray<i64, $PT>;
 
             fn builder(nrows: usize) -> Self::Builder {
-                MutableListArray::<i64, MutablePrimitiveArray<$T>>::with_capacity(nrows)
+                MutableListArray::<i64, $PT>::with_capacity(nrows)
             }
 
             #[inline]
@@ -114,12 +114,19 @@ macro_rules! impl_arrow_assoc_vec {
     };
 }
 
-impl_arrow_assoc_vec!(i32, ArrowDataType::Int32);
-impl_arrow_assoc_vec!(i64, ArrowDataType::Int64);
-impl_arrow_assoc_vec!(u32, ArrowDataType::UInt32);
-impl_arrow_assoc_vec!(u64, ArrowDataType::UInt64);
-impl_arrow_assoc_vec!(f32, ArrowDataType::Float32);
-impl_arrow_assoc_vec!(f64, ArrowDataType::Float64);
+macro_rules! impl_arrow_assoc_primitive_vec {
+    ($T:ty, $AT:expr) => {
+        impl_arrow_assoc_vec!($T, MutablePrimitiveArray<$T>, $AT);
+    };
+}
+
+impl_arrow_assoc_vec!(bool, MutableBooleanArray, ArrowDataType::Boolean);
+impl_arrow_assoc_primitive_vec!(i32, ArrowDataType::Int32);
+impl_arrow_assoc_primitive_vec!(i64, ArrowDataType::Int64);
+impl_arrow_assoc_primitive_vec!(u32, ArrowDataType::UInt32);
+impl_arrow_assoc_primitive_vec!(u64, ArrowDataType::UInt64);
+impl_arrow_assoc_primitive_vec!(f32, ArrowDataType::Float32);
+impl_arrow_assoc_primitive_vec!(f64, ArrowDataType::Float64);
 
 impl ArrowAssoc for &str {
     type Builder = MutableUtf8Array<i64>;

connectorx/src/destinations/arrow2/typesystem.rs

@@ -16,6 +16,7 @@ pub enum Arrow2TypeSystem {
     Date64(bool),
     Time64(bool),
     DateTimeTz(bool),
+    BoolArray(bool),
     Int32Array(bool),
     Int64Array(bool),
     UInt32Array(bool),
@@ -41,6 +42,7 @@ impl_typesystem! {
         { Date64     => NaiveDateTime }
         { Time64     => NaiveTime     }
         { DateTimeTz => DateTime<Utc> }
+        { BoolArray  => Vec<bool>     }
         { Int32Array => Vec<i32>      }
         { Int64Array => Vec<i64>      }
         { UInt32Array => Vec<u32>     }

connectorx/src/transports/postgres_arrow2.rs

@@ -61,6 +61,7 @@ macro_rules! impl_postgres_transport {
                 { ByteA[Vec<u8>]                    => LargeBinary[Vec<u8>]        | conversion auto }
                 { JSON[Value]                       => LargeUtf8[String]           | conversion option }
                 { JSONB[Value]                      => LargeUtf8[String]           | conversion none }
+                { BoolArray[Vec<bool>]              => BoolArray[Vec<bool>]        | conversion auto_vec }
                 { Int2Array[Vec<i16>]               => Int64Array[Vec<i64>]        | conversion auto_vec }
                 { Int4Array[Vec<i32>]               => Int64Array[Vec<i64>]        | conversion auto_vec }
                 { Int8Array[Vec<i64>]               => Int64Array[Vec<i64>]        | conversion auto }