use total count only on logs

sonroyaalmerol · Dec 22, 2024 · 19584b9 · 19584b9
1 parent e658d72
commit 19584b9
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/store/concurrency.go b/store/concurrency.go
@@ -90,12 +90,20 @@ func (cm *ConcurrencyManager) CheckConcurrency(m3uIndex string) bool {
 }
 
 func (cm *ConcurrencyManager) UpdateConcurrency(m3uIndex string, subIndex string, incr bool) {
+	cm.mu.Lock()
+	defer cm.mu.Unlock()
+
 	if incr {
 		cm.Increment(m3uIndex, subIndex)
 	} else {
 		cm.Decrement(m3uIndex, subIndex)
 	}
 
-	count := cm.GetCount(m3uIndex, subIndex)
-	utils.SafeLogf("Current number of connections for M3U_%s|%s: %d", m3uIndex, subIndex, count)
+	totalCount := 0
+	for subIndex := range cm.count[m3uIndex] {
+		count := cm.GetCount(m3uIndex, subIndex)
+		totalCount += count
+	}
+
+	utils.SafeLogf("Current number of connections for M3U_%s: %d", m3uIndex, totalCount)
 }