futureperfect · March 16, 2025 20:43 · Mar 16, 2025
diff --git a/async_app.py b/async_app.py
@@ -0,0 +1,41 @@
+import asyncio
+import bisect
+import logging
+
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+logger = logging.getLogger(__name__)
+
+async def sleep_and_sort(n, lock, result_list):
+    try:
+        logger.info(f"Sleeping for {n} seconds before doing work")
+        await asyncio.sleep(n)
+        logger.debug(f"Worker {n}: awake after {n} seconds")
+        async with lock:
+            logger.debug(f"Lock acquired, executing critical section")
+            bisect.insort(result_list, n)
+            logger.debug(f"Critical section finished, releasing lock")
+    except asyncio.CancelledError:
+        logger.error(f"Task {n}: cancelled")
+
+
+
+async def main() -> None:
+    input_nums = [3, 1, 5, 4, 2, 7, 12, 2, 7, 8]
+    lock = asyncio.Lock()
+    result: list[int] = []
+    try:
+        _tasks = []
+        async with asyncio.TaskGroup() as tg:
+            _tasks + [tg.create_task(sleep_and_sort(n, lock, result)) for n in input_nums]
+
+        logger.info(f"Result: {result}")
+    except asyncio.CancelledError as e:
+        logger.error(f"Task cancelled and captured outside TG: {e}")
+    finally:
+        # do final resource cleanup here
+        logger.info("main: exiting")
+
+
+if __name__ == "__main__":
+    with asyncio.Runner() as runner:
+        runner.run(main())