o Á¿iA ã@sdZddlZGdd„dƒZdS)zt Async version of gunicorn/http/unreader.py for ASGI workers. Provides async reading with pushback buffer support. éNc@s<eZdZdZddd„Zddd„Zdd „Zd d „Zd d „ZdS)Ú AsyncUnreaderzµAsync socket reader with pushback buffer support. This class wraps an asyncio StreamReader and provides the ability to "unread" data back into a buffer for re-parsing. é cCs||_t ¡|_||_dS)zInitialize the async unreader. Args: reader: asyncio.StreamReader instance max_chunk: Maximum bytes to read at once N)ÚreaderÚioÚBytesIOÚbufÚ max_chunk)Úselfrr©r úJ/home/ubuntu/.local/lib/python3.10/site-packages/gunicorn/asgi/unreader.pyÚ__init__s  zAsyncUnreader.__init__NcÃs|durt|tƒstdƒ‚|dur|dkrdS|dkrd}|j dtj¡|dur;|j ¡r;|j ¡}t  ¡|_|S|durH|  ¡IdH}|S|j ¡|krq|  ¡IdH}|sd|j ¡}t  ¡|_|S|j  |¡|j ¡|ksO|j ¡}t  ¡|_|j  ||d…¡|d|…S)a Read data from the stream, using buffered data first. Args: size: Number of bytes to read. If None, returns all buffered data or reads a single chunk. Returns: bytes: Data read from buffer or stream Nz&size parameter must be an int or long.ró) Ú isinstanceÚintÚ TypeErrorrÚseekrÚSEEK_ENDÚtellÚgetvaluerÚ _read_chunkÚwrite)r ÚsizeÚretÚchunkÚdatar r r Úread s8€      ú   zAsyncUnreader.readcÃs.z |j |j¡IdHWStyYdSw)z0Read a chunk of data from the underlying stream.Nr )rrrÚ Exception)r r r r rMs € ÿzAsyncUnreader._read_chunkcCs(|r|j dtj¡|j |¡dSdS)zkPush data back into the buffer for re-reading. Args: data: bytes to push back rN)rrrrr)r rr r r ÚunreadTsþzAsyncUnreader.unreadcCs8|j ¡}|j dtj¡|j ¡dk}|j |¡|S)z-Check if there's data in the pushback buffer.r)rrrrr)r ÚposÚhas_datar r r Úhas_buffered_data^s  zAsyncUnreader.has_buffered_data)r)N) Ú__name__Ú __module__Ú __qualname__Ú__doc__r rrrr r r r r rs  - r)r$rrr r r r Ús