Scott Baker | 4683159 | 2016-06-20 17:32:04 -0700 | [diff] [blame] | 1 | import os |
| 2 | import base64 |
| 3 | from collections import defaultdict |
Scott Baker | 71c20eb | 2017-03-14 21:35:18 -0700 | [diff] [blame] | 4 | from synchronizers.new_base.syncstep import SyncStep, DeferredException |
| 5 | from synchronizers.new_base.modelaccessor import * |
Scott Baker | 4683159 | 2016-06-20 17:32:04 -0700 | [diff] [blame] | 6 | from xos.logger import observer_logger as logger |
Scott Baker | 4683159 | 2016-06-20 17:32:04 -0700 | [diff] [blame] | 7 | import json |
| 8 | |
| 9 | class SyncVRouterTenant(SyncStep): |
| 10 | provides=[VRouterTenant] |
| 11 | observes = VRouterTenant |
| 12 | requested_interval=30 |
| 13 | playbook='sync_host.yaml' |
| 14 | |
| 15 | def get_fabric_onos_service(self): |
Scott Baker | 71c20eb | 2017-03-14 21:35:18 -0700 | [diff] [blame] | 16 | fs = FabricService.objects.first() |
Scott Baker | 7462a14 | 2017-07-19 16:34:29 -0700 | [diff] [blame] | 17 | for link in fs.subscribed_links.all(): |
| 18 | if link.provider_service_instance: |
| 19 | # cast from ServiceInstance to ONOSApp |
| 20 | apps = ONOSApp.objects.filter(id=link.provider_service_instance.id) |
| 21 | if apps: |
| 22 | # cast from Service to ONOSService |
| 23 | onos = ONOSService.objects.get(id=apps[0].owner.id) |
| 24 | return onos |
| 25 | return None |
Scott Baker | 4683159 | 2016-06-20 17:32:04 -0700 | [diff] [blame] | 26 | |
| 27 | def get_node_tag(self, node, tagname): |
Scott Baker | 8daf78d | 2017-03-15 11:39:44 -0700 | [diff] [blame] | 28 | tags = Tag.objects.filter(content_type=model_accessor.get_content_type_id(node), |
| 29 | object_id=node.id, |
| 30 | name=tagname) |
Scott Baker | 71c20eb | 2017-03-14 21:35:18 -0700 | [diff] [blame] | 31 | if tags: |
| 32 | return tags[0].value |
| 33 | else: |
Scott Baker | 4683159 | 2016-06-20 17:32:04 -0700 | [diff] [blame] | 34 | return None |
| 35 | |
Scott Baker | 71c20eb | 2017-03-14 21:35:18 -0700 | [diff] [blame] | 36 | def fetch_pending(self, deleted): |
Scott Baker | 8474a74 | 2017-06-15 10:34:07 -0700 | [diff] [blame] | 37 | # If fetch_pending is being called for delete, then just execute the standard delete logic. |
| 38 | if deleted: |
| 39 | return super(SyncVRouterTenant, self).fetch_pending(deleted) |
| 40 | |
Scott Baker | 71c20eb | 2017-03-14 21:35:18 -0700 | [diff] [blame] | 41 | fs = FabricService.objects.first() |
| 42 | if (not fs) or (not fs.autoconfig): |
| 43 | return None |
| 44 | |
| 45 | # TODO: Why is this a nonstandard synchronizer query? |
Scott Baker | 8474a74 | 2017-06-15 10:34:07 -0700 | [diff] [blame] | 46 | objs = VRouterTenant.objects.all() |
Scott Baker | 4683159 | 2016-06-20 17:32:04 -0700 | [diff] [blame] | 47 | |
Scott Baker | ef17d0c | 2017-02-20 08:45:45 -0800 | [diff] [blame] | 48 | objs = list(objs) |
| 49 | |
Andy Bavier | a3c7373 | 2016-06-27 15:24:59 -0400 | [diff] [blame] | 50 | # Check that each is a valid vCPE tenant or instance |
| 51 | for vroutertenant in objs: |
Scott Baker | 7462a14 | 2017-07-19 16:34:29 -0700 | [diff] [blame] | 52 | # TODO: hardcoded service dependency |
| 53 | vsg = self.get_vsg_subscriber(vroutertenant) |
| 54 | if vsg: |
| 55 | if not vsg.instance: |
Andy Bavier | a3c7373 | 2016-06-27 15:24:59 -0400 | [diff] [blame] | 56 | objs.remove(vroutertenant) |
| 57 | else: |
| 58 | # Maybe the VRouterTenant is for an instance |
Scott Baker | 71c20eb | 2017-03-14 21:35:18 -0700 | [diff] [blame] | 59 | # TODO: tenant_for_instance_id needs to be a real database field |
Andy Bavier | a3c7373 | 2016-06-27 15:24:59 -0400 | [diff] [blame] | 60 | instance_id = vroutertenant.get_attribute("tenant_for_instance_id") |
| 61 | if not instance_id: |
| 62 | objs.remove(vroutertenant) |
| 63 | else: |
| 64 | instance = Instance.objects.filter(id=instance_id)[0] |
| 65 | if not instance.instance_name: |
| 66 | objs.remove(vroutertenant) |
| 67 | |
Scott Baker | 4683159 | 2016-06-20 17:32:04 -0700 | [diff] [blame] | 68 | return objs |
| 69 | |
Scott Baker | 7462a14 | 2017-07-19 16:34:29 -0700 | [diff] [blame] | 70 | def get_vsg_subscriber(self, vroutertenant): |
| 71 | links = vroutertenant.provided_links.all() |
| 72 | for link in links: |
| 73 | if not link.subscriber_service_instance: |
| 74 | continue |
| 75 | # cast from ServiceInstance to VSGTEnant |
| 76 | vsgs = VSGTenant.objects.filter(id=link.subscriber_service_instance.id) |
| 77 | if vsgs: |
| 78 | return vsgs[0] |
| 79 | return None |
| 80 | |
Scott Baker | 4683159 | 2016-06-20 17:32:04 -0700 | [diff] [blame] | 81 | def map_sync_inputs(self, vroutertenant): |
| 82 | |
| 83 | fos = self.get_fabric_onos_service() |
| 84 | |
Scott Baker | 7462a14 | 2017-07-19 16:34:29 -0700 | [diff] [blame] | 85 | if not fos: |
| 86 | raise Exception("No fabric onos service") |
| 87 | |
Scott Baker | 4683159 | 2016-06-20 17:32:04 -0700 | [diff] [blame] | 88 | name = None |
| 89 | instance = None |
| 90 | # VRouterTenant setup is kind of hacky right now, we'll |
| 91 | # need to revisit. The idea is: |
| 92 | # * Look up the instance corresponding to the address |
| 93 | # * Look up the node running the instance |
| 94 | # * Get the "location" tag, push to the fabric |
Scott Baker | 7462a14 | 2017-07-19 16:34:29 -0700 | [diff] [blame] | 95 | |
| 96 | # TODO: hardcoded service dependency |
| 97 | vsg = self.get_vsg_subscriber(vroutertenant) |
| 98 | if vsg: |
Scott Baker | 71c20eb | 2017-03-14 21:35:18 -0700 | [diff] [blame] | 99 | instance = vsg.instance |
Scott Baker | 7462a14 | 2017-07-19 16:34:29 -0700 | [diff] [blame] | 100 | name = str(vsg) |
Scott Baker | 4683159 | 2016-06-20 17:32:04 -0700 | [diff] [blame] | 101 | else: |
Scott Baker | 4683159 | 2016-06-20 17:32:04 -0700 | [diff] [blame] | 102 | instance_id = vroutertenant.get_attribute("tenant_for_instance_id") |
Andy Bavier | a3c7373 | 2016-06-27 15:24:59 -0400 | [diff] [blame] | 103 | instance = Instance.objects.filter(id=instance_id)[0] |
| 104 | name = str(instance) |
Scott Baker | 4683159 | 2016-06-20 17:32:04 -0700 | [diff] [blame] | 105 | |
| 106 | node = instance.node |
| 107 | location = self.get_node_tag(node, "location") |
| 108 | |
Scott Baker | 71c20eb | 2017-03-14 21:35:18 -0700 | [diff] [blame] | 109 | if not location: |
| 110 | raise DeferredException("No location tag for node %s tenant %s -- skipping" % (str(node), str(vroutertenant))) |
| 111 | |
Andy Bavier | 82d8983 | 2016-06-28 15:31:06 -0400 | [diff] [blame] | 112 | # Create JSON |
| 113 | data = { |
| 114 | "%s/-1" % vroutertenant.public_mac : { |
| 115 | "basic" : { |
| 116 | "ips" : [ vroutertenant.public_ip ], |
| 117 | "location" : location |
| 118 | } |
| 119 | } |
| 120 | } |
| 121 | # Stupid Ansible... leading space so it doesn't think it's a dict |
| 122 | rest_body = " " + json.dumps(data) |
Andy Bavier | 92e3e00 | 2016-06-28 14:30:39 -0400 | [diff] [blame] | 123 | |
Scott Baker | 4683159 | 2016-06-20 17:32:04 -0700 | [diff] [blame] | 124 | # Is it a POST or DELETE? |
| 125 | |
Scott Baker | 4683159 | 2016-06-20 17:32:04 -0700 | [diff] [blame] | 126 | fields = { |
| 127 | 'rest_hostname': fos.rest_hostname, |
| 128 | 'rest_port': fos.rest_port, |
Scott Baker | 4683159 | 2016-06-20 17:32:04 -0700 | [diff] [blame] | 129 | 'rest_endpoint': "onos/v1/network/configuration/hosts", |
Andy Bavier | 82d8983 | 2016-06-28 15:31:06 -0400 | [diff] [blame] | 130 | 'rest_body': rest_body, |
Scott Baker | 4683159 | 2016-06-20 17:32:04 -0700 | [diff] [blame] | 131 | 'ansible_tag': '%s'%name, # name of ansible playbook |
| 132 | } |
| 133 | return fields |
| 134 | |
| 135 | def map_sync_outputs(self, controller_image, res): |
| 136 | pass |