Class QueueWriter

trinity/buffer/writer/queue_writer.py:11–25

Writer of the Queue buffer.

class QueueWriter(BufferWriter):
    """Writer of the Queue buffer."""

    def __init__(self, config: StorageConfig):
        assert config.storage_type == StorageType.QUEUE.value
        self.queue = QueueStorage.get_wrapper(config)

    async def write(self, data: List[Experience]) -> None:
        return await self.queue.put_batch.remote(Experience.serialize_many(data))

    async def acquire(self) -> int:
        return await self.queue.acquire.remote()

    async def release(self) -> int:
        return await self.queue.release.remote()