From ecaaeb0f33a6d8b97742aee65ae5217ce6901551 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Artjoms=20I=C5=A1kovs?= Date: Wed, 18 Sep 2024 10:46:08 +0100 Subject: [PATCH] Add VARCHAR support --- src/pg_arrow_source.rs | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/pg_arrow_source.rs b/src/pg_arrow_source.rs index a7509ff..8a0979a 100644 --- a/src/pg_arrow_source.rs +++ b/src/pg_arrow_source.rs @@ -92,7 +92,7 @@ impl ArrowBuilder { scale: scale.into(), } } - Type::TEXT => StringBuilder(array::StringBuilder::new()), + Type::TEXT | Type::VARCHAR => StringBuilder(array::StringBuilder::new()), Type::BYTEA => BinaryBuilder(array::BinaryBuilder::new()), _ => panic!("Unsupported type: {}", type_info.pg_type), } @@ -181,7 +181,7 @@ fn pg_type_to_arrow_type(type_info: &PgTypeInfo) -> DataType { numeric_typmod_precision(type_info.type_modifier).try_into().expect("Unsupported precision"), numeric_typmod_scale(type_info.type_modifier).try_into().expect("Unsupported scale"), ), - Type::TEXT => DataType::Utf8, + Type::TEXT | Type::VARCHAR => DataType::Utf8, Type::BYTEA => DataType::Binary, _ => panic!("Unsupported type: {}. Explicitly cast the relevant columns to text in order to store them as strings.", type_info.pg_type), }