wool 0.1rc8__tar.gz → 0.1rc9__tar.gz

{wool-0.1rc8 → wool-0.1rc9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wool
-Version: 0.1rc8
+Version: 0.1rc9
 Summary: A Python framework for distributed multiprocessing.
 Author-email: Conrad Bzura <conrad@wool.io>
 Maintainer-email: maintainers@wool.io
@@ -221,9 +221,10 @@ Provides-Extra: dev
 Requires-Dist: pytest; extra == 'dev'
 Requires-Dist: pytest-asyncio; extra == 'dev'
 Requires-Dist: pytest-grpc-aio~=0.2.0; extra == 'dev'
+Requires-Dist: pytest-mock; extra == 'dev'
 Requires-Dist: ruff; extra == 'dev'
 Provides-Extra: locking
-Requires-Dist: wool-locking==0.1rc8; extra == 'locking'
+Requires-Dist: wool-locking==0.1rc9; extra == 'locking'
 Description-Content-Type: text/markdown
 
 # Wool

{wool-0.1rc8 → wool-0.1rc9}/pyproject.toml

@@ -29,7 +29,13 @@ readme = "README.md"
 requires-python = ">=3.10"
 
 [project.optional-dependencies]
-dev = ["pytest", "pytest-asyncio", "pytest-grpc-aio~=0.2.0", "ruff"]
+dev = [
+    "pytest",
+    "pytest-asyncio",
+    "pytest-grpc-aio~=0.2.0",
+    "pytest-mock",
+    "ruff",
+]
 locking = ["wool-locking"]
 
 [project.scripts]

wool-0.1rc9/wool/_mempool/_client.py

@@ -0,0 +1,167 @@
+from __future__ import annotations
+
+import asyncio
+from typing import Final
+from typing import Protocol
+
+try:
+    from typing import Self
+except ImportError:
+    from typing_extensions import Self
+
+import grpc
+
+try:
+    from wool._protobuf.mempool import service_pb2 as pb
+    from wool._protobuf.mempool import service_pb2_grpc as rpc
+except ImportError as e:
+    from wool._protobuf import ProtobufImportError
+
+    raise ProtobufImportError(e) from e
+
+
+class EventHandler(Protocol):
+    @staticmethod
+    def __call__(client: MemoryPoolClient, response: pb.Event) -> None: ...
+
+
+class MemoryPoolClient:
+    _event_stream: Final[grpc.aio.UnaryStreamCall]
+    _listener: asyncio.Task
+    _session: pb.Session | None
+
+    def __init__(self, channel: grpc.aio.Channel, event_handler: EventHandler):
+        self._stub = rpc.MemoryPoolStub(channel)
+        self._event_stream = self._stub.session(pb.SessionRequest())
+        self._event_handler = event_handler
+        self._session = None
+
+    async def __aenter__(self) -> Self:
+        if self._session:
+            raise RuntimeError(
+                "Client session may not be re-entered. "
+                "Use 'async with MemoryPoolClient(...)' only once."
+            )
+        response = await self._event_stream.read()
+        assert isinstance(response, pb.SessionResponse)
+        self._session = response.session
+        self._listener = asyncio.create_task(self._listen())
+        return self
+
+    async def __aexit__(self, *_):
+        if not self._session:
+            raise RuntimeError(
+                "Client session has not been entered. "
+                "Use 'async with MemoryPoolClient(...)'."
+            )
+        assert self._listener
+        self._listener.cancel()
+        try:
+            await self._listener
+        except asyncio.CancelledError:
+            pass
+
+    def __del__(self):
+        try:
+            self._event_stream.cancel()
+        except Exception:
+            pass
+
+    @property
+    def id(self) -> str:
+        if not self._session:
+            raise RuntimeError(
+                "Client session has not been entered. "
+                "Use 'async with MemoryPoolClient(...)'."
+            )
+        return self._session.id
+
+    async def map(self, ref: str):
+        if not self._session:
+            raise RuntimeError(
+                "Client session has not been entered. "
+                "Use 'async with MemoryPoolClient(...)'."
+            )
+        request = pb.AcquireRequest(
+            session=self._session,
+            reference=pb.Reference(id=ref),
+        )
+        await self._stub.map(request)
+
+    async def get(self, ref: str) -> bytes:
+        if not self._session:
+            raise RuntimeError(
+                "Client session has not been entered. "
+                "Use 'async with MemoryPoolClient(...)'."
+            )
+        request = pb.GetRequest(
+            session=self._session,
+            reference=pb.Reference(id=ref),
+        )
+        response: pb.GetResponse = await self._stub.get(request)
+        return response.dump
+
+    async def put(self, dump: bytes, *, mutable: bool = False) -> str:
+        if not self._session:
+            raise RuntimeError(
+                "Client session has not been entered. "
+                "Use 'async with MemoryPoolClient(...)'."
+            )
+        request = pb.PutRequest(
+            session=self._session,
+            dump=dump,
+            mutable=mutable,
+        )
+        response: pb.PutResponse = await self._stub.put(request)
+        return response.reference.id
+
+    async def post(self, ref: str, dump: bytes) -> bool:
+        if not self._session:
+            raise RuntimeError(
+                "Client session has not been entered. "
+                "Use 'async with MemoryPoolClient(...)'."
+            )
+        request = pb.PostRequest(
+            session=self._session,
+            reference=pb.Reference(id=ref),
+            dump=dump,
+        )
+        response: pb.PostResponse = await self._stub.post(request)
+        return response.updated
+
+    async def acquire(self, ref: str):
+        if not self._session:
+            raise RuntimeError(
+                "Client session has not been entered. "
+                "Use 'async with MemoryPoolClient(...)'."
+            )
+        request = pb.AcquireRequest(
+            session=self._session,
+            reference=pb.Reference(id=ref),
+        )
+        await self._stub.acquire(request)
+
+    async def release(self, ref: str):
+        if not self._session:
+            raise RuntimeError(
+                "Client session has not been entered. "
+                "Use 'async with MemoryPoolClient(...)'."
+            )
+        request = pb.ReleaseRequest(
+            session=self._session,
+            reference=pb.Reference(id=ref),
+        )
+        await self._stub.release(request)
+
+    async def _listen(self):
+        assert self._event_stream
+        try:
+            while True:
+                if (response := await self._event_stream.read()) is None:
+                    break
+                assert isinstance(response, pb.SessionResponse), (
+                    f"Unexpected event type: {type(response)}"
+                )
+                self._event_handler(self, response.event)
+        except asyncio.CancelledError:
+            self._event_stream.cancel()

{wool-0.1rc8 → wool-0.1rc9}/wool/_mempool/_mempool.py

@@ -2,11 +2,11 @@ from __future__ import annotations
 
 import asyncio
 import hashlib
-import mmap
 import os
 import pathlib
 import shutil
 from contextlib import asynccontextmanager
+from mmap import mmap
 from typing import BinaryIO
 
 try:
@@ -23,7 +23,7 @@ class SharedObject:
     _id: str
     _mempool: MemoryPool
     _file: BinaryIO
-    _mmap: mmap.mmap
+    _mmap: mmap
     _size: int
     _md5: bytes
 
@@ -31,7 +31,7 @@ class SharedObject:
         self._id = id
         self._mempool = mempool
         self._file = open(self._path / "dump", "r+b")
-        self._mmap = mmap.mmap(self._file.fileno(), 0)
+        self._mmap = mmap(self._file.fileno(), 0)
         self._size = self.metadata.size
         self._md5 = self.metadata.md5
 
@@ -50,7 +50,7 @@ class SharedObject:
         return SharedObjectMetadata(self.id, mempool=self._mempool)
 
     @property
-    def mmap(self) -> mmap.mmap:
+    def mmap(self) -> mmap:
         return self._mmap
 
     @property
@@ -67,7 +67,7 @@ class SharedObject:
             self._mmap.close()
             self._file.close()
             self._file = open(self._path / "dump", "r+b")
-            self._mmap = mmap.mmap(self._file.fileno(), 0)
+            self._mmap = mmap(self._file.fileno(), 0)
             self._size = self.metadata.size
             self._md5 = self.metadata.md5
         return self
@@ -77,7 +77,7 @@ class SharedObjectMetadata:
     _id: str
     _mempool: MemoryPool
     _file: BinaryIO
-    _mmap: mmap.mmap
+    _mmap: mmap
     _instances: dict[str, SharedObjectMetadata] = {}
 
     def __new__(cls, id: str, *, mempool: MemoryPool):
@@ -89,7 +89,7 @@ class SharedObjectMetadata:
         self._id = id
         self._mempool = mempool
         self._file = open(self._path / "meta", "r+b")
-        self._mmap = mmap.mmap(self._file.fileno(), 0)
+        self._mmap = mmap(self._file.fileno(), 0)
         self._instances[id] = self
 
     def __del__(self):
@@ -115,7 +115,7 @@ class SharedObjectMetadata:
         return self._metadata.md5
 
     @property
-    def mmap(self) -> mmap.mmap:
+    def mmap(self) -> mmap:
         return self._mmap
 
     @property

{wool-0.1rc8 → wool-0.1rc9}/wool/_mempool/_metadata.py

@@ -1,16 +1,14 @@
 from __future__ import annotations
 
-import logging
+from dataclasses import asdict
 from dataclasses import dataclass
 
 try:
-    from wool._protobuf.mempool.metadata.metadata_pb2 import _MetadataMessage
-except ImportError:
-    logging.error(
-        "Failed to import _MetadataMessage. "
-        "Ensure protocol buffers are compiled."
-    )
-    raise
+    from wool._protobuf.mempool import metadata_pb2 as pb
+except ImportError as e:
+    from wool._protobuf import ProtobufImportError
+
+    raise ProtobufImportError(e) from e
 
 
 @dataclass
@@ -22,7 +20,7 @@ class MetadataMessage:
 
     @classmethod
     def loads(cls, data: bytes) -> MetadataMessage:
-        (metadata := _MetadataMessage()).ParseFromString(data)
+        (metadata := pb.MetadataMessage()).ParseFromString(data)
         return cls(
             ref=metadata.ref,
             mutable=metadata.mutable,
@@ -31,9 +29,7 @@ class MetadataMessage:
         )
 
     def dumps(self) -> bytes:
-        return _MetadataMessage(
-            ref=self.ref, mutable=self.mutable, size=self.size, md5=self.md5
-        ).SerializeToString()
+        return pb.MetadataMessage(**asdict(self)).SerializeToString()
 
 
 __all__ = ["MetadataMessage"]

wool-0.1rc9/wool/_mempool/_service.py

@@ -0,0 +1,227 @@
+from __future__ import annotations
+
+import asyncio
+from typing import AsyncGenerator
+from typing import Final
+from weakref import WeakSet
+from weakref import WeakValueDictionary
+
+import shortuuid
+from grpc.aio import ServicerContext
+
+from wool._mempool import MemoryPool
+
+try:
+    from wool._protobuf.mempool import service_pb2 as pb
+    from wool._protobuf.mempool import service_pb2_grpc as rpc
+except ImportError as e:
+    from wool._protobuf import ProtobufImportError
+
+    raise ProtobufImportError(e) from e
+
+
+class _Session:
+    id: Final[str]
+    queue: Final[asyncio.Queue[pb.SessionResponse]]
+    references: Final[set[_Reference]]
+
+    _sessions: Final[WeakValueDictionary[str, _Session]] = (
+        WeakValueDictionary()
+    )
+
+    @classmethod
+    def get(cls, id: str) -> _Session | None:
+        return cls._sessions.get(id)
+
+    def __init__(self):
+        self.id = shortuuid.uuid()
+        self.queue = asyncio.Queue()
+        self.references = set()
+        self._sessions[self.id] = self
+
+    def __eq__(self, other) -> bool:
+        if isinstance(other, _Session):
+            return self.id == other.id
+        return False
+
+    def __hash__(self) -> int:
+        return hash(self.id)
+
+
+class _Reference:
+    id: Final[str]
+    sessions: Final[WeakSet[_Session]]
+
+    _mempool: Final[MemoryPool]
+    _references: Final[WeakValueDictionary[str, _Reference]] = (
+        WeakValueDictionary()
+    )
+    _to_delete: Final[set[str]] = set()
+    _initialized: bool = False
+
+    @classmethod
+    def get(cls, id: str) -> _Reference | None:
+        return cls._references.get(id)
+
+    @classmethod
+    def new(cls, id: str, *, mempool: MemoryPool) -> _Reference:
+        if id in cls._references:
+            raise ValueError(f"Reference {id} already exists")
+        return cls(id, mempool=mempool)
+
+    def __new__(cls, id: str, *, mempool: MemoryPool):
+        if id in cls._references:
+            if id in cls._to_delete:
+                cls._to_delete.remove(id)
+            return cls._references[id]
+        return super().__new__(cls)
+
+    def __init__(self, id: str, *, mempool: MemoryPool):
+        if not self._initialized:
+            self.id = id
+            self.sessions = WeakSet()
+            self._mempool = mempool
+            self._references[id] = self
+            self._initialized = True
+
+    def __eq__(self, other) -> bool:
+        if isinstance(other, _Reference):
+            return self.id == other.id
+        return False
+
+    def __hash__(self) -> int:
+        return hash(self.id)
+
+    def __del__(self):
+        self._to_delete.add(self.id)
+
+        async def _delete(id, to_delete, mempool):
+            if id in to_delete:
+                try:
+                    to_delete.remove(id)
+                    await mempool.delete(id)
+                except FileNotFoundError:
+                    pass
+
+        try:
+            asyncio.get_running_loop().create_task(
+                _delete(self.id, self._to_delete, self._mempool)
+            )
+        except RuntimeError:
+            asyncio.new_event_loop().run_until_complete(
+                _delete(self.id, self._to_delete, self._mempool)
+            )
+
+
+class MemoryPoolService(rpc.MemoryPoolServicer):
+    def __init__(self, mempool: MemoryPool | None = None):
+        self._mempool = mempool or MemoryPool()
+        self._shutdown = asyncio.Event()
+
+    async def session(
+        self, request: pb.SessionRequest, context: ServicerContext
+    ) -> AsyncGenerator[pb.SessionResponse]:
+        session = _Session()
+        yield pb.SessionResponse(session=pb.Session(id=session.id))
+        while True:
+            yield await session.queue.get()
+
+    async def map(
+        self, request: pb.AcquireRequest, context: ServicerContext
+    ) -> pb.AcquireResponse:
+        if not (session := _Session.get(request.session.id)):
+            raise ValueError(f"Session {request.session.id} not found")
+        await self._mempool.map(request.reference.id or None)
+        reference = _Reference(request.reference.id, mempool=self._mempool)
+        await self.acquire(
+            pb.AcquireRequest(
+                session=pb.Session(id=session.id),
+                reference=pb.Reference(id=reference.id),
+            ),
+            context,
+        )
+        return pb.AcquireResponse()
+
+    async def put(
+        self, request: pb.PutRequest, context: ServicerContext
+    ) -> pb.PutResponse:
+        if not (session := _Session.get(request.session.id)):
+            raise ValueError(f"Session {request.session.id} not found")
+        reference = _Reference(
+            id=await self._mempool.put(request.dump, mutable=request.mutable),
+            mempool=self._mempool,
+        )
+        await self.acquire(
+            pb.AcquireRequest(
+                session=pb.Session(id=session.id),
+                reference=pb.Reference(id=reference.id),
+            ),
+            context,
+        )
+        return pb.PutResponse(reference=pb.Reference(id=reference.id))
+
+    async def get(
+        self, request: pb.GetRequest, context: ServicerContext
+    ) -> pb.GetResponse:
+        if not (session := _Session.get(request.session.id)):
+            raise ValueError(f"Session {request.session.id} not found")
+        if _Reference.get(request.reference.id) is None:
+            await self.acquire(
+                pb.AcquireRequest(
+                    session=pb.Session(id=session.id),
+                    reference=pb.Reference(id=request.reference.id),
+                ),
+                context,
+            )
+        dump = await self._mempool.get(request.reference.id)
+        return pb.GetResponse(dump=dump)
+
+    async def post(
+        self, request: pb.PostRequest, context: ServicerContext
+    ) -> pb.PostResponse:
+        if not (session := _Session.get(request.session.id)):
+            raise ValueError(f"Session {request.session.id} not found")
+        if not (reference := _Reference.get(request.reference.id)):
+            raise ValueError(f"Reference {request.reference.id} not found")
+        if reference not in session.references:
+            await self.acquire(
+                pb.AcquireRequest(
+                    session=pb.Session(id=session.id),
+                    reference=pb.Reference(id=reference.id),
+                ),
+                context,
+            )
+        updated = await self._mempool.post(request.reference.id, request.dump)
+        if updated:
+            for session in _Reference(
+                id=request.reference.id, mempool=self._mempool
+            ).sessions:
+                if session.id is not request.session.id:
+                    event = pb.Event(
+                        reference=request.reference,
+                        event_type="post",
+                    )
+                    await session.queue.put(pb.SessionResponse(event=event))
+        return pb.PostResponse(updated=updated)
+
+    async def acquire(
+        self, request: pb.AcquireRequest, context: ServicerContext
+    ) -> pb.AcquireResponse:
+        if not (session := _Session.get(request.session.id)):
+            raise ValueError(f"Session {request.session.id} not found")
+        if not (reference := _Reference.get(request.reference.id)):
+            raise ValueError(f"Reference {request.reference.id} not found")
+        session.references.add(reference)
+        reference.sessions.add(session)
+        return pb.AcquireResponse()
+
+    async def release(
+        self, request: pb.ReleaseRequest, context: ServicerContext
+    ) -> pb.ReleaseResponse:
+        if not (session := _Session.get(request.session.id)):
+            raise ValueError(f"Session {request.session.id} not found")
+        if not (reference := _Reference.get(request.reference.id)):
+            raise ValueError(f"Reference {request.reference.id} not found")
+        session.references.remove(reference)
+        reference.sessions.remove(session)
+        return pb.ReleaseResponse()

wool-0.1rc9/wool/_protobuf/__init__.py

@@ -0,0 +1,11 @@
+import os
+import sys
+
+sys.path.insert(0, os.path.abspath(os.path.dirname(__file__)))
+
+
+class ProtobufImportError(ImportError):
+    def __init__(self, exception: ImportError):
+        super().__init__(
+            f"{str(exception)} - ensure protocol buffers are compiled."
+        )

{wool-0.1rc8/wool/_protobuf/mempool/metadata → wool-0.1rc9/wool/_protobuf/mempool}/metadata_pb2.py

@@ -1,7 +1,7 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # NO CHECKED-IN PROTOBUF GENCODE
-# source: mempool/metadata/metadata.proto
+# source: mempool/metadata.proto
 # Protobuf Python Version: 6.31.0
 """Generated protocol buffer code."""
 from google.protobuf import descriptor as _descriptor
@@ -15,7 +15,7 @@ _runtime_version.ValidateProtobufRuntimeVersion(
     31,
     0,
     '',
-    'mempool/metadata/metadata.proto'
+    'mempool/metadata.proto'
 )
 # @@protoc_insertion_point(imports)
 
@@ -24,13 +24,13 @@ _sym_db = _symbol_database.Default()
 
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1fmempool/metadata/metadata.proto\x12\x16wool._mempool.metadata\"K\n\x10_MetadataMessage\x12\x0b\n\x03ref\x18\x01 \x01(\t\x12\x0f\n\x07mutable\x18\x02 \x01(\x08\x12\x0c\n\x04size\x18\x03 \x01(\x03\x12\x0b\n\x03md5\x18\x04 \x01(\x0c\x62\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x16mempool/metadata.proto\x12\x16wool._protobuf.mempool\"J\n\x0fMetadataMessage\x12\x0b\n\x03ref\x18\x01 \x01(\t\x12\x0f\n\x07mutable\x18\x02 \x01(\x08\x12\x0c\n\x04size\x18\x03 \x01(\x03\x12\x0b\n\x03md5\x18\x04 \x01(\x0c\x62\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
-_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mempool.metadata.metadata_pb2', _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mempool.metadata_pb2', _globals)
 if not _descriptor._USE_C_DESCRIPTORS:
   DESCRIPTOR._loaded_options = None
