giambio/tests/timeout3.py

37 lines
1.0 KiB
Python

import giambio
from debugger import Debugger
async def child(name: int):
print(f"[child {name}] Child spawned!! Sleeping for {name} seconds")
await giambio.sleep(name)
print(f"[child {name}] Had a nice nap!")
return name
async def worker(coro, *args):
try:
async with giambio.with_timeout(2):
await coro(*args)
except giambio.exceptions.TooSlowError:
return True
return False
async def main():
start = giambio.clock()
try:
async with giambio.with_timeout(5) as pool:
task = await pool.spawn(worker, child, 5)
print(f"[main] Child has returned: {await task.join()}")
print(f"[main] Sleeping")
await giambio.sleep(500) # This will trigger the timeout
except giambio.exceptions.TooSlowError:
print("[main] One or more children have timed out!")
print(f"[main] Children execution complete in {giambio.clock() - start:.2f} seconds")
return 12
if __name__ == "__main__":
giambio.run(main, debugger=Debugger())