✨ 新增线程安全共享内存储存器

liteyuki/comm/storage.py

@@ -30,6 +30,13 @@ class KeyValueStore:
         self.passive_chan = Channel(_id="shared_memory-passive")
 
     def set(self, key: str, value: any) -> None:
+        """
+        设置键值对
+        Args:
+            key: 键
+            value: 值
+
+        """
         if IS_MAIN_PROCESS:
             lock = _get_lock(key)
             with lock:
@@ -39,6 +46,15 @@ class KeyValueStore:
             self.passive_chan.send(("set", key, value))
 
     def get(self, key: str, default: Optional[any] = None) -> any:
+        """
+        获取键值对
+        Args:
+            key: 键
+            default: 默认值
+
+        Returns:
+            any: 值
+        """
         if IS_MAIN_PROCESS:
             lock = _get_lock(key)
             with lock:
@@ -47,18 +63,35 @@ class KeyValueStore:
             self.passive_chan.send(("get", key, default))
             return self.active_chan.receive()
 
-    def delete(self, key: str) -> None:
+    def delete(self, key: str, ignore_key_error: bool = True) -> None:
+        """
+        删除键值对
+        Args:
+            key: 键
+            ignore_key_error: 是否忽略键不存在的错误
+
+        Returns:
+        """
         if IS_MAIN_PROCESS:
             lock = _get_lock(key)
             with lock:
                 if key in self._store:
-                    del self._store[key]
-                    del _locks[key]
+                    try:
+                        del self._store[key]
+                        del _locks[key]
+                    except KeyError as e:
+                        if not ignore_key_error:
+                            raise e
         else:
             # 向主进程发送请求删除
             self.passive_chan.send(("delete", key))
 
     def get_all(self) -> dict[str, any]:
+        """
+        获取所有键值对
+        Returns:
+            dict[str, any]: 键值对
+        """
         if IS_MAIN_PROCESS:
             return self._store
         else:
@@ -91,9 +124,7 @@ if IS_MAIN_PROCESS:
 
     @shared_memory.passive_chan.on_receive(lambda d: d[0] == "get")
     def on_get(d):
-        print(shared_memory.get_all())
         shared_memory.active_chan.send(shared_memory.get(d[1], d[2]))
-        print("发送数据：", shared_memory.get(d[1], d[2]))
 
 
     @shared_memory.passive_chan.on_receive(lambda d: d[0] == "set")
@@ -104,6 +135,11 @@ if IS_MAIN_PROCESS:
     @shared_memory.passive_chan.on_receive(lambda d: d[0] == "delete")
     def on_delete(d):
         shared_memory.delete(d[1])
+
+
+    @shared_memory.passive_chan.on_receive(lambda d: d[0] == "get_all")
+    def on_get_all(d):
+        shared_memory.active_chan.send(shared_memory.get_all())
 else:
     shared_memory = None