Merge pull request #2 from leroyguillaume/main

Support BTreeSet

README.md

@@ -12,6 +12,7 @@
 ## Support trim
 * `String`
 * `Vec<String>`
+* `BTreeSet<String>`
 * `Option<String>`
 
 
@@ -55,5 +56,21 @@ fn main() {
     let json = r#"{"name":["   ","foo","b ar","hello ","  rust"]}"#;
     let foo = serde_json::from_str::<VecFoo>(json).unwrap();
     assert_eq!(foo.name, vec!["", "foo", "b ar", "hello", "rust"]);
+
+    #[derive(Deserialize)]
+    struct BTreeSetFoo {
+        #[serde(deserialize_with = "btreeset_string_trim")]
+        name: BTreeSet<String>,
+    }
+    let json = r#"{"name":["   ","foo","b ar","hello ","  rust"]}"#;
+    let foo = serde_json::from_str::<BTreeSetFoo>(json).unwrap();
+    let expected: BTreeSet<String> = BTreeSet::from_iter([
+        "".into(),
+        "foo".into(),
+        "b ar".into(),
+        "hello".into(),
+        "rust".into(),
+    ]);
+    assert_eq!(foo.name, expected);
 }
 ```

lib.rs

@@ -1,3 +1,5 @@
+use std::collections::BTreeSet;
+
 use serde::{de, Deserialize};
 pub use trim_in_place::*;
 
@@ -34,6 +36,17 @@ where
     Ok(de_string)
 }
 
+pub fn btreeset_string_trim<'de, D>(d: D) -> Result<BTreeSet<String>, D::Error>
+where
+    D: de::Deserializer<'de>,
+{
+    let de_string: BTreeSet<String> = BTreeSet::<String>::deserialize(d)?
+        .into_iter()
+        .map(|mut x| x.trim_in_place().to_string())
+        .collect();
+    Ok(de_string)
+}
+
 #[test]
 fn test_vec_string_trim() {
     #[derive(Deserialize)]
@@ -80,3 +93,22 @@ fn test_option_string_trim() {
     assert_eq!(foo.name, None);
     assert_eq!(foo.addr, "ABC");
 }
+
+#[test]
+fn test_btreeset_string_trim() {
+    #[derive(Deserialize)]
+    struct BTreeSetFoo {
+        #[serde(deserialize_with = "btreeset_string_trim")]
+        name: BTreeSet<String>,
+    }
+    let json = r#"{"name":["   ","foo","b ar","hello ","  rust"]}"#;
+    let foo = serde_json::from_str::<BTreeSetFoo>(json).unwrap();
+    let expected: BTreeSet<String> = BTreeSet::from_iter([
+        "".into(),
+        "foo".into(),
+        "b ar".into(),
+        "hello".into(),
+        "rust".into(),
+    ]);
+    assert_eq!(foo.name, expected);
+}