impl arrow_assoc for postgres for vec<bool>

Author: Amar1729

connectorx/src/destinations/arrow2/arrow_assoc.rs

@@ -121,6 +121,57 @@ impl_arrow_assoc_vec!(u64, ArrowDataType::UInt64);
 impl_arrow_assoc_vec!(f32, ArrowDataType::Float32);
 impl_arrow_assoc_vec!(f64, ArrowDataType::Float64);
 
+impl ArrowAssoc for Vec<bool> {
+    // type Builder = MutableListArray<i64, MutablePrimitiveArray<i8>>;
+    type Builder = MutableListArray<i64, MutableBooleanArray>;
+
+    fn builder(nrows: usize) -> Self::Builder {
+        MutableListArray::<i64, MutableBooleanArray>::with_capacity(nrows)
+    }
+
+    #[inline]
+    fn push(builder: &mut Self::Builder, value: Self) {
+        // hmm
+        let val: Vec<Option<bool>> = value.into_iter().map(|v| Some(v)).collect();
+        builder.try_push(Some(val)).unwrap();
+    }
+
+    fn field(header: &str) -> Field {
+        Field::new(
+            header,
+            ArrowDataType::LargeList(Box::new(Field::new("", ArrowDataType::Boolean, false))),
+            false,
+        )
+    }
+}
+
+impl ArrowAssoc for Option<Vec<bool>> {
+    type Builder = MutableListArray<i64, MutableBooleanArray>;
+
+    fn builder(nrows: usize) -> Self::Builder {
+        MutableListArray::<i64, MutableBooleanArray>::with_capacity(nrows)
+    }
+
+    #[inline]
+    fn push(builder: &mut Self::Builder, value: Self) {
+        match value {
+            Some(values) => {
+                let val: Vec<Option<bool>> = values.into_iter().map(|v| Some(v)).collect();
+                builder.try_push(Some(val)).unwrap();
+            }
+            None => builder.push_null(),
+        }
+    }
+
+    fn field(header: &str) -> Field {
+        Field::new(
+            header,
+            ArrowDataType::LargeList(Box::new(Field::new("", ArrowDataType::Boolean, false))),
+            true,
+        )
+    }
+}
+
 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 }