Add map to ScalarValue::try_from as a list

apache · Dec 11, 2023 · 67ea868 · 67ea868
1 parent d091b55
commit 67ea868
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/datafusion/common/src/scalar.rs b/datafusion/common/src/scalar.rs
@@ -2872,6 +2872,14 @@ impl TryFrom<&DataType> for ScalarValue {
                 ))),
                 1,
             )),
+            DataType::Map(field, _) => ScalarValue::List(new_null_array(
+                &DataType::List(Arc::new(Field::new(
+                    "entries",
+                    field.data_type().clone(),
+                    true,
+                ))),
+                1,
+            )),
             DataType::Struct(fields) => ScalarValue::Struct(None, fields.clone()),
             DataType::Null => ScalarValue::Null,
             _ => {
@@ -4330,6 +4338,17 @@ mod tests {
         assert_eq!(scalar, ScalarValue::Utf8(Some("foo".to_string())));
     }
 
+    #[test]
+    fn test_scalar_map(){
+        let key = Field::new("key", DataType::Int64, false);
+        let value = Field::new("value", DataType::Int64, true);
+        let map_field = Field::new("entries", DataType::Struct(vec![key, value].into()), false);
+
+        let scalar = ScalarValue::try_from(DataType::Map(map_field.into(), false)).unwrap();
+
+        assert!(matches!(scalar, ScalarValue::List(_)));
+    }
+
     #[test]
     fn test_scalar_struct() {
         let field_a = Arc::new(Field::new("A", DataType::Int32, false));