Various additions to make sure timed out and failed hosts are taken out of the playbook even if failed

during setup, async tasks, or poll operations.  This introduced some redundancy so I've made some
notes in the code where things need to be refactored to remove them.
pull/70/head
Michael DeHaan 12 years ago
parent 4ae98ed92d
commit e3b4981feb

@ -60,6 +60,9 @@ class PlaybookCallbacks(object):
def on_async_poll(self, jid, host, clock, host_result): def on_async_poll(self, jid, host, clock, host_result):
print async_poll_status(jid, host, clock, host_result) print async_poll_status(jid, host, clock, host_result)
def on_dark_host(self, host, msg):
print "exception: [%s] => %s" % (host, msg)
def main(args): def main(args):
''' run ansible-playbook operations ''' ''' run ansible-playbook operations '''

@ -202,6 +202,17 @@ class PlayBook(object):
runner.background = async_seconds runner.background = async_seconds
results = runner.run() results = runner.run()
# TODO -- this block is repeated in lots of places, refactor
dark_hosts = results.get('dark',{})
contacted_hosts = results.get('contacted',{})
for (host, error) in dark_hosts.iteritems():
self.callbacks.on_dark_host(host, error)
self.dark[host] = 1
for (host, host_result) in contacted_hosts.iteritems():
if 'failed' in host_result:
self.callbacks.on_failed(host, host_result)
self.failures[host] = 1
if async_poll_interval <= 0: if async_poll_interval <= 0:
# if not polling, playbook requested fire and forget # if not polling, playbook requested fire and forget
# trust the user wanted that and return immediately # trust the user wanted that and return immediately
@ -220,7 +231,9 @@ class PlayBook(object):
return results return results
clock = async_seconds clock = async_seconds
runner.hosts = self.hosts_to_poll(results) runner.hosts = self.hosts_to_poll(results)
runner.hosts = self._prune_failed_hosts(runner.hosts)
poll_results = results poll_results = results
while (clock >= 0): while (clock >= 0):
runner.hosts = poll_hosts runner.hosts = poll_hosts
@ -234,20 +247,33 @@ class PlayBook(object):
runner.pattern = '*' runner.pattern = '*'
runner.hosts = self.hosts_to_poll(poll_results) runner.hosts = self.hosts_to_poll(poll_results)
poll_results = runner.run() poll_results = runner.run()
if len(runner.hosts) == 0: if len(runner.hosts) == 0:
break break
if poll_results is None: if poll_results is None:
break break
# TODO -- this block is repeated in lots of places, refactor
dark_hosts = poll_results.get('dark',{})
contacted_hosts = poll_results.get('contacted',{})
for (host, error) in dark_hosts.iteritems():
self.callbacks.on_dark_host(host, error)
self.dark[host] = 1
for (host, host_result) in contacted_hosts.iteritems():
if 'failed' in host_result:
self.callbacks.on_failed(host, host_result)
self.failures[host] = 1
for (host, host_result) in poll_results['contacted'].iteritems(): for (host, host_result) in poll_results['contacted'].iteritems():
# override last result with current status result for report
results['contacted'][host] = host_result results['contacted'][host] = host_result
# output if requested if not host in self.dark and not host in self.failures:
self.callbacks.on_async_poll(jid, host, clock, host_result) # override last result with current status result for report
# run down the clock # output if requested
clock = clock - async_poll_interval self.callbacks.on_async_poll(jid, host, clock, host_result)
time.sleep(async_poll_interval) # run down the clock
# do not have to poll the completed hosts, smaller list clock = clock - async_poll_interval
runner.hosts = self.hosts_to_poll(poll_results) time.sleep(async_poll_interval)
# do not have to poll the completed hosts, smaller list
# mark any hosts that are still listed as started as failed # mark any hosts that are still listed as started as failed
# since these likely got killed by async_wrapper # since these likely got killed by async_wrapper
for (host, host_result) in results['contacted'].iteritems(): for (host, host_result) in results['contacted'].iteritems():
@ -274,9 +300,16 @@ class PlayBook(object):
) )
if async_seconds == 0: if async_seconds == 0:
return runner.run() rc = runner.run()
else: else:
return self._async_poll(runner, async_seconds, async_poll_interval) rc = self._async_poll(runner, async_seconds, async_poll_interval)
dark_hosts = rc.get('dark',{})
for (host, error) in dark_hosts.iteritems():
self.callbacks.on_dark_host(host, error)
return rc
def _run_task(self, pattern=None, task=None, host_list=None, def _run_task(self, pattern=None, task=None, host_list=None,
remote_user=None, handlers=None, conditional=False): remote_user=None, handlers=None, conditional=False):
@ -429,6 +462,17 @@ class PlayBook(object):
setup_cache=SETUP_CACHE setup_cache=SETUP_CACHE
).run() ).run()
# FIXME: similar logic up in run_task, refactor
dark_hosts = setup_results.get('dark',{})
contacted_hosts = setup_results.get('contacted',{})
for (host, error) in dark_hosts.iteritems():
self.callbacks.on_dark_host(host, error)
self.dark[host] = 1
for (host, host_result) in contacted_hosts.iteritems():
if 'failed' in host_result:
self.callbacks.on_failed(host, host_result)
self.failed[hosts] = 1
# now for each result, load into the setup cache so we can # now for each result, load into the setup cache so we can
# let runner template out future commands # let runner template out future commands
setup_ok = setup_results.get('contacted', {}) setup_ok = setup_results.get('contacted', {})

Loading…
Cancel
Save