Allow initial elements as variadic into NewOrderedMap

elliotchance · Sep 13, 2024 · aed8ebe · aed8ebe
1 parent 9767ae7
commit aed8ebe
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/v2/orderedmap.go b/v2/orderedmap.go
@@ -5,10 +5,14 @@ type OrderedMap[K comparable, V any] struct {
 	ll list[K, V]
 }
 
-func NewOrderedMap[K comparable, V any]() *OrderedMap[K, V] {
-	return &OrderedMap[K, V]{
-		kv: make(map[K]*Element[K, V]),
+func NewOrderedMap[K comparable, V any](els ...*Element[K, V]) *OrderedMap[K, V] {
+	om := &OrderedMap[K, V]{
+		kv: make(map[K]*Element[K, V], len(els)),
 	}
+	for _, el := range els {
+		om.Set(el.Key, el.Value)
+	}
+	return om
 }
 
 // Get returns the value for a key. If the key does not exist, the second return