From f2172e6743dc97bf9e53a396b2ac4d33ee53683e Mon Sep 17 00:00:00 2001
From: CrazyboyQCD <53971641+CrazyboyQCD@users.noreply.github.com>
Date: Wed, 24 Jan 2024 12:45:06 +0800
Subject: [PATCH] add rust version of sonic to serde-json (#427)

---
 bench/algorithm/json-serde/5-i.rs | 70 +++++++++++++++++++++++++++++++
 bench/bench_rust.yaml             |  1 +
 bench/include/rust/Cargo.toml     |  1 +
 3 files changed, 72 insertions(+)
 create mode 100644 bench/algorithm/json-serde/5-i.rs

diff --git a/bench/algorithm/json-serde/5-i.rs b/bench/algorithm/json-serde/5-i.rs
new file mode 100644
index 00000000..1d9abde7
--- /dev/null
+++ b/bench/algorithm/json-serde/5-i.rs
@@ -0,0 +1,70 @@
+use serde::{Deserialize, Serialize, Serializer};
+use std::fs;
+
+fn main() -> anyhow::Result<()> {
+    let file_name = std::env::args_os()
+        .nth(1)
+        .and_then(|s| s.into_string().ok())
+        .unwrap_or("sample".to_string());
+    let n = std::env::args_os()
+        .nth(2)
+        .and_then(|s| s.into_string().ok())
+        .and_then(|s| s.parse().ok())
+        .unwrap_or(10);
+    let mut json_str = fs::read(format!("{}.json", file_name))?;
+    let json: GeoData = sonic_rs::from_slice(&mut json_str)?;
+
+    print_hash(sonic_rs::to_vec(&json)?);
+    let mut array = Vec::with_capacity(n);
+    for _i in 0..n {
+        let json: GeoData = sonic_rs::from_slice(&mut json_str)?;
+        array.push(json);
+    }
+    print_hash(sonic_rs::to_vec(&array)?);
+    Ok(())
+}
+
+fn print_hash(bytes: impl AsRef<[u8]>) {
+    let digest = md5::compute(&bytes);
+    println!("{:x}", digest);
+}
+
+#[derive(Deserialize, Debug, Default)]
+struct MyF64(f64);
+
+impl Serialize for MyF64 {
+    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
+    where
+        S: Serializer,
+    {
+        if self.0.fract() == 0.0 {
+            serializer.serialize_i64(self.0 as i64)
+        } else {
+            serializer.serialize_f64(self.0)
+        }
+    }
+}
+
+#[derive(Deserialize, Serialize, Debug, Default)]
+struct GeoData {
+    r#type: String,
+    features: Vec<Feature>,
+}
+
+#[derive(Deserialize, Serialize, Debug, Default)]
+struct Feature {
+    r#type: String,
+    properties: Properties,
+    geometry: Geometry,
+}
+
+#[derive(Deserialize, Serialize, Debug, Default)]
+struct Properties {
+    name: String,
+}
+
+#[derive(Deserialize, Serialize, Debug, Default)]
+struct Geometry {
+    r#type: String,
+    coordinates: Vec<Vec<[MyF64; 2]>>,
+}
diff --git a/bench/bench_rust.yaml b/bench/bench_rust.yaml
index bc2d5328..f5889baf 100644
--- a/bench/bench_rust.yaml
+++ b/bench/bench_rust.yaml
@@ -60,6 +60,7 @@ problems:
       - 2.rs
       - 3.rs
       - 4-i.rs
+      - 5-i.rs
   - name: coro-prime-sieve
     source:
       - 1.rs
diff --git a/bench/include/rust/Cargo.toml b/bench/include/rust/Cargo.toml
index 1e0d8b0b..8358097c 100644
--- a/bench/include/rust/Cargo.toml
+++ b/bench/include/rust/Cargo.toml
@@ -43,6 +43,7 @@ regex = "1"
 serde = {version = "1", features = ["derive"]}
 serde_json = {version = "1", features = ["float_roundtrip", "preserve_order"]}
 simd-json = "0.13"
+sonic-rs = "0.3"
 static-rc = "0"
 
 async-channel = {version = "2", optional = true}