Fixed crash due to concurrent map iteration

Forceu · Jan 11, 2025 · d11a140 · d11a140
1 parent 82e3d9d
commit d11a140
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/internal/storage/processingstatus/pstatusdb/PStatusDb.go b/internal/storage/processingstatus/pstatusdb/PStatusDb.go
@@ -43,13 +43,15 @@ func Set(status models.UploadStatus) {
 func deleteAllExpiredStatus() {
 	allStatus := GetAll()
 	cutOff := time.Now().Add(-24 * time.Hour).Unix()
+	statusMutex.Lock()
+	newStatusMap := make(map[string]models.UploadStatus)
 	for _, status := range allStatus {
-		if status.Creation < cutOff {
-			statusMutex.Lock()
-			delete(statusMap, status.ChunkId)
-			statusMutex.Unlock()
+		if status.Creation > cutOff {
+			newStatusMap[status.ChunkId] = status
 		}
 	}
+	statusMap = newStatusMap
+	statusMutex.Unlock()
 }
 
 func doGarbageCollection(runPeriodically bool) {