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

made the master's participant count an optional parameter

Browse Source
This commit is contained in:
Stephen McQuay 2011-04-02 01:32:23 -06:00
parent 0f5356867d
commit d61a9dadba
1 changed files with 20 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
bin/master.py
View File

@ -26,6 +26,10 @@ if __name__ == '__main__':
action="store_true", dest="last", default=False,
help="when finished, send shutdown signal to connected nodes (default: %default)")
parser.add_option('-n', '--node-count',
type="int", dest="participants", default=None,
help="specify how many participants we should wait for (default: %default)")
parser.add_option('-p', '--port',
type="int", dest="port", default=6666,
help="specify the port to use on the server (default: %default)")
@ -55,15 +59,22 @@ if __name__ == '__main__':
tasksq, resultsq, masterq, slavesq = get_qs(m)
print "wait on all participants"
participants = 0
while not masterq.empty():
participants += 1
worker = masterq.get()
print "%d: %s is ready" % (participants, worker)
if participants == 0:
print "nobody found"
sys.exit(1)
if not options.participants:
print "wait on all announced participants"
participants = 0
while not masterq.empty():
participants += 1
worker = masterq.get()
print "%d: %s is ready" % (participants, worker)
if participants == 0:
print "nobody found"
sys.exit(1)
else:
participants = options.participants
print "wait on %d participants" % participants
for i in xrange(participants):
worker = masterq.get()
print "%d of %d: %s is ready" % (i+1, participants, worker)
results = []