-  _globals['__METADATAMESSAGE']._serialized_start=59
-  _globals['__METADATAMESSAGE']._serialized_end=134
+  _globals['_METADATAMESSAGE']._serialized_start=50
+  _globals['_METADATAMESSAGE']._serialized_end=124
 # @@protoc_insertion_point(module_scope)

{wool-0.1rc8/wool/_protobuf/mempool/metadata → wool-0.1rc9/wool/_protobuf/mempool}/metadata_pb2.pyi

@@ -4,7 +4,7 @@ from typing import ClassVar as _ClassVar, Optional as _Optional
 
 DESCRIPTOR: _descriptor.FileDescriptor
 
-class _MetadataMessage(_message.Message):
+class MetadataMessage(_message.Message):
     __slots__ = ("ref", "mutable", "size", "md5")
     REF_FIELD_NUMBER: _ClassVar[int]
     MUTABLE_FIELD_NUMBER: _ClassVar[int]

{wool-0.1rc8/wool/_protobuf/mempool/metadata → wool-0.1rc9/wool/_protobuf/mempool}/metadata_pb2_grpc.py

@@ -17,7 +17,7 @@ except ImportError:
 if _version_not_supported:
     raise RuntimeError(
         f'The grpc package installed is at version {GRPC_VERSION},'
-        + f' but the generated code in mempool/metadata/metadata_pb2_grpc.py depends on'
+        + f' but the generated code in mempool/metadata_pb2_grpc.py depends on'
         + f' grpcio>={GRPC_GENERATED_VERSION}.'
         + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
         + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'

wool-0.1rc8/wool/_protobuf/mempool/mempool_pb2.py → wool-0.1rc9/wool/_protobuf/mempool/service_pb2.py

@@ -1,7 +1,7 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # NO CHECKED-IN PROTOBUF GENCODE
-# source: mempool/mempool.proto
+# source: mempool/service.proto
 # Protobuf Python Version: 6.31.0
 """Generated protocol buffer code."""
 from google.protobuf import descriptor as _descriptor
@@ -15,7 +15,7 @@ _runtime_version.ValidateProtobufRuntimeVersion(
     31,
     0,
     '',
-    'mempool/mempool.proto'
+    'mempool/service.proto'
 )
 # @@protoc_insertion_point(imports)
 
@@ -24,11 +24,11 @@ _sym_db = _symbol_database.Default()
 
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x15mempool/mempool.proto\x12\x16wool._protobuf.mempool\"\x10\n\x0eSessionRequest\"\x81\x01\n\x0fSessionResponse\x12\x32\n\x07session\x18\x01 \x01(\x0b\x32\x1f.wool._protobuf.mempool.SessionH\x00\x12.\n\x05\x65vent\x18\x02 \x01(\x0b\x32\x1d.wool._protobuf.mempool.EventH\x00\x42\n\n\x08response\"x\n\x0e\x41\x63quireRequest\x12\x34\n\treference\x18\x01 \x01(\x0b\x32!.wool._protobuf.mempool.Reference\x12\x30\n\x07session\x18\x02 \x01(\x0b\x32\x1f.wool._protobuf.mempool.Session\"\x11\n\x0f\x41\x63quireResponse\"]\n\nPutRequest\x12\x30\n\x07session\x18\x01 \x01(\x0b\x32\x1f.wool._protobuf.mempool.Session\x12\x0f\n\x07mutable\x18\x02 \x01(\x08\x12\x0c\n\x04\x64ump\x18\x03 \x01(\x0c\"C\n\x0bPutResponse\x12\x34\n\treference\x18\x01 \x01(\x0b\x32!.wool._protobuf.mempool.Reference\"\x83\x01\n\x0bPostRequest\x12\x30\n\x07session\x18\x01 \x01(\x0b\x32\x1f.wool._protobuf.mempool.Session\x12\x34\n\treference\x18\x02 \x01(\x0b\x32!.wool._protobuf.mempool.Reference\x12\x0c\n\x04\x64ump\x18\x03 \x01(\x0c\"\x1f\n\x0cPostResponse\x12\x0f\n\x07updated\x18\x01 \x01(\x08\"t\n\nGetRequest\x12\x34\n\treference\x18\x01 \x01(\x0b\x32!.wool._protobuf.mempool.Reference\x12\x30\n\x07session\x18\x02 \x01(\x0b\x32\x1f.wool._protobuf.mempool.Session\"\x1b\n\x0bGetResponse\x12\x0c\n\x04\x64ump\x18\x01 \x01(\x0c\"x\n\x0eReleaseRequest\x12\x34\n\treference\x18\x01 \x01(\x0b\x32!.wool._protobuf.mempool.Reference\x12\x30\n\x07session\x18\x02 \x01(\x0b\x32\x1f.wool._protobuf.mempool.Session\"\x11\n\x0fReleaseResponse\"\x17\n\tReference\x12\n\n\x02id\x18\x01 \x01(\t\"\x15\n\x07Session\x12\n\n\x02id\x18\x01 \x01(\t\"Q\n\x05\x45vent\x12\x34\n\treference\x18\x01 \x01(\x0b\x32!.wool._protobuf.mempool.Reference\x12\x12\n\nevent_type\x18\x02 \x01(\t2\x95\x04\n\nMemoryPool\x12\\\n\x07session\x12&.wool._protobuf.mempool.SessionRequest\x1a\'.wool._protobuf.mempool.SessionResponse0\x01\x12Z\n\x07\x61\x63quire\x12&.wool._protobuf.mempool.AcquireRequest\x1a\'.wool._protobuf.mempool.AcquireResponse\x12N\n\x03put\x12\".wool._protobuf.mempool.PutRequest\x1a#.wool._protobuf.mempool.PutResponse\x12Q\n\x04post\x12#.wool._protobuf.mempool.PostRequest\x1a$.wool._protobuf.mempool.PostResponse\x12N\n\x03get\x12\".wool._protobuf.mempool.GetRequest\x1a#.wool._protobuf.mempool.GetResponse\x12Z\n\x07release\x12&.wool._protobuf.mempool.ReleaseRequest\x1a\'.wool._protobuf.mempool.ReleaseResponseb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x15mempool/service.proto\x12\x16wool._protobuf.mempool\"\x10\n\x0eSessionRequest\"\x81\x01\n\x0fSessionResponse\x12\x32\n\x07session\x18\x01 \x01(\x0b\x32\x1f.wool._protobuf.mempool.SessionH\x00\x12.\n\x05\x65vent\x18\x02 \x01(\x0b\x32\x1d.wool._protobuf.mempool.EventH\x00\x42\n\n\x08response\"x\n\x0e\x41\x63quireRequest\x12\x34\n\treference\x18\x01 \x01(\x0b\x32!.wool._protobuf.mempool.Reference\x12\x30\n\x07session\x18\x02 \x01(\x0b\x32\x1f.wool._protobuf.mempool.Session\"\x11\n\x0f\x41\x63quireResponse\"]\n\nPutRequest\x12\x30\n\x07session\x18\x01 \x01(\x0b\x32\x1f.wool._protobuf.mempool.Session\x12\x0f\n\x07mutable\x18\x02 \x01(\x08\x12\x0c\n\x04\x64ump\x18\x03 \x01(\x0c\"C\n\x0bPutResponse\x12\x34\n\treference\x18\x01 \x01(\x0b\x32!.wool._protobuf.mempool.Reference\"\x83\x01\n\x0bPostRequest\x12\x30\n\x07session\x18\x01 \x01(\x0b\x32\x1f.wool._protobuf.mempool.Session\x12\x34\n\treference\x18\x02 \x01(\x0b\x32!.wool._protobuf.mempool.Reference\x12\x0c\n\x04\x64ump\x18\x03 \x01(\x0c\"\x1f\n\x0cPostResponse\x12\x0f\n\x07updated\x18\x01 \x01(\x08\"t\n\nGetRequest\x12\x34\n\treference\x18\x01 \x01(\x0b\x32!.wool._protobuf.mempool.Reference\x12\x30\n\x07session\x18\x02 \x01(\x0b\x32\x1f.wool._protobuf.mempool.Session\"\x1b\n\x0bGetResponse\x12\x0c\n\x04\x64ump\x18\x01 \x01(\x0c\"x\n\x0eReleaseRequest\x12\x34\n\treference\x18\x01 \x01(\x0b\x32!.wool._protobuf.mempool.Reference\x12\x30\n\x07session\x18\x02 \x01(\x0b\x32\x1f.wool._protobuf.mempool.Session\"\x11\n\x0fReleaseResponse\"\x17\n\tReference\x12\n\n\x02id\x18\x01 \x01(\t\"\x15\n\x07Session\x12\n\n\x02id\x18\x01 \x01(\t\"Q\n\x05\x45vent\x12\x34\n\treference\x18\x01 \x01(\x0b\x32!.wool._protobuf.mempool.Reference\x12\x12\n\nevent_type\x18\x02 \x01(\t2\xed\x04\n\nMemoryPool\x12\\\n\x07session\x12&.wool._protobuf.mempool.SessionRequest\x1a\'.wool._protobuf.mempool.SessionResponse0\x01\x12Z\n\x07\x61\x63quire\x12&.wool._protobuf.mempool.AcquireRequest\x1a\'.wool._protobuf.mempool.AcquireResponse\x12V\n\x03map\x12&.wool._protobuf.mempool.AcquireRequest\x1a\'.wool._protobuf.mempool.AcquireResponse\x12N\n\x03put\x12\".wool._protobuf.mempool.PutRequest\x1a#.wool._protobuf.mempool.PutResponse\x12Q\n\x04post\x12#.wool._protobuf.mempool.PostRequest\x1a$.wool._protobuf.mempool.PostResponse\x12N\n\x03get\x12\".wool._protobuf.mempool.GetRequest\x1a#.wool._protobuf.mempool.GetResponse\x12Z\n\x07release\x12&.wool._protobuf.mempool.ReleaseRequest\x1a\'.wool._protobuf.mempool.ReleaseResponseb\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
-_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mempool.mempool_pb2', _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mempool.service_pb2', _globals)
 if not _descriptor._USE_C_DESCRIPTORS:
   DESCRIPTOR._loaded_options = None
   _globals['_SESSIONREQUEST']._serialized_start=49
