ԋgjddlmZddlZddlZddlZddlZddlmZddlm Z ddl m Z Gdde Z y)) annotationsN)Literal)parse_timedelta) WorkerPluginc<eZdZdZ d d dZdZdZdZdZy) KillWorkeraKill Workers Randomly This kills workers in a cluster randomly. It is intended to be used in stress testing. Parameters ---------- delay: str The expected amount of time for a worker to live. The actual time will vary, treating worker death as a poisson process. mode: str or "graceful" which calls worker.close(...) Either "sys.exit" which calls sys.exit(0) or "segfault" which triggers a segfault cVt||_|dvrtd|||_y)N)sys.exitgracefulsegfaultzCThree modes supported, 'sys.exit', 'graceful', and 'segfault'. got )rdelay ValueErrormode)selfr rs 1lib/python3.12/site-packages/distributed/chaos.py__init__zKillWorker.__init__s> %U+ ; ;h   cbK||_|jdk(r |j}n7|jdk(r |j}n|jdk(r |j}|jj j jtjd|jz yw)Nr r r )r callback) workerrr sys_exitr loop asyncio_loop call_laterrandom expovariater )rrfs rsetupzKillWorker.setup,s 99 " A YY* $ A YY* $ A %%00$$Q^4 1 sB-B/cdtj|jjddy)NF)nanny executor_wait)asyncio create_taskrclosers rr zKillWorker.graceful:s$DKK--E-OPrc.tjdy)Nr)sysexitr&s rrzKillWorker.sys_exit=s   rc.tjdy)z Magic, from https://gist.github.com/coolreader18/6dbe0be2ae2192e90e1a809f1624c694?permalink_comment_id=3874116#gistcomment-3874116 rN)ctypes string_atr&s rr zKillWorker.segfault@s rN)z100 sr )r zstr | int | floatrz+Literal['sys.exit', 'graceful', 'segfault']) __name__ __module__ __qualname____doc__rrr rr rrrrs;$$+r6s)" &766r