Fixes #13605: Specify batch size for cached counter migrations (#13610)

* Specify batch size for cached counter migrations

* Remove list() casting of querysets
This commit is contained in:
Jeremy Stretch
2023-08-30 09:18:24 -04:00
committed by GitHub
parent bccf89a9d7
commit 6faf5d80f6
3 changed files with 7 additions and 7 deletions

View File

@@ -7,12 +7,12 @@ import utilities.fields
def populate_virtualmachine_counts(apps, schema_editor):
VirtualMachine = apps.get_model('virtualization', 'VirtualMachine')
vms = list(VirtualMachine.objects.annotate(_interface_count=Count('interfaces', distinct=True)))
vms = VirtualMachine.objects.annotate(_interface_count=Count('interfaces', distinct=True))
for vm in vms:
vm.interface_count = vm._interface_count
VirtualMachine.objects.bulk_update(vms, ['interface_count'])
VirtualMachine.objects.bulk_update(vms, ['interface_count'], batch_size=100)
class Migration(migrations.Migration):