fixed a bug where multiple threads tried accessing the db context

cs-23-sw-7-06 · Dec 19, 2023 · e1e4759 · e1e4759
1 parent 8d1420e
commit e1e4759
Showing 1 changed file with 15 additions and 3 deletions.
diff --git a/CentralHub.Api/Services/LocalizationService.cs b/CentralHub.Api/Services/LocalizationService.cs
@@ -44,6 +44,8 @@ public async Task AggregateMeasurementsAsync(CancellationToken stoppingToken)
         var roomMeasurementGroups = await aggregatorRepository.GetRoomMeasurementGroupsAsync(stoppingToken);
         var allRooms = await roomRepository.GetRoomsAsync(stoppingToken);
         using var measuredRoomsMutex = new CancellableMutex<List<RoomDto>>(new List<RoomDto>());
+        using var aggregatedMeasurementsMutex = new CancellableMutex<List<AggregatedMeasurementDto>>(new List<AggregatedMeasurementDto>());
+
 
         await Parallel.ForEachAsync(
             roomMeasurementGroups,
@@ -60,9 +62,8 @@ await Parallel.ForEachAsync(
 
             await measuredRoomsMutex.Lock(m => m.Add(room), cancellationToken);
 
-            await aggregatorRepository.AddAggregatedMeasurementAsync(
-                CreateAggregatedMeasurement(room, roomMeasurementGroup.Value),
-                cancellationToken);
+            var aggregatedMeasurement = CreateAggregatedMeasurement(room, roomMeasurementGroup.Value);
+            await aggregatedMeasurementsMutex.Lock(am => am.Add(aggregatedMeasurement), cancellationToken);
         });
 
         await measuredRoomsMutex.Lock(async m =>
@@ -75,6 +76,17 @@ await aggregatorRepository.AddAggregatedMeasurementAsync(
             }
         },
         stoppingToken);
+
+        await aggregatedMeasurementsMutex.Lock(async am =>
+        {
+            foreach (var aggregatedMeasurement in am)
+            {
+                await aggregatorRepository.AddAggregatedMeasurementAsync(
+                    aggregatedMeasurement,
+                    stoppingToken);
+            }
+        },
+        stoppingToken);
     }
 
     private static AggregatedMeasurementDto CreateAggregatedMeasurement(RoomDto room, IReadOnlyCollection<MeasurementGroup> measurementGroups)