mirror of
https://github.com/ansible/awx.git
synced 2026-02-12 07:04:45 -03:30
29d60844a8
* Track host_status_counts and use that to process notifications * Remove now unused setting * Back out changes to callback class not needed after all * Skirt the need for duck typing by leaning on the cached field * Delete tests for deleted task * Revert "Back out changes to callback class not needed after all" This reverts commit 3b8ae350d218991d42bffd65ce4baac6f41926b2. * Directly hardcode stats_event_type for callback class * Fire notifications if stats event was never sent * Remove test content for deleted methods * Add placeholder for when no hosts matched * Make field default be None, denote events processed with empty dict * Make UI process null value for host_status_counts * Fix tracking of EOF dispatch for system jobs * Reorganize EVENT_MAP into class properties * Consolidate conditional I missed from EVENT_MAP refactor * Give up on the null condition, also applies for empty hosts * Remove cls position argument not being used * Move wrapup method out of class, add tests
44 lines
1.7 KiB
Python
44 lines
1.7 KiB
Python
from django.db import transaction, DatabaseError, InterfaceError
|
|
|
|
import logging
|
|
import time
|
|
|
|
|
|
logger = logging.getLogger('awx.main.tasks.utils')
|
|
|
|
|
|
def update_model(model, pk, _attempt=0, _max_attempts=5, select_for_update=False, **updates):
|
|
"""Reload the model instance from the database and update the
|
|
given fields.
|
|
"""
|
|
try:
|
|
with transaction.atomic():
|
|
# Retrieve the model instance.
|
|
if select_for_update:
|
|
instance = model.objects.select_for_update().get(pk=pk)
|
|
else:
|
|
instance = model.objects.get(pk=pk)
|
|
|
|
# Update the appropriate fields and save the model
|
|
# instance, then return the new instance.
|
|
if updates:
|
|
update_fields = ['modified']
|
|
for field, value in updates.items():
|
|
setattr(instance, field, value)
|
|
update_fields.append(field)
|
|
if field == 'status':
|
|
update_fields.append('failed')
|
|
instance.save(update_fields=update_fields)
|
|
return instance
|
|
except (DatabaseError, InterfaceError) as e:
|
|
# Log out the error to the debug logger.
|
|
logger.debug('Database error updating %s, retrying in 5 seconds (retry #%d): %s', model._meta.object_name, _attempt + 1, e)
|
|
|
|
# Attempt to retry the update, assuming we haven't already
|
|
# tried too many times.
|
|
if _attempt < _max_attempts:
|
|
time.sleep(5)
|
|
return update_model(model, pk, _attempt=_attempt + 1, _max_attempts=_max_attempts, **updates)
|
|
else:
|
|
logger.error('Failed to update %s after %d retries.', model._meta.object_name, _attempt)
|