feat: add put slice (#1084)

Co-authored-by: David Ragot <[email protected]>
formancehq · Dec 27, 2023 · 804743c · 804743c
1 parent 70cd991
commit 804743c
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/libs/go-libs/collectionutils/slice.go b/libs/go-libs/collectionutils/slice.go
@@ -70,15 +70,25 @@ func Contains[T any](slice []T, t T) bool {
 
 type Set[T comparable] map[T]struct{}
 
-func (s Set[T]) Put(t T) {
-	s[t] = struct{}{}
+func (s Set[T]) Put(t ...T) {
+	for _, t2 := range t {
+		s[t2] = struct{}{}
+	}
 }
 
 func (s Set[T]) Contains(t T) bool {
 	_, ok := s[t]
 	return ok
 }
 
+func (s Set[T]) ToSlice() []T {
+	ret := make([]T, 0)
+	for k := range s {
+		ret = append(ret, k)
+	}
+	return ret
+}
+
 func NewSet[T comparable]() Set[T] {
 	return make(Set[T], 0)
 }