| 12345678910111213141516171819202122232425262728293031 |
- # (c) Nelen & Schuurmans
- import asyncio
- from dramatiq.middleware import SkipMessage
- from .async_actor import async_actor
- __all__ = ["sleep_task"]
- @async_actor(
- retry_when=lambda x, y: isinstance(y, KeyError),
- max_retries=1,
- )
- async def sleep_task(seconds: int, return_value=None, event="success"):
- event = event.lower()
- if event == "success":
- await asyncio.sleep(int(seconds))
- elif event == "crash":
- import ctypes
- ctypes.string_at(0) # segfault
- elif event == "skip":
- raise SkipMessage("skipping")
- elif event == "retry":
- raise KeyError("will-retry")
- else:
- raise ValueError(f"Unknown event '{event}'")
- return return_value
|
