brainlessdb 0.7.2__tar.gz → 0.8.0__tar.gz

{brainlessdb-0.7.2 → brainlessdb-0.8.0}/CHANGELOG.md

@@ -3,6 +3,13 @@
 All notable changes to this project will be documented in this file.
 
 
+## [0.8.0] - 2026-02-09
+
+### 🚀 Features
+
+- Replace N+1 bucket.all() with watchall stream and parallelize collection load
+
+
 ## [0.7.2] - 2026-02-08
 
 ### 🐛 Bug Fixes

{brainlessdb-0.7.2 → brainlessdb-0.8.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: brainlessdb
-Version: 0.7.2
+Version: 0.8.0
 Summary: Typed collections backed by NATS JetStream KV
 Author-email: "INSOFT s.r.o." <helpdesk@insoft.cz>
 License-Expression: MIT

{brainlessdb-0.7.2 → brainlessdb-0.8.0}/brainlessdb/__init__.py

@@ -1,6 +1,6 @@
 """Brainless DB - Typed collections backed by NATS JetStream KV."""
 
-__version__ = "0.7.2"
+__version__ = "0.8.0"
 
 from typing import Any, Optional, TypeVar
 

{brainlessdb-0.7.2 → brainlessdb-0.8.0}/brainlessdb/bucket.py

@@ -71,10 +71,18 @@ class Bucket:
 
     async def all(self) -> dict[str, bytes]:
         result = {}
-        for key in await self.keys():
-            value = await self.get(key)
-            if value is not None:
-                result[key] = value
+        watcher = await self._kv.watchall(include_history=True)
+        try:
+            while True:
+                entry = await watcher.updates(timeout=5)
+                if entry is None:
+                    break
+                op = str(entry.operation) if entry.operation else ""
+                is_del = "DEL" in op.upper() or "PURGE" in op.upper()
+                if not is_del and entry.value is not None:
+                    result[entry.key] = entry.value
+        finally:
+            await watcher.stop()
         return result
 
     async def watch(self, include_history: bool = False) -> AsyncIterator[WatchEntry]:

{brainlessdb-0.7.2 → brainlessdb-0.8.0}/brainlessdb/collection.py

@@ -197,6 +197,10 @@ class Collection(Generic[T]):
             self._local_bucket = await self._client.get_bucket(a.local_bucket)
         return True
 
+    @staticmethod
+    async def _load_raw(bucket: Optional[Bucket]) -> dict[str, bytes]:
+        return await bucket.all() if bucket else {}
+
     @staticmethod
     async def _load_bucket(bucket: Optional[Bucket]) -> dict[str, dict]:
         if not bucket:
@@ -337,20 +341,22 @@ class Collection(Generic[T]):
             self._loaded = True
             return 0
 
-        # Load from all buckets
-        config_entries: dict[str, tuple[dict, Any]] = {}
-        if self._config_bucket:
-            for uid, raw in (await self._config_bucket.all()).items():
-                try:
-                    w = msgspec.json.decode(raw, type=ConfigWrapper)
-                except Exception as ex:
-                    _log.error("Failed to decode config for '%s' in '%s': %s", uid, self.name, ex)
-                    continue
-                config_entries[uid] = (w.d, w.m)
-                self._metadata[uid] = w.m
+        # Load all buckets in parallel
+        config_bytes, state, local = await asyncio.gather(
+            self._load_raw(self._config_bucket),
+            self._load_bucket(self._state_bucket),
+            self._load_bucket(self._local_bucket),
+        )
 
-        state = await self._load_bucket(self._state_bucket)
-        local = await self._load_bucket(self._local_bucket)
+        config_entries: dict[str, tuple[dict, Any]] = {}
+        for uid, raw in config_bytes.items():
+            try:
+                w = msgspec.json.decode(raw, type=ConfigWrapper)
+            except Exception as ex:
+                _log.error("Failed to decode config for '%s' in '%s': %s", uid, self.name, ex)
+                continue
+            config_entries[uid] = (w.d, w.m)
+            self._metadata[uid] = w.m
 
         # Build entities
         for uid, (cfg, _) in config_entries.items():

{brainlessdb-0.7.2 → brainlessdb-0.8.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "brainlessdb"
-version = "0.7.2"
+version = "0.8.0"
 description = "Typed collections backed by NATS JetStream KV"
 authors = [
     {name = "INSOFT s.r.o.", email = "helpdesk@insoft.cz"}

{brainlessdb-0.7.2 → brainlessdb-0.8.0}/tests/mock_nats.py

@@ -50,6 +50,10 @@ class MockKV:
     async def watchall(self, include_history: bool = False) -> "MockWatcher":
         q: asyncio.Queue = asyncio.Queue()
         self._watchers.append(q)
+        if include_history:
+            for key, value in self._data.items():
+                await q.put(MockEntry(key=key, value=value, revision=self._revision))
+            await q.put(None)
         return MockWatcher(q, self._watchers)