fix: sync map impl

corazawaf · Dec 30, 2024 · cad794a · cad794a
1 parent d5a0d6d
commit cad794a
Showing 1 changed file with 6 additions and 8 deletions.
diff --git a/internal/corazawaf/rule.go b/internal/corazawaf/rule.go
@@ -591,14 +591,12 @@ var transformationNameToID = sync.Map{} // map[string]int
 func transformationID(currentID int, transformationName string) int {
 	currName := transformationIDToName[currentID]
 	nextName := fmt.Sprintf("%s+%s", currName, transformationName)
-	if id, ok := transformationNameToID.Load(nextName); ok {
-		return id.(int)
-	}
-
-	id := len(transformationIDToName)
-	transformationIDToName = append(transformationIDToName, nextName)
-	transformationNameToID.Store(nextName, id)
-	return id
+	id, _ := transformationNameToID.LoadOrStore(nextName, func() interface{} {
+		txid := len(transformationIDToName)
+		transformationIDToName = append(transformationIDToName, nextName)
+		return txid
+	}())
+	return id.(int)
 }
 
 // AddTransformation adds a transformation to the rule