External-Mirrors/awx
2
0
Fork 0
mirror of https://github.com/ansible/awx.git synced 2026-04-22 10:20:24 -02:30
Code Issues Packages Projects Releases Wiki Activity

properly handle host summary bulk updates if hosts go missing

Browse Source
This commit is contained in:
Ryan Petrello
2020-05-12 09:38:59 -04:00
parent e7347d15c1
commit 3cb2475307
2 changed files with 51 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
awx/main/models/events.py
View File

@@ -489,25 +489,23 @@ class JobEvent(BasePlaybookEvent):
return
job = self.job
from awx.main.models.jobs import JobHostSummary # circular import
from awx.main.models import Host, JobHostSummary # circular import
existing = Host.objects.filter(id__in=self.host_map.values()).values_list('id', flat=True)
summaries = dict()
for host in hostnames:
host_id = self.host_map.get(host, None)
if host_id not in existing:
host_id = None
host_stats = {}
for stat in ('changed', 'dark', 'failures', 'ignored', 'ok', 'processed', 'rescued', 'skipped'):
try:
host_stats[stat] = self.event_data.get(stat, {}).get(host, 0)
except AttributeError: # in case event_data[stat] isn't a dict.
pass
host_id = self.host_map.get(host, None)
summaries.setdefault(
(host_id, host),
JobHostSummary(created=now(), modified=now(), job_id=job.id, host_id=host_id, host_name=host)
summaries[(host_id, host)] = JobHostSummary(
created=now(), modified=now(), job_id=job.id, host_id=host_id, host_name=host, **host_stats
)
host_summary = summaries[(host_id, host)]
for stat, value in host_stats.items():
if getattr(host_summary, stat) != value:
setattr(host_summary, stat, value)
JobHostSummary.objects.bulk_create(summaries.values())