limit fixed

javiorfo · Nov 8, 2024 · cff67de · cff67de
1 parent 4cb4522
commit cff67de
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/list.go b/list.go
@@ -79,10 +79,11 @@ func (list List[T]) FlatMapToAny(mapper func(T) Steam[any]) Steam[any] {
 
 // Limit restricts the number of elements in the List to the specified limit and returns a new List.
 func (list List[T]) Limit(limit int) Steam[T] {
-	results := make(List[T], 0)
-	for i := 0; i < len(list) && i < limit; i++ {
-		results = append(results, list[i])
+	if limit > list.Count() {
+		limit = list.Count()
 	}
+	results := make(List[T], limit)
+	copy(results, list[:limit])
 	return results
 }