reset database cache

spinn_front_end_common/interface/buffer_management/storage_objects/buffer_database.py

@@ -45,6 +45,15 @@ class BufferDatabase(BaseDatabase):
 
     __slots__ = ()
 
+    @classmethod
+    def reset_db(cls, segment_number: int,
+                 read_only: Optional[bool] = None) -> "BufferDatabase":
+        """
+        Retrieves a NeoBufferDatabase for this segment.
+        """
+        database_file = cls.reset_file(segment_number)
+        return BufferDatabase(database_file, read_only)
+
     def clear_recording_region(
             self, x: int, y: int, p: int, region: int) -> bool:
         """

spinn_front_end_common/interface/provenance/provenance_reader.py

@@ -76,6 +76,15 @@ def __init__(self, provenance_data_path: Optional[str] = None):
         super().__init__(provenance_data_path, read_only=True,
                          row_factory=None, text_factory=None)
 
+    @classmethod
+    def reset_db(cls, segment_number: int,
+                 read_only: Optional[bool] = None) -> "ProvenanceReader":
+        """
+        Retrieves a NeoBufferDatabase for this segment.
+        """
+        database_file = cls.reset_file(segment_number)
+        return ProvenanceReader(database_file, read_only)
+
     def run_query(self, query: str, params: Iterable[_SqliteTypes] = ()
                   ) -> List[Sequence[_SqliteTypes]]:
         """

spinn_front_end_common/utilities/base_database.py

@@ -15,7 +15,7 @@
 import os
 import sqlite3
 import time
-from typing import Optional, Union
+from typing import Dict, Optional, Union
 from typing_extensions import TypeAlias
 from spinn_front_end_common.data import FecDataView
 from spinn_front_end_common.utilities.sqlite_db import SQLiteDB
@@ -25,11 +25,14 @@
 _SECONDS_TO_MICRO_SECONDS_CONVERSION = 1000
 _SqliteTypes: TypeAlias = Union[str, int, float, bytes, None]
 
+reset_cache: Dict[int, str] = {}
+
 
 def _timestamp():
     return int(time.time() * _SECONDS_TO_MICRO_SECONDS_CONVERSION)
 
 
+
 class BaseDatabase(SQLiteDB):
     """
     Specific implementation of the Database for SQLite 3.
@@ -65,6 +68,14 @@ def __init__(self, database_file: Optional[str] = None, *,
         super().__init__(
             self._database_file, read_only=read_only, row_factory=row_factory,
             text_factory=text_factory, ddl_file=_DDL_FILE)
+        reset = FecDataView.get_reset_number()
+        if (reset not in reset_cache or
+                reset_cache[reset] != database_file):
+            reset_cache[reset] = database_file
+
+    @classmethod
+    def reset_file(cls, reset_number) -> str:
+        return reset_cache[reset_number]
 
     @classmethod
     def default_database_file(cls) -> str: