blob: bcda884890cb32cd467c7cfc06a65b743c994396 [file] [log] [blame]
from synchronizers.new_base.modelaccessor import *
from synchronizers.new_base.policy import Policy
class SlicePolicy(Policy):
model_name = "Slice"
def handle_create(self, slice):
return self.handle_update(slice)
def handle_update(self, slice):
support_nat_net = False # Assume we're using VTN rather than nat-net
# slice = Slice.get(slice_id)
controller_slices = ControllerSlice.objects.filter(slice_id=slice.id)
existing_controllers = [cs.controller for cs in controller_slices]
existing_controllers_ids = [x.id for x in existing_controllers]
self.logger.info("MODEL POLICY: slice existing_controllers=%s" % existing_controllers)
all_controllers = Controller.objects.all()
for controller in all_controllers:
if controller.id not in existing_controllers_ids:
self.logger.info("MODEL POLICY: slice adding controller %s" % controller)
sd = ControllerSlice(slice=slice, controller=controller)
sd.save()
if slice.network in ["host", "bridged"]:
# Host and Bridged docker containers need no networks and they will
# only get in the way.
self.logger.info("MODEL POLICY: Skipping network creation")
elif slice.network in ["noauto"]:
# do nothing
pass
else:
# make sure slice has at least 1 public and 1 private networkd
public_nets = []
private_nets = []
networks = Network.objects.filter(owner_id=slice.id)
for network in networks:
if not network.autoconnect:
continue
if network.template.name == 'Public dedicated IPv4':
public_nets.append(network)
elif network.template.name == 'Public shared IPv4':
public_nets.append(network)
elif network.template.name == 'Private':
private_nets.append(network)
if support_nat_net and (not public_nets):
# ensure there is at least one public network, and default it to dedicated
nat_net = Network(
name = slice.name+'-nat',
template = NetworkTemplate.objects.get(name='Public shared IPv4'),
owner = slice
)
if slice.exposed_ports:
nat_net.ports = slice.exposed_ports
nat_net.save()
public_nets.append(nat_net)
self.logger.info("MODEL POLICY: slice %s made nat-net" % slice)
if not private_nets:
private_net = Network(
name = slice.name+'-private',
template = NetworkTemplate.objects.get(name='Private'),
owner = slice
)
private_net.save()
self.logger.info("MODEL POLICY: slice %s made private net" % slice)
private_nets = [private_net]
# create slice networks
public_net_slice = None
private_net_slice = None
public_net_ids = [x.id for x in public_nets]
private_net_ids = [x.id for x in private_nets]
net_slices = NetworkSlice.objects.filter(slice_id=slice.id)
net_slices = [x for x in net_slices if x.network_id in public_net_ids+private_net_ids]
for net_slice in net_slices:
if net_slice.network.id in public_nets_ids:
public_net_slice = net_slice
elif net_slice.network.id in private_nets_ids:
private_net_slice = net_slice
if support_nat_net and (not public_net_slice):
public_net_slice = NetworkSlice(slice=slice, network=public_nets[0])
public_net_slice.save()
self.logger.info("MODEL POLICY: slice %s made public_net_slice" % slice)
if not private_net_slice:
private_net_slice = NetworkSlice(slice=slice, network=private_nets[0])
private_net_slice.save()
self.logger.info("MODEL POLICY: slice %s made private_net_slice" % slice)
# TODO: This feels redundant with the reaper
def handle_delete(slice):
public_nets = []
private_net = None
networks = Network.objects.filter(owner_id=slice.id)
for n in networks:
n.delete()
# Note that sliceprivileges and slicecontrollers are autodeleted, through the dependency graph