External-Mirrors/awx
2
0
Fork 0
mirror of https://github.com/ansible/awx.git synced 2026-05-07 09:27:36 -02:30
Code Issues Packages Projects Releases Wiki Activity

replace the smart inventory membership lock with a new Django 2.2 flag

Browse Source
This commit is contained in:
Ryan Petrello
2019-07-22 11:11:36 -04:00
parent f6c357659d
commit 1cedf244b7
1 changed files with 19 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
awx/main/tasks.py
View File

@@ -602,26 +602,25 @@ def update_inventory_computed_fields(inventory_id, should_update_hosts=True):
def update_smart_memberships_for_inventory(smart_inventory): def update_smart_memberships_for_inventory(smart_inventory):
with advisory_lock('update_smart_memberships_for_inventory-{}'.format(smart_inventory.id)): current = set(SmartInventoryMembership.objects.filter(inventory=smart_inventory).values_list('host_id', flat=True))
current = set(SmartInventoryMembership.objects.filter(inventory=smart_inventory).values_list('host_id', flat=True)) new = set(smart_inventory.hosts.values_list('id', flat=True))
new = set(smart_inventory.hosts.values_list('id', flat=True)) additions = new - current
additions = new - current removals = current - new
removals = current - new if additions or removals:
if additions or removals: with transaction.atomic():
with transaction.atomic(): if removals:
if removals: SmartInventoryMembership.objects.filter(inventory=smart_inventory, host_id__in=removals).delete()
SmartInventoryMembership.objects.filter(inventory=smart_inventory, host_id__in=removals).delete() if additions:
if additions: add_for_inventory = [
add_for_inventory = [ SmartInventoryMembership(inventory_id=smart_inventory.id, host_id=host_id)
SmartInventoryMembership(inventory_id=smart_inventory.id, host_id=host_id) for host_id in additions
for host_id in additions ]
] SmartInventoryMembership.objects.bulk_create(add_for_inventory, ignore_conflicts=True)
SmartInventoryMembership.objects.bulk_create(add_for_inventory) logger.debug('Smart host membership cached for {}, {} additions, {} removals, {} total count.'.format(
logger.debug('Smart host membership cached for {}, {} additions, {} removals, {} total count.'.format( smart_inventory.pk, len(additions), len(removals), len(new)
smart_inventory.pk, len(additions), len(removals), len(new) ))
)) return True # changed
return True # changed return False
return False
@task() @task()