voussoir/voussoirkit
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Use randomized uris for memory databases.

Browse source
This commit is contained in:
voussoir 2023-02-25 14:14:21 -08:00
parent 04817df227
commit bb846a3c97
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
voussoirkit/worms.py
View file

@ -139,6 +139,7 @@ class Database(metaclass=abc.ABCMeta):
# But if your application uses string IDs, set self.id_type = str # But if your application uses string IDs, set self.id_type = str
self.id_type = int self.id_type = int
self.last_commit_id = None self.last_commit_id = None
self._memdb_random = RNG.getrandbits(16)
@abc.abstractmethod @abc.abstractmethod
def _init_column_index(self): def _init_column_index(self):
@ -173,7 +174,7 @@ class Database(metaclass=abc.ABCMeta):
if isinstance(path, pathclass.Path): if isinstance(path, pathclass.Path):
path = path.absolute_path path = path.absolute_path
if path == ':memory:': if path == ':memory:':
sql_read = sqlite3.connect('file:memdb1?mode=memory&cache=shared&mode=ro', uri=True) sql_read = sqlite3.connect(f'file:{self._memdb_random}?mode=memory&cache=shared', uri=True)
sql_read.row_factory = sqlite3.Row sql_read.row_factory = sqlite3.Row
else: else:
log.debug('Connecting to sqlite file "%s".', path) log.debug('Connecting to sqlite file "%s".', path)
@ -186,7 +187,7 @@ class Database(metaclass=abc.ABCMeta):
path = path.absolute_path path = path.absolute_path
if path == ':memory:': if path == ':memory:':
sql_write = sqlite3.connect('file:memdb1?mode=memory&cache=shared', uri=True) sql_write = sqlite3.connect(f'file:{self._memdb_random}?mode=memory&cache=shared', uri=True)
sql_write.row_factory = sqlite3.Row sql_write.row_factory = sqlite3.Row
else: else:
log.debug('Connecting to sqlite file "%s".', path) log.debug('Connecting to sqlite file "%s".', path)