Pre-allocate slice in generic Map function

PuerkitoBio · Feb 28, 2024 · ce7330a · ce7330a
1 parent 4cbe087
commit ce7330a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/iteration.go b/iteration.go
@@ -37,8 +37,10 @@ func (s *Selection) Map(f func(int, *Selection) string) (result []string) {
 // Map is the generic version of Selection.Map, allowing any type to be
 // returned.
 func Map[E any](s *Selection, f func(int, *Selection) E) (result []E) {
+	result = make([]E, len(s.Nodes))
+
 	for i, n := range s.Nodes {
-		result = append(result, f(i, newSingleSelection(n, s.document)))
+		result[i] = f(i, newSingleSelection(n, s.document))
 	}
 
 	return result