add Set::to_json

moonbitlang · Oct 12, 2024 · b14746d · b14746d
1 parent 7d69952
commit b14746d
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 0 deletions.
diff --git a/builtin/builtin.mbti b/builtin/builtin.mbti
@@ -330,6 +330,7 @@ impl Set {
   size[K](Self[K]) -> Int
   symmetric_difference[K : Hash + Eq](Self[K], Self[K]) -> Self[K]
   to_array[K](Self[K]) -> Array[K]
+  to_json[X : ToJson](Self[X]) -> Json
   to_string[K : Show](Self[K]) -> String
   union[K : Hash + Eq](Self[K], Self[K]) -> Self[K]
 }

diff --git a/builtin/json.mbt b/builtin/json.mbt
@@ -134,6 +134,14 @@ pub fn Map::to_json[K : Show, V : ToJson](self : Map[K, V]) -> Json {
   Object(object)
 }
 
+pub fn Set::to_json[X : ToJson](self : Set[X]) -> Json {
+  let array = []
+  for v in self {
+    array.push(v.to_json())
+  }
+  Array(array)
+}
+
 pub fn Option::to_json[T : ToJson](self : T?) -> Json {
   match self {
     None => Null

diff --git a/builtin/linked_hash_set_wbtest.mbt → builtin/linked_hash_set_test.mbt b/builtin/linked_hash_set_wbtest.mbt → builtin/linked_hash_set_test.mbt
@@ -146,6 +146,15 @@ test "insert_and_grow" {
   }
   assert_eq!(m.size(), 10)
   assert_eq!(m.capacity(), 16)
+  @json.inspect!(m, content=["0", "1", "2", "3", "4", "5", "6", "7", "8", "9"])
+}
+
+test "array unique via Set" {
+  let v = [1, 2, 3, 4, 5, 3, 2, 4, 5]
+  let h = Set::from_iter(v.iter())
+  let v = [..h]
+  // @json.inspect!([..h])
+  @json.inspect!(v, content=[1, 2, 3, 4, 5])
 }
 
 test "remove_and_shift_back" {