Enable independent memory repo

davidlatwe · Jul 22, 2023 · 9c621f1 · 9c621f1
1 parent 8e00af1
commit 9c621f1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/montydb/configure.py b/montydb/configure.py
@@ -247,7 +247,7 @@ def provide_storage(repository):
         A Subclass of `montydb.storage.abcs.AbstractStorage`
 
     """
-    if repository == MEMORY_REPOSITORY:
+    if repository.startswith(MEMORY_REPOSITORY):
         storage_name = MEMORY_STORAGE
         if not memory.is_memory_storage_set():
             set_storage(repository, storage_name)

diff --git a/montydb/storage/memory.py b/montydb/storage/memory.py
@@ -1,6 +1,6 @@
 
 from itertools import islice
-from collections import OrderedDict
+from collections import defaultdict, OrderedDict
 
 from ..types import bson
 from . import (
@@ -13,7 +13,7 @@
 )
 
 
-_repo = OrderedDict()
+_repos = defaultdict(OrderedDict)
 _config = {"_": {}}
 
 
@@ -27,7 +27,7 @@ class MemoryStorage(AbstractStorage):
 
     def __init__(self, repository, storage_config):
         super(MemoryStorage, self).__init__(repository, storage_config)
-        self._repo = _repo
+        self._repo = _repos[repository]
 
     @classmethod
     def nice_name(cls):