Fix issue openwpm#1068: add extra guard, check for store_record_tasks

MohammadMahdiJavid · Nov 22, 2023 · 57adefc · 57adefc
1 parent 553b1f3
commit 57adefc
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/openwpm/storage/storage_controller.py b/openwpm/storage/storage_controller.py
@@ -216,6 +216,14 @@ async def finalize_visit_id(
         self.logger.info("Awaiting all tasks for visit_id %d", visit_id)
         for task in self.store_record_tasks[visit_id]:
             await task
+
+        if visit_id not in self.store_record_tasks:
+            self.logger.error(
+                "visit_id %d, is already awaited, skipping...",
+                visit_id,
+            )
+            return None
+
         del self.store_record_tasks[visit_id]
         self.logger.debug(
             "Awaited all tasks for visit_id %d while finalizing", visit_id