feat(parquet): add support decimal256 read/write in parquet

jorgecarleitao · Feb 21, 2023 · a87ae9d · a87ae9d
1 parent 615300b
commit a87ae9d
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/io/parquet/write/schema.rs b/src/io/parquet/write/schema.rs
@@ -295,6 +295,20 @@ pub fn to_parquet_type(field: &Field) -> Result<ParquetType> {
                 None,
             )?)
         }
+        DataType::Decimal256(precision, scale) => {
+            let precision = *precision;
+            let scale = *scale;
+            let logical_type = Some(PrimitiveLogicalType::Decimal(precision, scale));
+
+            Ok(ParquetType::try_from_primitive(
+                name,
+                PhysicalType::FixedLenByteArray(16),
+                repetition,
+                None,
+                logical_type,
+                None,
+            )?)
+        }
         DataType::Interval(_) => Ok(ParquetType::try_from_primitive(
             name,
             PhysicalType::FixedLenByteArray(12),