sm/smbinterp
1
0
Fork 0
Code Issues Pull Requests Activity

made iqmanager more complicated and better

Browse Source
This commit is contained in:
Stephen Mardson McQuay 2011-05-26 10:12:01 -06:00
parent c08504a089
commit a1527b25c8
1 changed files with 70 additions and 44 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

60
bin/iqmgr.py
View File

@ -4,26 +4,41 @@ import sys
import time
import interp.bootstrap
from interp.cluster import QueueManager, get_qs
from optparse import OptionParser
m = QueueManager(address=('localhost', 6666), authkey='asdf')
if __name__ == '__main__':
parser = OptionParser(usage = "usage: %s [options] <status|watch|add|fresult #|slay|clear|clearall|clearresults>")
parser.add_option('-p', '--port',
type="int", dest="port", default=6666,
help="specify the port to use on the server (default: %default)")
parser.add_option('-a', '--auth-key',
type="str", dest="authkey", default='asdf',
help="authkey (default: %default)")
(options, args) = parser.parse_args()
if len(args) == 0:
cmd = 'status'
else:
cmd = args[0]
m = QueueManager(address=('', options.port), authkey=options.authkey)
m.connect()
tq,rq,mq,sq = get_qs(m)
if len(sys.argv) == 1:
sys.argv.append("st")
if sys.argv[1].startswith("st"):
if cmd.startswith("st"):
print "interp queue status:"
print " tasksq : %d" % tq.qsize()
print " resultsq : %d" % rq.qsize()
print " masterq : %d" % mq.qsize()
print " minionsq : %d" % sq.qsize()
if sys.argv[1].startswith("wa"):
if len(sys.argv) == 3:
sleeptime = float(sys.argv[2])
if cmd.startswith("wa"):
if len(args) == 2:
sleeptime = float(args[1])
else:
sleeptime = 1
@ -39,23 +54,34 @@ if sys.argv[1].startswith("wa"):
sq.qsize(),)
i += 1
if sys.argv[1] == 'add':
for i in xrange(int(sys.argv[2])):
mq.put('jane')
if cmd == 'add':
for i in xrange(int(args[1])):
mq.put('jane%d' % i)
if sys.argv[1] == 'slay':
if len(sys.argv) == 2:
if cmd == 'fresult':
for i in xrange(int(args[1])):
rq.put('fake.%d' % i)
if cmd == 'slay':
if len(args) == 1:
for i in xrange(mq.qsize()):
print i, "killing", mq.get()
sq.put("slay")
elif len(sys.argv) == 3:
for i in xrange(int(sys.argv[2])):
elif len(args) == 2:
for i in xrange(int(args[1])):
print i, "killing", mq.get()
sq.put("slay")
if sys.argv[1] == 'clear':
if cmd == 'clear':
for i in xrange(tq.qsize()): print tq.get()
for i in xrange(rq.qsize()): print rq.get()
if cmd == 'clearall':
for i in xrange(tq.qsize()): print tq.get()
for i in xrange(rq.qsize()): print rq.get()
for i in xrange(mq.qsize()): print mq.get()
for i in xrange(sq.qsize()): print sq.get()
if cmd == 'clearresults':
for i in xrange(rq.qsize()): print rq.get()