Add a new initilization function for OrderedMap

Signed-off-by: Dušan Mitrović <dmitrovic@nanointeractive.com>

Author: dusnm

types/ordered_map.go

@@ -12,17 +12,38 @@ type (
 	}
 )
 
+// NewOrderedMap
+//
+// Initializes a new instance of the OrderedMap
 func NewOrderedMap[T comparable, V any](length int) OrderedMap[T, V] {
 	return OrderedMap[T, V]{
 		insertionOrder: make([]T, 0, length),
 		values:         make(map[T]V, length),
+		cmpFunc:        nil,
 	}
 }
 
+func NewOrderedMapWithCompareFunc[T comparable, V any](
+	length int,
+	cmpFunc func(a, b T) int,
+) OrderedMap[T, V] {
+	return OrderedMap[T, V]{
+		insertionOrder: make([]T, 0, length),
+		values:         make(map[T]V, length),
+		cmpFunc:        cmpFunc,
+	}
+}
+
+// SetCompareFunc
+//
+// Sets the comparison function to determine ordering.
 func (om *OrderedMap[T, V]) SetCompareFunc(cmpFunc func(a, b T) int) {
 	om.cmpFunc = cmpFunc
 }
 
+// UnsetCompareFunc
+//
+// Unsets the comparison function
 func (om *OrderedMap[T, V]) UnsetCompareFunc() {
 	om.cmpFunc = nil
 }

types/ordered_map_test.go

@@ -93,21 +93,24 @@ func TestOrderedMapIterWithCustomOrder(t *testing.T) {
 
 	assert := require.New(t)
 
-	om := types.NewOrderedMap[string, int](3)
+	om := types.NewOrderedMapWithCompareFunc[string, int](
+		3,
+		func(a, b string) int {
+			if a == b {
+				return 0
+			}
+
+			if a < b {
+				return -1
+			}
+
+			return 1
+		},
+	)
+
 	om.Set("foo", 78)
 	om.Set("var", 90)
 	om.Set("bar", 81)
-	om.SetCompareFunc(func(a, b string) int {
-		if a == b {
-			return 0
-		}
-
-		if a < b {
-			return -1
-		}
-
-		return 1
-	})
 
 	expectedKeys := []string{"bar", "foo", "var"}
 	actualKeys := make([]string, 0, len(expectedKeys))