Method read

trinity/buffer/reader/sql_reader.py:38–51

(self, batch_size: Optional[int] = None, **kwargs)

        return self._async_storage

    async def read(self, batch_size: Optional[int] = None, **kwargs) -> List:
        batch_size = self.read_batch_size if batch_size is None else batch_size
        if self.wrap_in_ray:
            try:
                return await self.storage.read.remote(batch_size, **kwargs)
            except (StopIteration, StopAsyncIteration):
                raise StopAsyncIteration()
            except Exception as e:
                if "StopAsyncIteration" in traceback.format_exc():
                    raise StopAsyncIteration() from e
                raise
        else:
            storage = await self._get_async_storage()
            return await storage.read(batch_size, **kwargs)

    def state_dict(self) -> Dict:
        return {"current_index": 0}