fix orderMap set method

projectdiscovery · Jun 28, 2023 · 96f8f2c · 96f8f2c
1 parent da6ed76
commit 96f8f2c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/maps/ordered_map.go b/maps/ordered_map.go
@@ -13,8 +13,10 @@ type OrderedMap[k comparable, v any] struct {
 
 // Set sets a value in the OrderedMap (if the key already exists, it will be overwritten)
 func (o *OrderedMap[k, v]) Set(key k, value v) {
+	if _, ok := o.m[key]; !ok {
+		o.keys = append(o.keys, key)
+	}
 	o.m[key] = value
-	o.keys = append(o.keys, key)
 }
 
 // Get gets a value from the OrderedMap