fmt

Author: zhuqi-lucas

datafusion/core/src/datasource/file_format/json.rs

@@ -517,7 +517,16 @@ mod tests {
             .await?;
 
         // Scan and read data
-        let exec = scan_format(&ctx, &format, Some(file_schema), tmp_dir.path().to_str().unwrap(), "array.json", None, None).await?;
+        let exec = scan_format(
+            &ctx,
+            &format,
+            Some(file_schema),
+            tmp_dir.path().to_str().unwrap(),
+            "array.json",
+            None,
+            None,
+        )
+        .await?;
         let batches = collect(exec, task_ctx).await?;
 
         assert_eq!(1, batches.len());
@@ -526,7 +535,10 @@ mod tests {
 
         // Verify data
         let array_a = as_int64_array(batches[0].column(0))?;
-        assert_eq!(vec![1, 2, 3], (0..3).map(|i| array_a.value(i)).collect::<Vec<_>>());
+        assert_eq!(
+            vec![1, 2, 3],
+            (0..3).map(|i| array_a.value(i)).collect::<Vec<_>>()
+        );
 
         Ok(())
     }
@@ -540,18 +552,24 @@ mod tests {
 
         let tmp_dir = tempfile::TempDir::new()?;
         let path = format!("{}/array.json", tmp_dir.path().to_string_lossy());
-        std::fs::write(
-            &path,
-            r#"[{"a": 1, "b": "hello"}, {"a": 2, "b": "world"}]"#,
-        )?;
+        std::fs::write(&path, r#"[{"a": 1, "b": "hello"}, {"a": 2, "b": "world"}]"#)?;
 
         let format = JsonFormat::default().with_format_array(true);
         let file_schema = format
             .infer_schema(&ctx, &store, &[local_unpartitioned_file(&path)])
             .await?;
 
         // Project only column "a"
-        let exec = scan_format(&ctx, &format, Some(file_schema), tmp_dir.path().to_str().unwrap(), "array.json", Some(vec![0]), None).await?;
+        let exec = scan_format(
+            &ctx,
+            &format,
+            Some(file_schema),
+            tmp_dir.path().to_str().unwrap(),
+            "array.json",
+            Some(vec![0]),
+            None,
+        )
+        .await?;
         let batches = collect(exec, task_ctx).await?;
 
         assert_eq!(1, batches.len());

datafusion/datasource-json/src/file_format.rs

@@ -211,9 +211,8 @@ fn infer_json_schema_from_json_array<R: Read>(
     })?;
 
     // Parse as JSON array using serde_json
-    let values: Vec<serde_json::Value> = serde_json::from_str(&content).map_err(|e| {
-        ArrowError::JsonError(format!("Failed to parse JSON array: {e}"))
-    })?;
+    let values: Vec<serde_json::Value> = serde_json::from_str(&content)
+        .map_err(|e| ArrowError::JsonError(format!("Failed to parse JSON array: {e}")))?;
 
     // Take only max_records for schema inference
     let values_to_infer: Vec<_> = values.into_iter().take(max_records).collect();