mirror of
https://github.com/ceph/ceph
synced 2024-12-28 14:34:13 +00:00
62 lines
2.1 KiB
Python
62 lines
2.1 KiB
Python
|
import beanstalkc
|
||
|
|
import yaml
|
||
|
|
import logging
|
||
|
|
|
||
|
|
from .config import config
|
||
|
|
|
||
|
|
log = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
def beanstalk_connect(machine_type):
|
||
|
|
qhost = config.queue_host
|
||
|
|
qport = config.queue_port
|
||
|
|
if qhost is None or qport is None:
|
||
|
|
raise RuntimeError(
|
||
|
|
'Beanstalk queue information not found in {conf_path}'.format(
|
||
|
|
conf_path=config.teuthology_yaml))
|
||
|
|
log.info("Checking Beanstalk Queue...")
|
||
|
|
beanstalk = beanstalkc.Connection(host=qhost, port=qport)
|
||
|
|
beanstalk.watch(machine_type)
|
||
|
|
beanstalk.ignore('default')
|
||
|
|
return beanstalk
|
||
|
|
|
||
|
|
def walk_jobs(beanstalk, machine_type, delete=False):
|
||
|
|
curjobs = beanstalk.stats_tube(machine_type)['current-jobs-ready']
|
||
|
|
if curjobs != 0:
|
||
|
|
x=1
|
||
|
|
while x != curjobs:
|
||
|
|
x += 1
|
||
|
|
job = beanstalk.reserve(timeout=20)
|
||
|
|
if job.body is not None:
|
||
|
|
job_config = yaml.safe_load(job.body)
|
||
|
|
job_name=job_config['name']
|
||
|
|
job_id = job.stats()['id']
|
||
|
|
job_description = job_config['description']
|
||
|
|
if delete:
|
||
|
|
if delete in job_name:
|
||
|
|
print 'Deleted Job: {job_id} from queue. Name: {job_name}'.format(
|
||
|
|
job_id = job_id,
|
||
|
|
job_name = job_name,
|
||
|
|
)
|
||
|
|
job.delete()
|
||
|
|
else:
|
||
|
|
print "Searching queue... Checked " + str(x) + "/" + str(curjobs)," Jobs\r",
|
||
|
|
else:
|
||
|
|
print 'Job: {x}/{curjobs} ID: {job_id} Name: {job_name}'.format(
|
||
|
|
x=x,
|
||
|
|
curjobs=curjobs,
|
||
|
|
job_id=job_id,
|
||
|
|
job_name=job_name,
|
||
|
|
)
|
||
|
|
print '\t {desc}'.format(desc=job_description)
|
||
|
|
|
||
|
|
print "\nDone"
|
||
|
|
else:
|
||
|
|
log.info('No jobs in Beanstalk Queue')
|
||
|
|
|
||
|
|
def main(args):
|
||
|
|
machine_type = args['--machine_type']
|
||
|
|
delete = args['--delete']
|
||
|
|
beanstalk = beanstalk_connect(machine_type)
|
||
|
|
walk_jobs(beanstalk, machine_type, delete)
|
||
|
|
beanstalk.close()
|