Scott Baker | af599eb | 2017-03-21 12:43:26 -0700 | [diff] [blame] | 1 | from synchronizers.new_base.modelaccessor import * |
| 2 | from collections import defaultdict |
Scott Baker | 46a3ee9 | 2017-05-30 16:44:56 -0700 | [diff] [blame^] | 3 | from synchronizers.new_base.policy import Policy |
Scott Baker | b63ea79 | 2016-08-11 10:24:48 -0700 | [diff] [blame] | 4 | |
Scott Baker | 46a3ee9 | 2017-05-30 16:44:56 -0700 | [diff] [blame^] | 5 | class NetworkPolicy(Policy): |
| 6 | model_name = "Network" |
Scott Baker | b63ea79 | 2016-08-11 10:24:48 -0700 | [diff] [blame] | 7 | |
Scott Baker | 46a3ee9 | 2017-05-30 16:44:56 -0700 | [diff] [blame^] | 8 | def handle_create(self, network): |
| 9 | return self.handle_update(network) |
| 10 | |
| 11 | def handle_update(self, network): |
| 12 | # network controllers are not visible to users. We must ensure |
| 13 | # networks are deployed at all deploymets available to their slices. |
Scott Baker | a0a688a | 2017-03-28 11:59:56 -0700 | [diff] [blame] | 14 | |
| 15 | # TODO: should be possible to query only the ControllerSlice objects |
| 16 | # associated with network.owner rather than iterating through |
| 17 | # all ControllerSlice. |
| 18 | |
Scott Baker | 46a3ee9 | 2017-05-30 16:44:56 -0700 | [diff] [blame^] | 19 | slice_controllers = ControllerSlice.objects.all() |
| 20 | slice_deploy_lookup = defaultdict(list) |
| 21 | for slice_controller in slice_controllers: |
| 22 | slice_deploy_lookup[slice_controller.slice.id].append(slice_controller.controller) |
Scott Baker | b63ea79 | 2016-08-11 10:24:48 -0700 | [diff] [blame] | 23 | |
Scott Baker | 46a3ee9 | 2017-05-30 16:44:56 -0700 | [diff] [blame^] | 24 | network_controllers = ControllerNetwork.objects.all() |
| 25 | network_deploy_lookup = defaultdict(list) |
| 26 | for network_controller in network_controllers: |
| 27 | network_deploy_lookup[network_controller.network.id].append(network_controller.controller.id) |
Scott Baker | b63ea79 | 2016-08-11 10:24:48 -0700 | [diff] [blame] | 28 | |
Scott Baker | 46a3ee9 | 2017-05-30 16:44:56 -0700 | [diff] [blame^] | 29 | expected_controllers = slice_deploy_lookup[network.owner.id] |
| 30 | for expected_controller in expected_controllers: |
| 31 | if network.id not in network_deploy_lookup or expected_controller.id not in network_deploy_lookup[network.id]: |
| 32 | lazy_blocked=True |
Scott Baker | b63ea79 | 2016-08-11 10:24:48 -0700 | [diff] [blame] | 33 | |
Scott Baker | 46a3ee9 | 2017-05-30 16:44:56 -0700 | [diff] [blame^] | 34 | # check and see if some instance already exists |
| 35 | for networkslice in network.networkslices.all(): |
| 36 | found = False |
| 37 | for instance in networkslice.slice.instances.all(): |
| 38 | if instance.node.site_deployment.controller.id == expected_controller.id: |
| 39 | found = True |
| 40 | if found: |
| 41 | self.logger.info("MODEL_POLICY: network %s setting lazy_blocked to false because instance on controller already exists" % network) |
| 42 | lazy_blocked=False |
Scott Baker | b63ea79 | 2016-08-11 10:24:48 -0700 | [diff] [blame] | 43 | |
Scott Baker | 46a3ee9 | 2017-05-30 16:44:56 -0700 | [diff] [blame^] | 44 | nd = ControllerNetwork(network=network, controller=expected_controller, lazy_blocked=lazy_blocked) |
| 45 | self.logger.info("MODEL POLICY: network %s create ControllerNetwork %s lazy_blocked %s" % (network, nd, lazy_blocked)) |
| 46 | if network.subnet: |
| 47 | # XXX: Possibly unpredictable behavior if there is |
| 48 | # more than one ControllerNetwork and the subnet |
| 49 | # is specified. |
| 50 | nd.subnet = network.subnet |
| 51 | nd.save() |