Lenaisten/advent22
1
0
Fork 0
mirror of https://code.lenaisten.de/Lenaisten/advent22.git synced 2024-11-23 08:13:01 +00:00
Code Issues Projects Releases Packages Wiki Activity

experimental success

Browse source
This commit is contained in:
Jörn-Michael Miehe 2023-09-08 16:08:10 +00:00
parent 5223efddb4
commit e894aad746
1 changed files with 23 additions and 36 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

59
api/asynccontextmanager.py
View file

@ -1,38 +1,32 @@
import asyncio import asyncio
import functools import functools
from contextlib import asynccontextmanager, contextmanager
from io import BytesIO from io import BytesIO
from typing import AsyncContextManager, AsyncIterator, ContextManager, Iterator
from cache import AsyncLRU from cache import AsyncLRU
FILES = __file__, "pyproject.toml"
REPS = 3
# #
# sync impl # sync impl
# #
def get_buf_sync() -> ContextManager[BytesIO]: @functools.lru_cache
if getattr(get_buf_sync, "inner", None) is None: def get_bytes_sync(fn) -> bytes:
print("sync open")
with open(fn, "rb") as file:
return file.readline()
@functools.lru_cache
def inner() -> bytes:
print("sync open")
with open(__file__, "rb") as file:
return file.readline()
setattr(get_buf_sync, "inner", inner) def get_buf_sync(fn) -> BytesIO:
return BytesIO(get_bytes_sync(fn))
@contextmanager
def ctx() -> Iterator[BytesIO]:
yield BytesIO(get_buf_sync.inner())
return ctx()
def main_sync() -> None: def main_sync() -> None:
for _ in range(2): for fn in FILES:
with get_buf_sync() as buffer: for _ in range(REPS):
print(buffer.read()) print(get_buf_sync(fn).read())
# #
@ -40,28 +34,21 @@ def main_sync() -> None:
# #
async def get_buf_async() -> AsyncContextManager[BytesIO]: @AsyncLRU()
if getattr(get_buf_async, "inner", None) is None: async def get_bytes_async(fn) -> bytes:
print("async open")
with open(fn, "rb") as file:
return file.readline()
@AsyncLRU()
async def inner() -> bytes:
print("async open")
with open(__file__, "rb") as file:
return file.readline()
setattr(get_buf_async, "inner", inner) async def get_buf_async(fn) -> BytesIO:
return BytesIO(await get_bytes_async(fn))
@asynccontextmanager
async def ctx() -> AsyncIterator[BytesIO]:
yield BytesIO(await get_buf_async.inner())
return ctx()
async def main_async() -> None: async def main_async() -> None:
for _ in range(2): for fn in FILES:
async with await get_buf_async() as buffer: for _ in range(REPS):
print(buffer.read()) print((await get_buf_async(fn)).read())
if __name__ == "__main__": if __name__ == "__main__":