Sapan Bhatia | 2810e03 | 2017-11-03 00:31:00 -0400 | [diff] [blame] | 1 | # Copyright 2017-present Open Networking Foundation |
| 2 | # |
| 3 | # Licensed under the Apache License, Version 2.0 (the "License"); |
| 4 | # you may not use this file except in compliance with the License. |
| 5 | # You may obtain a copy of the License at |
| 6 | # |
| 7 | # http://www.apache.org/licenses/LICENSE-2.0 |
| 8 | # |
| 9 | # Unless required by applicable law or agreed to in writing, software |
| 10 | # distributed under the License is distributed on an "AS IS" BASIS, |
| 11 | # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 12 | # See the License for the specific language governing permissions and |
| 13 | # limitations under the License. |
| 14 | |
| 15 | from synchronizers.new_base.modelaccessor import * |
Sapan Bhatia | 503ad90 | 2017-11-13 13:06:17 -0500 | [diff] [blame] | 16 | from synchronizers.new_base.model_policies.model_policy_tenantwithcontainer import Policy |
Sapan Bhatia | 2810e03 | 2017-11-03 00:31:00 -0400 | [diff] [blame] | 17 | from synchronizers.new_base.exceptions import * |
| 18 | |
| 19 | from xosconfig import Config |
| 20 | from multistructlog import create_logger |
| 21 | |
| 22 | log = create_logger(Config().get('logging')) |
| 23 | blueprints = Config().get('blueprints') |
| 24 | |
| 25 | def service_of_service_instance(si): |
| 26 | if si.endswith('Tenant'): |
| 27 | return si[:-len('Tenant')] + 'Service' |
| 28 | elif si.endswith('ServiceInstance'): |
| 29 | return si[:-len('ServiceInstance')] + 'Service' |
| 30 | else: |
| 31 | raise Exception('Could not translate service instance into service: %s'%si) |
| 32 | |
Sapan Bhatia | 503ad90 | 2017-11-13 13:06:17 -0500 | [diff] [blame] | 33 | class VEPCServiceInstancePolicy(Policy): |
Sapan Bhatia | 2810e03 | 2017-11-03 00:31:00 -0400 | [diff] [blame] | 34 | model_name = "VEPCServiceInstance" |
| 35 | |
Sapan Bhatia | 503ad90 | 2017-11-13 13:06:17 -0500 | [diff] [blame] | 36 | def __init__(self): |
| 37 | self.in_memory_instances = [] |
Sapan Bhatia | 491a080 | 2017-11-22 19:37:28 -0500 | [diff] [blame^] | 38 | self.network_map = {} |
| 39 | |
Sapan Bhatia | 503ad90 | 2017-11-13 13:06:17 -0500 | [diff] [blame] | 40 | super(VEPCServiceInstancePolicy, self).__init__() |
| 41 | |
Sapan Bhatia | 2810e03 | 2017-11-03 00:31:00 -0400 | [diff] [blame] | 42 | """TODO: Update the following to not be service-specific |
| 43 | This code assumes there is only one vendor installed |
| 44 | """ |
| 45 | def configure_service_instance(self, service_instance): |
| 46 | if service_instance.leaf_model_name == 'VSPGWUTenant': |
| 47 | vendor = VSPGWUVendor.objects.first() |
| 48 | if not vendor: |
| 49 | raise Exception('No VSPGWU vendors') |
| 50 | service_instance.vspgwu_vendor = vendor |
| 51 | elif service_instance.leaf_model_name == 'VSPGWCTenant': |
| 52 | vendor = VSPGWCVendor.objects.first() |
| 53 | if not vendor: |
| 54 | raise Exception('No VSPGWC vendors') |
| 55 | service_instance.vspgwc_vendor = vendor |
| 56 | |
| 57 | def child_service_instance_from_name(self, name): |
| 58 | service_instances = self.obj.child_serviceinstances.all() |
Sapan Bhatia | 503ad90 | 2017-11-13 13:06:17 -0500 | [diff] [blame] | 59 | service_instances.extend(self.in_memory_instances) |
Sapan Bhatia | 2810e03 | 2017-11-03 00:31:00 -0400 | [diff] [blame] | 60 | |
| 61 | try: |
| 62 | service_instance = next(s for s in service_instances if s.leaf_model_name == name) |
| 63 | except StopIteration: |
| 64 | service_instance = None |
| 65 | |
| 66 | return service_instance |
| 67 | |
| 68 | def get_service_for_service_instance(self, si): |
| 69 | service = service_of_service_instance(si) |
| 70 | service_class = getattr(Slice().stub, service) |
| 71 | service_obj = service_class.objects.first() # There's only one service object |
| 72 | return service_obj |
| 73 | |
Sapan Bhatia | 503ad90 | 2017-11-13 13:06:17 -0500 | [diff] [blame] | 74 | def create_service_instance(self, si): |
Sapan Bhatia | 2810e03 | 2017-11-03 00:31:00 -0400 | [diff] [blame] | 75 | service = self.get_service_for_service_instance(si) |
| 76 | if not service: |
| 77 | raise Exception('No service object for %s'%service) |
| 78 | |
| 79 | si_class = getattr(Slice().stub, si) |
Sapan Bhatia | 503ad90 | 2017-11-13 13:06:17 -0500 | [diff] [blame] | 80 | s = si_class(owner = service, name = 'epc-%s-%d'%(si.lower(), self.obj.id)) |
Sapan Bhatia | 2810e03 | 2017-11-03 00:31:00 -0400 | [diff] [blame] | 81 | s.master_serviceinstance = self.obj |
| 82 | |
| 83 | self.configure_service_instance(s) |
| 84 | s.save() |
Sapan Bhatia | 503ad90 | 2017-11-13 13:06:17 -0500 | [diff] [blame] | 85 | |
| 86 | self.in_memory_instances.append(s) |
Sapan Bhatia | 2810e03 | 2017-11-03 00:31:00 -0400 | [diff] [blame] | 87 | return s |
| 88 | |
Sapan Bhatia | 491a080 | 2017-11-22 19:37:28 -0500 | [diff] [blame^] | 89 | def create_service_instance_with_networks(self, si_name, networks): |
| 90 | instance = self.child_service_instance_from_name(si_name) |
| 91 | if not instance: |
| 92 | instance = self.create_service_instance(si_name) |
Sapan Bhatia | 2810e03 | 2017-11-03 00:31:00 -0400 | [diff] [blame] | 93 | |
Sapan Bhatia | 491a080 | 2017-11-22 19:37:28 -0500 | [diff] [blame^] | 94 | for n in networks: |
| 95 | one_and_only_slice_hopefully = instance.owner.slices.all()[0] |
| 96 | ns_object = NetworkSlice.objects.filter(network = n, slice = one_and_only_slice_hopefully) |
| 97 | if not ns_object: |
| 98 | ns_object = NetworkSlice(network = n, slice = one_and_only_slice_hopefully) |
| 99 | ns_object.save() |
| 100 | |
| 101 | return instance |
| 102 | |
| 103 | def create_link(self, src_instance, dst_instance): |
Sapan Bhatia | 2810e03 | 2017-11-03 00:31:00 -0400 | [diff] [blame] | 104 | src_service = self.get_service_for_service_instance(src) |
| 105 | dst_service = self.get_service_for_service_instance(dst) |
| 106 | |
| 107 | service_dependency = ServiceDependency.objects.filter(provider_service_id = dst_service.id, subscriber_service_id = src_service.id) |
| 108 | if not service_dependency: |
| 109 | service_dependency = ServiceDependency(provider_service = dst_service, subscriber_service = src_service) |
| 110 | service_dependency.save() |
| 111 | |
| 112 | service_instance_link = ServiceInstanceLink.objects.filter(provider_service_instance_id = dst_instance.id, subscriber_service_instance_id = src_instance.id) |
| 113 | if not service_instance_link: |
| 114 | service_instance_link = ServiceInstanceLink(provider_service_instance = dst_instance, subscriber_service_instance = src_instance) |
| 115 | service_instance_link.save() |
| 116 | |
| 117 | def recursive_create_links(self, blueprint, src): |
Sapan Bhatia | 491a080 | 2017-11-22 19:37:28 -0500 | [diff] [blame^] | 118 | for node in blueprint: |
| 119 | k = node['name'] |
| 120 | networks = node.get('networks', []) |
| 121 | instance = self.create_service_instance_with_networks(k, networks) |
| 122 | |
Sapan Bhatia | 503ad90 | 2017-11-13 13:06:17 -0500 | [diff] [blame] | 123 | if src: |
Sapan Bhatia | 491a080 | 2017-11-22 19:37:28 -0500 | [diff] [blame^] | 124 | self.create_link(src, instance) |
Sapan Bhatia | 2810e03 | 2017-11-03 00:31:00 -0400 | [diff] [blame] | 125 | |
Sapan Bhatia | 491a080 | 2017-11-22 19:37:28 -0500 | [diff] [blame^] | 126 | links = node.get('links', []) |
| 127 | self.recursive_create_links(links, instance) |
Sapan Bhatia | 2810e03 | 2017-11-03 00:31:00 -0400 | [diff] [blame] | 128 | |
Sapan Bhatia | 491a080 | 2017-11-22 19:37:28 -0500 | [diff] [blame^] | 129 | def create_epc_network(self, n): |
| 130 | network_name = n['name'] |
| 131 | site_name = self.obj.site.name |
| 132 | slice_name = '%s_%s'%(site_name, network_name) |
| 133 | |
| 134 | slice = Slice.objects.get(name=slice_name) |
| 135 | if not slice: |
| 136 | slice = Slice(name = slice_name, default_isolation = "vm", network = "noauto", site = self.obj.site) |
| 137 | slice.save() |
| 138 | |
| 139 | net = Network.objects.get(name=network_name) |
| 140 | if not net: |
| 141 | net = Network(name = network_name, subnet = n['subnet'], permit_all_slices = n.get('permit_all_slices', False), template = n.get('template', 'public'), owner = slice) |
| 142 | net.save() |
| 143 | elif net.subnet != n['subnet']: |
| 144 | net.subnet = n['subnet'] |
| 145 | net.save() |
| 146 | |
| 147 | self.network_map[network_name] = net |
| 148 | |
| 149 | def create_networks(self, networks): |
| 150 | for n in networks: |
| 151 | self.create_epc_network(n) |
| 152 | |
| 153 | def create_networks_and_child_services(self, service_instance): |
Sapan Bhatia | 2810e03 | 2017-11-03 00:31:00 -0400 | [diff] [blame] | 154 | self.obj = service_instance |
| 155 | # Create service graph based on blueprint |
| 156 | chosen_blueprint = service_instance.blueprint |
| 157 | try: |
| 158 | blueprint = next(b for b in blueprints if b['name'] == chosen_blueprint) |
| 159 | except StopIteration: |
| 160 | log.error('Chosen blueprint (%s) not found' % chosen_blueprint) |
| 161 | |
Sapan Bhatia | 491a080 | 2017-11-22 19:37:28 -0500 | [diff] [blame^] | 162 | self.create_networks(blueprint['networks']) |
Sapan Bhatia | 2810e03 | 2017-11-03 00:31:00 -0400 | [diff] [blame] | 163 | self.recursive_create_links(blueprint['graph'], None) |
| 164 | |
Sapan Bhatia | 503ad90 | 2017-11-13 13:06:17 -0500 | [diff] [blame] | 165 | def handle_create(self, service_instance): |
| 166 | self.handle_update(service_instance) |
| 167 | |
Sapan Bhatia | 2810e03 | 2017-11-03 00:31:00 -0400 | [diff] [blame] | 168 | def handle_update(self, service_instance): |
Sapan Bhatia | 491a080 | 2017-11-22 19:37:28 -0500 | [diff] [blame^] | 169 | self.create_networks_and_child_services(service_instance) |
Sapan Bhatia | 2810e03 | 2017-11-03 00:31:00 -0400 | [diff] [blame] | 170 | |
Sapan Bhatia | 2810e03 | 2017-11-03 00:31:00 -0400 | [diff] [blame] | 171 | def handle_delete(self, service_instance): |
Sapan Bhatia | 503ad90 | 2017-11-13 13:06:17 -0500 | [diff] [blame] | 172 | raise Exception("Not implemented") |