add argument 'capacity~' for @hashset.new

moonbitlang · Jan 14, 2025 · 19ebaa5 · 19ebaa5
1 parent 639ffb2
commit 19ebaa5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/hashset/hashset.mbt b/hashset/hashset.mbt
@@ -18,10 +18,10 @@ let default_init_capacity = 8
 
 ///|
 /// Create new hash set.
-pub fn T::new[K]() -> T[K] {
+pub fn T::new[K](capacity~ : Int = default_init_capacity) -> T[K] {
   {
     size: 0,
-    capacity: default_init_capacity,
+    capacity,
     growAt: calc_grow_threshold(default_init_capacity),
     entries: FixedArray::make(default_init_capacity, None),
   }

diff --git a/hashset/hashset.mbti b/hashset/hashset.mbti
@@ -20,7 +20,7 @@ impl T {
   intersection[K : Hash + Eq](Self[K], Self[K]) -> Self[K]
   is_empty[K](Self[K]) -> Bool
   iter[K](Self[K]) -> Iter[K]
-  new[K]() -> Self[K]
+  new[K](capacity~ : Int = ..) -> Self[K]
   of[K : Hash + Eq](FixedArray[K]) -> Self[K]
   remove[K : Hash + Eq](Self[K], K) -> Unit
   size[K](Self[K]) -> Int