mirror of
https://github.com/ceph/ceph
synced 2024-12-24 20:33:27 +00:00
64 lines
1.9 KiB
Python
64 lines
1.9 KiB
Python
|
import logging
|
||
|
|
import gevent
|
||
|
|
import random
|
||
|
|
import time
|
||
|
|
|
||
|
|
from ..orchestra import run
|
||
|
|
|
||
|
|
log = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
class ProcThrasher:
|
||
|
|
""" Kills and restarts some number of the specified process on the specified
|
||
|
|
remote
|
||
|
|
"""
|
||
|
|
def __init__(self, config, remote, *proc_args, **proc_kwargs):
|
||
|
|
self.proc_kwargs = proc_kwargs
|
||
|
|
self.proc_args = proc_args
|
||
|
|
self.config = config
|
||
|
|
self.greenlet = None
|
||
|
|
self.logger = proc_kwargs.get("logger", log.getChild('proc_thrasher'))
|
||
|
|
self.remote = remote
|
||
|
|
|
||
|
|
# config:
|
||
|
|
self.num_procs = self.config.get("num_procs", 5)
|
||
|
|
self.rest_period = self.config.get("rest_period", 100) # seconds
|
||
|
|
self.run_time = self.config.get("run_time", 1000) # seconds
|
||
|
|
|
||
|
|
def log(self, msg):
|
||
|
|
self.logger.info(msg)
|
||
|
|
|
||
|
|
def start(self):
|
||
|
|
if self.greenlet is not None:
|
||
|
|
return
|
||
|
|
self.greenlet = gevent.Greenlet(self.loop)
|
||
|
|
self.greenlet.start()
|
||
|
|
|
||
|
|
def join(self):
|
||
|
|
self.greenlet.join()
|
||
|
|
|
||
|
|
def loop(self):
|
||
|
|
time_started = time.time()
|
||
|
|
procs = []
|
||
|
|
self.log("Starting")
|
||
|
|
while time_started + self.run_time > time.time():
|
||
|
|
if len(procs) > 0:
|
||
|
|
self.log("Killing proc")
|
||
|
|
proc = random.choice(procs)
|
||
|
|
procs.remove(proc)
|
||
|
|
proc.stdin.close()
|
||
|
|
self.log("About to wait")
|
||
|
|
run.wait([proc])
|
||
|
|
self.log("Killed proc")
|
||
|
|
|
||
|
|
while len(procs) < self.num_procs:
|
||
|
|
self.log("Creating proc " + str(len(procs) + 1))
|
||
|
|
self.log("args are " + str(self.proc_args) + " kwargs: " + str(self.proc_kwargs))
|
||
|
|
procs.append(self.remote.run(
|
||
|
|
*self.proc_args,
|
||
|
|
** self.proc_kwargs))
|
||
|
|
self.log("About to sleep")
|
||
|
|
time.sleep(self.rest_period)
|
||
|
|
self.log("Just woke")
|
||
|
|
|
||
|
|
run.wait(procs)
|