From 99140a941328646001bcd2a04ae8f0c8221d77d5 Mon Sep 17 00:00:00 2001 From: Jeremy Stretch Date: Wed, 30 Aug 2023 08:47:20 -0400 Subject: [PATCH] Specify batch size for cached counter migrations --- netbox/dcim/migrations/0176_device_component_counters.py | 6 +++--- .../dcim/migrations/0178_virtual_chassis_member_counter.py | 2 +- .../migrations/0035_virtualmachine_interface_count.py | 2 +- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/netbox/dcim/migrations/0176_device_component_counters.py b/netbox/dcim/migrations/0176_device_component_counters.py index b570ddbd5..a911d7fd7 100644 --- a/netbox/dcim/migrations/0176_device_component_counters.py +++ b/netbox/dcim/migrations/0176_device_component_counters.py @@ -6,7 +6,7 @@ import utilities.fields def recalculate_device_counts(apps, schema_editor): Device = apps.get_model("dcim", "Device") - devices = list(Device.objects.all().annotate( + devices = Device.objects.annotate( _console_port_count=Count('consoleports', distinct=True), _console_server_port_count=Count('consoleserverports', distinct=True), _power_port_count=Count('powerports', distinct=True), @@ -17,7 +17,7 @@ def recalculate_device_counts(apps, schema_editor): _device_bay_count=Count('devicebays', distinct=True), _module_bay_count=Count('modulebays', distinct=True), _inventory_item_count=Count('inventoryitems', distinct=True), - )) + ) for device in devices: device.console_port_count = device._console_port_count @@ -42,7 +42,7 @@ def recalculate_device_counts(apps, schema_editor): 'device_bay_count', 'module_bay_count', 'inventory_item_count', - ]) + ], batch_size=100) class Migration(migrations.Migration): diff --git a/netbox/dcim/migrations/0178_virtual_chassis_member_counter.py b/netbox/dcim/migrations/0178_virtual_chassis_member_counter.py index e3ade1344..a8d0b41bf 100644 --- a/netbox/dcim/migrations/0178_virtual_chassis_member_counter.py +++ b/netbox/dcim/migrations/0178_virtual_chassis_member_counter.py @@ -12,7 +12,7 @@ def populate_virtualchassis_members(apps, schema_editor): for vc in vcs: vc.member_count = vc._member_count - VirtualChassis.objects.bulk_update(vcs, ['member_count']) + VirtualChassis.objects.bulk_update(vcs, ['member_count'], batch_size=100) class Migration(migrations.Migration): diff --git a/netbox/virtualization/migrations/0035_virtualmachine_interface_count.py b/netbox/virtualization/migrations/0035_virtualmachine_interface_count.py index 5f52d32e0..a5c09ba0b 100644 --- a/netbox/virtualization/migrations/0035_virtualmachine_interface_count.py +++ b/netbox/virtualization/migrations/0035_virtualmachine_interface_count.py @@ -12,7 +12,7 @@ def populate_virtualmachine_counts(apps, schema_editor): 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):