@@ -62,5 +62,5 @@ if not _descriptor._USE_C_DESCRIPTORS:
   _globals['_EVENT']._serialized_start=1007
   _globals['_EVENT']._serialized_end=1088
   _globals['_MEMORYPOOL']._serialized_start=1091
-  _globals['_MEMORYPOOL']._serialized_end=1624
+  _globals['_MEMORYPOOL']._serialized_end=1712
 # @@protoc_insertion_point(module_scope)

wool-0.1rc8/wool/_protobuf/mempool/mempool_pb2_grpc.py → wool-0.1rc9/wool/_protobuf/mempool/service_pb2_grpc.py

@@ -3,7 +3,7 @@
 import grpc
 import warnings
 
-from mempool import mempool_pb2 as mempool_dot_mempool__pb2
+from mempool import service_pb2 as mempool_dot_service__pb2
 
 GRPC_GENERATED_VERSION = '1.73.1'
 GRPC_VERSION = grpc.__version__
@@ -18,7 +18,7 @@ except ImportError:
 if _version_not_supported:
     raise RuntimeError(
         f'The grpc package installed is at version {GRPC_VERSION},'
-        + f' but the generated code in mempool/mempool_pb2_grpc.py depends on'
+        + f' but the generated code in mempool/service_pb2_grpc.py depends on'
         + f' grpcio>={GRPC_GENERATED_VERSION}.'
         + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
         + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
@@ -36,33 +36,38 @@ class MemoryPoolStub(object):
         """
         self.session = channel.unary_stream(
                 '/wool._protobuf.mempool.MemoryPool/session',
-                request_serializer=mempool_dot_mempool__pb2.SessionRequest.SerializeToString,
-                response_deserializer=mempool_dot_mempool__pb2.SessionResponse.FromString,
+                request_serializer=mempool_dot_service__pb2.SessionRequest.SerializeToString,
+                response_deserializer=mempool_dot_service__pb2.SessionResponse.FromString,
                 _registered_method=True)
         self.acquire = channel.unary_unary(
                 '/wool._protobuf.mempool.MemoryPool/acquire',
-                request_serializer=mempool_dot_mempool__pb2.AcquireRequest.SerializeToString,
-                response_deserializer=mempool_dot_mempool__pb2.AcquireResponse.FromString,
+                request_serializer=mempool_dot_service__pb2.AcquireRequest.SerializeToString,
+                response_deserializer=mempool_dot_service__pb2.AcquireResponse.FromString,
+                _registered_method=True)
+        self.map = channel.unary_unary(
+                '/wool._protobuf.mempool.MemoryPool/map',
+                request_serializer=mempool_dot_service__pb2.AcquireRequest.SerializeToString,
+                response_deserializer=mempool_dot_service__pb2.AcquireResponse.FromString,
                 _registered_method=True)
         self.put = channel.unary_unary(
                 '/wool._protobuf.mempool.MemoryPool/put',
-                request_serializer=mempool_dot_mempool__pb2.PutRequest.SerializeToString,
-                response_deserializer=mempool_dot_mempool__pb2.PutResponse.FromString,
+                request_serializer=mempool_dot_service__pb2.PutRequest.SerializeToString,
+                response_deserializer=mempool_dot_service__pb2.PutResponse.FromString,
                 _registered_method=True)
         self.post = channel.unary_unary(
                 '/wool._protobuf.mempool.MemoryPool/post',
-                request_serializer=mempool_dot_mempool__pb2.PostRequest.SerializeToString,
-                response_deserializer=mempool_dot_mempool__pb2.PostResponse.FromString,
+                request_serializer=mempool_dot_service__pb2.PostRequest.SerializeToString,
+                response_deserializer=mempool_dot_service__pb2.PostResponse.FromString,
                 _registered_method=True)
         self.get = channel.unary_unary(
                 '/wool._protobuf.mempool.MemoryPool/get',
-                request_serializer=mempool_dot_mempool__pb2.GetRequest.SerializeToString,
-                response_deserializer=mempool_dot_mempool__pb2.GetResponse.FromString,
+                request_serializer=mempool_dot_service__pb2.GetRequest.SerializeToString,
+                response_deserializer=mempool_dot_service__pb2.GetResponse.FromString,
                 _registered_method=True)
         self.release = channel.unary_unary(
                 '/wool._protobuf.mempool.MemoryPool/release',
-                request_serializer=mempool_dot_mempool__pb2.ReleaseRequest.SerializeToString,
-                response_deserializer=mempool_dot_mempool__pb2.ReleaseResponse.FromString,
+                request_serializer=mempool_dot_service__pb2.ReleaseRequest.SerializeToString,
+                response_deserializer=mempool_dot_service__pb2.ReleaseResponse.FromString,
                 _registered_method=True)
 
 
@@ -81,6 +86,12 @@ class MemoryPoolServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def map(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
     def put(self, request, context):
         """Missing associated documentation comment in .proto file."""
         context.set_code(grpc.StatusCode.UNIMPLEMENTED)
@@ -110,33 +121,38 @@ def add_MemoryPoolServicer_to_server(servicer, server):
     rpc_method_handlers = {
             'session': grpc.unary_stream_rpc_method_handler(
                     servicer.session,
-                    request_deserializer=mempool_dot_mempool__pb2.SessionRequest.FromString,
-                    response_serializer=mempool_dot_mempool__pb2.SessionResponse.SerializeToString,
+                    request_deserializer=mempool_dot_service__pb2.SessionRequest.FromString,
+                    response_serializer=mempool_dot_service__pb2.SessionResponse.SerializeToString,
             ),
             'acquire': grpc.unary_unary_rpc_method_handler(
                     servicer.acquire,
-                    request_deserializer=mempool_dot_mempool__pb2.AcquireRequest.FromString,
-                    response_serializer=mempool_dot_mempool__pb2.AcquireResponse.SerializeToString,
+                    request_deserializer=mempool_dot_service__pb2.AcquireRequest.FromString,
+                    response_serializer=mempool_dot_service__pb2.AcquireResponse.SerializeToString,
+            ),
+            'map': grpc.unary_unary_rpc_method_handler(
+                    servicer.map,
+                    request_deserializer=mempool_dot_service__pb2.AcquireRequest.FromString,
+                    response_serializer=mempool_dot_service__pb2.AcquireResponse.SerializeToString,
             ),
             'put': grpc.unary_unary_rpc_method_handler(
                     servicer.put,
-                    request_deserializer=mempool_dot_mempool__pb2.PutRequest.FromString,
-                    response_serializer=mempool_dot_mempool__pb2.PutResponse.SerializeToString,
+                    request_deserializer=mempool_dot_service__pb2.PutRequest.FromString,
+                    response_serializer=mempool_dot_service__pb2.PutResponse.SerializeToString,
             ),
             'post': grpc.unary_unary_rpc_method_handler(
                     servicer.post,
-                    request_deserializer=mempool_dot_mempool__pb2.PostRequest.FromString,
-                    response_serializer=mempool_dot_mempool__pb2.PostResponse.SerializeToString,
+                    request_deserializer=mempool_dot_service__pb2.PostRequest.FromString,
+                    response_serializer=mempool_dot_service__pb2.PostResponse.SerializeToString,
             ),
             'get': grpc.unary_unary_rpc_method_handler(
                     servicer.get,
-                    request_deserializer=mempool_dot_mempool__pb2.GetRequest.FromString,
-                    response_serializer=mempool_dot_mempool__pb2.GetResponse.SerializeToString,
+                    request_deserializer=mempool_dot_service__pb2.GetRequest.FromString,
+                    response_serializer=mempool_dot_service__pb2.GetResponse.SerializeToString,
             ),
             'release': grpc.unary_unary_rpc_method_handler(
                     servicer.release,
-                    request_deserializer=mempool_dot_mempool__pb2.ReleaseRequest.FromString,
-                    response_serializer=mempool_dot_mempool__pb2.ReleaseResponse.SerializeToString,
+                    request_deserializer=mempool_dot_service__pb2.ReleaseRequest.FromString,
+                    response_serializer=mempool_dot_service__pb2.ReleaseResponse.SerializeToString,
             ),
     }
     generic_handler = grpc.method_handlers_generic_handler(
@@ -164,8 +180,8 @@ class MemoryPool(object):
             request,
             target,
             '/wool._protobuf.mempool.MemoryPool/session',
-            mempool_dot_mempool__pb2.SessionRequest.SerializeToString,
-            mempool_dot_mempool__pb2.SessionResponse.FromString,
+            mempool_dot_service__pb2.SessionRequest.SerializeToString,
+            mempool_dot_service__pb2.SessionResponse.FromString,
             options,
             channel_credentials,
             insecure,
@@ -191,8 +207,35 @@ class MemoryPool(object):
             request,
             target,
             '/wool._protobuf.mempool.MemoryPool/acquire',
-            mempool_dot_mempool__pb2.AcquireRequest.SerializeToString,
-            mempool_dot_mempool__pb2.AcquireResponse.FromString,
+            mempool_dot_service__pb2.AcquireRequest.SerializeToString,
+            mempool_dot_service__pb2.AcquireResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def map(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/wool._protobuf.mempool.MemoryPool/map',
+            mempool_dot_service__pb2.AcquireRequest.SerializeToString,
+            mempool_dot_service__pb2.AcquireResponse.FromString,
             options,
             channel_credentials,
             insecure,
@@ -218,8 +261,8 @@ class MemoryPool(object):
             request,
             target,
             '/wool._protobuf.mempool.MemoryPool/put',
-            mempool_dot_mempool__pb2.PutRequest.SerializeToString,
-            mempool_dot_mempool__pb2.PutResponse.FromString,
+            mempool_dot_service__pb2.PutRequest.SerializeToString,
+            mempool_dot_service__pb2.PutResponse.FromString,
             options,
             channel_credentials,
             insecure,
@@ -245,8 +288,8 @@ class MemoryPool(object):
             request,
             target,
             '/wool._protobuf.mempool.MemoryPool/post',
-            mempool_dot_mempool__pb2.PostRequest.SerializeToString,
-            mempool_dot_mempool__pb2.PostResponse.FromString,
+            mempool_dot_service__pb2.PostRequest.SerializeToString,
+            mempool_dot_service__pb2.PostResponse.FromString,
             options,
             channel_credentials,
             insecure,
@@ -272,8 +315,8 @@ class MemoryPool(object):
             request,
             target,
             '/wool._protobuf.mempool.MemoryPool/get',
-            mempool_dot_mempool__pb2.GetRequest.SerializeToString,
-            mempool_dot_mempool__pb2.GetResponse.FromString,
+            mempool_dot_service__pb2.GetRequest.SerializeToString,
+            mempool_dot_service__pb2.GetResponse.FromString,
             options,
             channel_credentials,
             insecure,
@@ -299,8 +342,8 @@ class MemoryPool(object):
             request,
             target,
             '/wool._protobuf.mempool.MemoryPool/release',
-            mempool_dot_mempool__pb2.ReleaseRequest.SerializeToString,
-            mempool_dot_mempool__pb2.ReleaseResponse.FromString,
+            mempool_dot_service__pb2.ReleaseRequest.SerializeToString,
+            mempool_dot_service__pb2.ReleaseResponse.FromString,
             options,
             channel_credentials,
             insecure,

wool-0.1rc8/wool/_mempool/_service.py

@@ -1,225 +0,0 @@
-from __future__ import annotations
-
-import asyncio
-from typing import AsyncGenerator
-from typing import Final
-from weakref import WeakSet
-from weakref import WeakValueDictionary
-
-import shortuuid
-from grpc.aio import ServicerContext
-
-from wool._mempool import MemoryPool
-from wool._protobuf.mempool import mempool_pb2 as proto
-from wool._protobuf.mempool import mempool_pb2_grpc as rpc
-
-
-class Session:
-    """
-    A session represents a client connection to the memory pool service and
-    serves as the scope for any shared references acquired over its duration.
-    """
-
-    id: Final[str]
-    queue: Final[asyncio.Queue[proto.SessionResponse]]
-    references: Final[set[Reference]]
-    sessions: Final[WeakValueDictionary[str, Session]] = WeakValueDictionary()
-
-    @classmethod
-    def get(cls, id: str) -> Session | None:
-        return cls.sessions.get(id)
-
-    def __init__(self):
-        self.id = shortuuid.uuid()
-        self.queue = asyncio.Queue()
-        self.references = set()
-        self.sessions[self.id] = self
-
-    def __eq__(self, other) -> bool:
-        if isinstance(other, Session):
-            return self.id == other.id
-        return False
-
-    def __hash__(self) -> int:
-        return hash(self.id)
-
-
-class Reference:
-    id: Final[str]
-    mempool: Final[MemoryPool]
-    sessions: Final[WeakSet[Session]]
-
-    _references: Final[WeakValueDictionary[str, Reference]] = (
-        WeakValueDictionary()
-    )
-    _to_delete: Final[set[str]] = set()
-    _initialized: bool = False
-
-    @classmethod
-    def get(cls, id: str) -> Reference | None:
-        return cls._references.get(id)
-
-    @classmethod
-    def new(cls, id: str, *, mempool: MemoryPool) -> Reference:
-        if id in cls._references:
-            raise ValueError(f"Reference {id} already exists")
-        return cls(id, mempool=mempool)
-
-    def __new__(cls, id: str, *, mempool: MemoryPool):
-        if id in cls._references:
-            if id in cls._to_delete:
-                cls._to_delete.remove(id)
-            return cls._references[id]
-        return super().__new__(cls)
-
-    def __init__(self, id: str, *, mempool: MemoryPool):
-        if not self._initialized:
-            self.id = id
-            self.mempool = mempool
-            self.sessions = WeakSet()
-            self._references[id] = self
-            self._initialized = True
-
-    def __eq__(self, other) -> bool:
-        if isinstance(other, Reference):
-            return self.id == other.id
-        return False
-
-    def __hash__(self) -> int:
-        return hash(self.id)
-
-    def __del__(self):
-        self._to_delete.add(self.id)
-
-        id = self.id
-        to_delete = self._to_delete
-        mempool = self.mempool
-
-        async def _delete():
-            if id in to_delete:
-                try:
-                    to_delete.remove(id)
-                    await mempool.delete(id)
-                except FileNotFoundError:
-                    pass
-
-        try:
-            asyncio.get_running_loop().create_task(_delete())
-        except RuntimeError:
-            asyncio.new_event_loop().run_until_complete(_delete())
-
-
-class MemoryPoolService(rpc.MemoryPoolServicer):
-    def __init__(self, mempool: MemoryPool | None = None):
-        self._mempool = mempool or MemoryPool()
-        self._shutdown = asyncio.Event()
-
-    async def session(
-        self, request: proto.SessionRequest, context: ServicerContext
-    ) -> AsyncGenerator[proto.SessionResponse]:
-        session = Session()
-        yield proto.SessionResponse(session=proto.Session(id=session.id))
-        while True:
-            yield await session.queue.get()
-
-    async def acquire(
-        self, request: proto.AcquireRequest, context: ServicerContext
-    ) -> proto.AcquireResponse:
-        if not (session := Session.get(request.session.id)):
-            raise ValueError(f"Session {request.session.id} not found")
-        if not (reference := Reference.get(request.reference.id)):
-            raise ValueError(f"Reference {request.reference.id} not found")
-        session.references.add(reference)
-        reference.sessions.add(session)
-        return proto.AcquireResponse()
-
-    async def map(
-        self, request: proto.AcquireRequest, context: ServicerContext
-    ) -> proto.AcquireResponse:
-        if not (session := Session.get(request.session.id)):
-            raise ValueError(f"Session {request.session.id} not found")
-        await self._mempool.map(request.reference.id)
-        reference = Reference(request.reference.id, mempool=self._mempool)
-        await self.acquire(
-            proto.AcquireRequest(
-                session=proto.Session(id=session.id),
-                reference=proto.Reference(id=reference.id),
-            ),
-            context,
-        )
-        return proto.AcquireResponse()
-
-    async def put(
-        self, request: proto.PutRequest, context: ServicerContext
-    ) -> proto.PutResponse:
-        if not (session := Session.get(request.session.id)):
-            raise ValueError(f"Session {request.session.id} not found")
-        reference = Reference(
-            id=await self._mempool.put(request.dump, mutable=request.mutable),
-            mempool=self._mempool,
-        )
-        await self.acquire(
-            proto.AcquireRequest(
-                session=proto.Session(id=session.id),
-                reference=proto.Reference(id=reference.id),
-            ),
-            context,
-        )
-        return proto.PutResponse(reference=proto.Reference(id=reference.id))
-
-    async def get(
-        self, request: proto.GetRequest, context: ServicerContext
-    ) -> proto.GetResponse:
-        if not (session := Session.get(request.session.id)):
-            raise ValueError(f"Session {request.session.id} not found")
-        if not (reference := Reference.get(request.reference.id)):
-            raise ValueError(f"Reference {request.reference.id} not found")
-        if reference not in session.references:
-            await self.acquire(
-                proto.AcquireRequest(
-                    session=proto.Session(id=session.id),
-                    reference=proto.Reference(id=reference.id),
-                ),
-                context,
-            )
-        dump = await self._mempool.get(reference.id)
-        return proto.GetResponse(dump=dump)
-
-    async def post(
-        self, request: proto.PostRequest, context: ServicerContext
-    ) -> proto.PostResponse:
-        if not (session := Session.get(request.session.id)):
-            raise ValueError(f"Session {request.session.id} not found")
-        if not (reference := Reference.get(request.reference.id)):
-            raise ValueError(f"Reference {request.reference.id} not found")
-        if reference not in session.references:
-            await self.acquire(
-                proto.AcquireRequest(
-                    session=proto.Session(id=session.id),
-                    reference=proto.Reference(id=reference.id),
-                ),
-                context,
-            )
-        updated = await self._mempool.post(request.reference.id, request.dump)
-        if updated:
-            for session in Reference(
-                id=request.reference.id, mempool=self._mempool
-            ).sessions:
-                if session.id is not request.session.id:
-                    event = proto.Event(
-                        reference=request.reference,
-                        event_type="post",
-                    )
-                    await session.queue.put(proto.SessionResponse(event=event))
-        return proto.PostResponse(updated=updated)
-
-    async def release(
-        self, request: proto.ReleaseRequest, context: ServicerContext
-    ) -> proto.ReleaseResponse:
-        if not (session := Session.get(request.session.id)):
-            raise ValueError(f"Session {request.session.id} not found")
-        if not (reference := Reference.get(request.reference.id)):
-            raise ValueError(f"Reference {request.reference.id} not found")
-        session.references.remove(reference)
-        reference.sessions.remove(session)
-        return proto.ReleaseResponse()

wool-0.1rc8/wool/_protobuf/__init__.py

@@ -1,4 +0,0 @@
-import os
-import sys
-
-sys.path.insert(0, os.path.abspath(os.path.dirname(__file__)))