o ²µjć@sdZddlZGdd„dƒZdS)zt Async version of gunicorn/http/unreader.py for ASGI workers. Provides async reading with pushback buffer support. éNc@sTeZdZdZddd„Zdd„Zdd„Zd d „Zdd d „Zdd„Z dd„Z dd„Z d S)Ś AsyncUnreadera8Async 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. Performance optimization: Reuses BytesIO buffer with truncate/seek instead of creating new objects to reduce GC pressure. é cCs ||_t ”|_||_d|_dS)zInitialize the async unreader. Args: reader: asyncio.StreamReader instance max_chunk: Maximum bytes to read at once rN)ŚreaderŚioŚBytesIOŚbufŚ max_chunkŚ _buf_start)Śselfrr©r śK/var/www/ai-app/venv/lib/python3.10/site-packages/gunicorn/asgi/unreader.pyŚ__init__s  zAsyncUnreader.__init__cCs"|j d”|j d”d|_dS)z7Reset buffer for reuse instead of creating new BytesIO.rN)rŚseekŚtruncater ©r r r r Ś _reset_buffer$s   zAsyncUnreader._reset_buffercCs$|j |j”|j ”}| ”|S)z'Get all buffered data and reset buffer.)rrr Śreadr)r Śdatar r r Ś_get_buffered_data*s z AsyncUnreader._get_buffered_datacCs|j dtj”}||jS)zGet size of buffered data.r)rrrŚSEEK_ENDr )r Śendr r r Ś _buffer_size1s zAsyncUnreader._buffer_sizeNcĆs |durt|tƒstdƒ‚|dur|dkrdS|dkrd}| ”}|dur.|dkr.| ”S|dur;| ”IdH}|S||krd| ”IdH}|sL| ”S|j dtj ”|j  |”|t |ƒ7}||ks?|j |j ”|j  |”}|j |7_ |j dkrŽ|j  ”}| ”|rŽ|j  |”|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ór)Ś isinstanceŚintŚ TypeErrorrrŚ _read_chunkrrrrŚwriteŚlenr rr)r ŚsizeŚbuf_sizeŚchunkrŚ remainingr r r r6s>€   ś    zAsyncUnreader.readcĆs.z |j |j”IdHWStyYdSw)z0Read a chunk of data from the underlying stream.Nr)rrrŚ Exceptionrr r r rks € ’zAsyncUnreader._read_chunkcCsL|r"|j |j”|j ”}| ”|j |”|r$|j |”dSdSdS)zµPush data back into the buffer for re-reading. Args: data: bytes to push back Note: This prepends data to the buffer so it will be read first. N)rrr rrr)r rŚexistingr r r Śunreadrs  ÷zAsyncUnreader.unreadcCs | ”dkS)z-Check if there's data in the pushback buffer.r)rrr r r Śhas_buffered_data…s zAsyncUnreader.has_buffered_data)r)N) Ś__name__Ś __module__Ś __qualname__Ś__doc__r rrrrrr%r&r r r r rs   5 r)r*rrr r r r Śs