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): |
| 16 | fos = None |
Scott Baker | 71c20eb | 2017-03-14 21:35:18 -0700 | [diff] [blame] | 17 | fs = FabricService.objects.first() |
Scott Baker | 4683159 | 2016-06-20 17:32:04 -0700 | [diff] [blame] | 18 | if fs.subscribed_tenants.exists(): |
Scott Baker | 71c20eb | 2017-03-14 21:35:18 -0700 | [diff] [blame] | 19 | app = fs.subscribed_tenants.first() |
Scott Baker | 4683159 | 2016-06-20 17:32:04 -0700 | [diff] [blame] | 20 | if app.provider_service: |
| 21 | ps = app.provider_service |
Scott Baker | 71c20eb | 2017-03-14 21:35:18 -0700 | [diff] [blame] | 22 | fos = ONOSService.objects.filter(id=ps.id)[0] |
Scott Baker | 4683159 | 2016-06-20 17:32:04 -0700 | [diff] [blame] | 23 | return fos |
| 24 | |
| 25 | def get_node_tag(self, node, tagname): |
Scott Baker | 8daf78d | 2017-03-15 11:39:44 -0700 | [diff] [blame] | 26 | tags = Tag.objects.filter(content_type=model_accessor.get_content_type_id(node), |
| 27 | object_id=node.id, |
| 28 | name=tagname) |
Scott Baker | 71c20eb | 2017-03-14 21:35:18 -0700 | [diff] [blame] | 29 | if tags: |
| 30 | return tags[0].value |
| 31 | else: |
Scott Baker | 4683159 | 2016-06-20 17:32:04 -0700 | [diff] [blame] | 32 | return None |
| 33 | |
Scott Baker | 71c20eb | 2017-03-14 21:35:18 -0700 | [diff] [blame] | 34 | def fetch_pending(self, deleted): |
| 35 | fs = FabricService.objects.first() |
| 36 | if (not fs) or (not fs.autoconfig): |
| 37 | return None |
| 38 | |
| 39 | # TODO: Why is this a nonstandard synchronizer query? |
| 40 | |
Scott Baker | 4683159 | 2016-06-20 17:32:04 -0700 | [diff] [blame] | 41 | if (not deleted): |
Scott Baker | 71c20eb | 2017-03-14 21:35:18 -0700 | [diff] [blame] | 42 | objs = VRouterTenant.objects.all() |
Scott Baker | 4683159 | 2016-06-20 17:32:04 -0700 | [diff] [blame] | 43 | else: |
Scott Baker | 8daf78d | 2017-03-15 11:39:44 -0700 | [diff] [blame] | 44 | objs = super(SyncVRouterTenant, self).fetch_pending(deleted) |
Scott Baker | 4683159 | 2016-06-20 17:32:04 -0700 | [diff] [blame] | 45 | |
Scott Baker | ef17d0c | 2017-02-20 08:45:45 -0800 | [diff] [blame] | 46 | objs = list(objs) |
| 47 | |
Andy Bavier | a3c7373 | 2016-06-27 15:24:59 -0400 | [diff] [blame] | 48 | # Check that each is a valid vCPE tenant or instance |
| 49 | for vroutertenant in objs: |
| 50 | # Do we have a vCPE subscriber_tenant? |
| 51 | if vroutertenant.subscriber_tenant: |
| 52 | sub = vroutertenant.subscriber_tenant |
Scott Baker | 71c20eb | 2017-03-14 21:35:18 -0700 | [diff] [blame] | 53 | if sub.kind != 'vCPE': |
Andy Bavier | a3c7373 | 2016-06-27 15:24:59 -0400 | [diff] [blame] | 54 | objs.remove(vroutertenant) |
Scott Baker | 71c20eb | 2017-03-14 21:35:18 -0700 | [diff] [blame] | 55 | else: |
| 56 | # coerce the subscriber tenant over to the VSGTenant |
| 57 | vsg = VSGTenant.objects.filter(id=sub.id).first() |
| 58 | if not vsg.instance: |
| 59 | objs.remove(vroutertenant) |
Andy Bavier | a3c7373 | 2016-06-27 15:24:59 -0400 | [diff] [blame] | 60 | else: |
| 61 | # Maybe the VRouterTenant is for an instance |
Scott Baker | 71c20eb | 2017-03-14 21:35:18 -0700 | [diff] [blame] | 62 | # TODO: tenant_for_instance_id needs to be a real database field |
Andy Bavier | a3c7373 | 2016-06-27 15:24:59 -0400 | [diff] [blame] | 63 | instance_id = vroutertenant.get_attribute("tenant_for_instance_id") |
| 64 | if not instance_id: |
| 65 | objs.remove(vroutertenant) |
| 66 | else: |
| 67 | instance = Instance.objects.filter(id=instance_id)[0] |
| 68 | if not instance.instance_name: |
| 69 | objs.remove(vroutertenant) |
| 70 | |
Scott Baker | 4683159 | 2016-06-20 17:32:04 -0700 | [diff] [blame] | 71 | return objs |
| 72 | |
| 73 | def map_sync_inputs(self, vroutertenant): |
| 74 | |
| 75 | fos = self.get_fabric_onos_service() |
| 76 | |
| 77 | name = None |
| 78 | instance = None |
| 79 | # VRouterTenant setup is kind of hacky right now, we'll |
| 80 | # need to revisit. The idea is: |
| 81 | # * Look up the instance corresponding to the address |
| 82 | # * Look up the node running the instance |
| 83 | # * Get the "location" tag, push to the fabric |
Andy Bavier | a3c7373 | 2016-06-27 15:24:59 -0400 | [diff] [blame] | 84 | if vroutertenant.subscriber_tenant: |
Scott Baker | 4683159 | 2016-06-20 17:32:04 -0700 | [diff] [blame] | 85 | sub = vroutertenant.subscriber_tenant |
Scott Baker | 71c20eb | 2017-03-14 21:35:18 -0700 | [diff] [blame] | 86 | assert(sub.kind == 'vCPE') |
| 87 | vsg = VSGTenant.objects.filter(id=sub.id).first() |
| 88 | instance = vsg.instance |
Andy Bavier | a3c7373 | 2016-06-27 15:24:59 -0400 | [diff] [blame] | 89 | name = str(sub) |
Scott Baker | 4683159 | 2016-06-20 17:32:04 -0700 | [diff] [blame] | 90 | else: |
Scott Baker | 4683159 | 2016-06-20 17:32:04 -0700 | [diff] [blame] | 91 | instance_id = vroutertenant.get_attribute("tenant_for_instance_id") |
Andy Bavier | a3c7373 | 2016-06-27 15:24:59 -0400 | [diff] [blame] | 92 | instance = Instance.objects.filter(id=instance_id)[0] |
| 93 | name = str(instance) |
Scott Baker | 4683159 | 2016-06-20 17:32:04 -0700 | [diff] [blame] | 94 | |
| 95 | node = instance.node |
| 96 | location = self.get_node_tag(node, "location") |
| 97 | |
Scott Baker | 71c20eb | 2017-03-14 21:35:18 -0700 | [diff] [blame] | 98 | if not location: |
| 99 | raise DeferredException("No location tag for node %s tenant %s -- skipping" % (str(node), str(vroutertenant))) |
| 100 | |
Andy Bavier | 82d8983 | 2016-06-28 15:31:06 -0400 | [diff] [blame] | 101 | # Create JSON |
| 102 | data = { |
| 103 | "%s/-1" % vroutertenant.public_mac : { |
| 104 | "basic" : { |
| 105 | "ips" : [ vroutertenant.public_ip ], |
| 106 | "location" : location |
| 107 | } |
| 108 | } |
| 109 | } |
| 110 | # Stupid Ansible... leading space so it doesn't think it's a dict |
| 111 | rest_body = " " + json.dumps(data) |
Andy Bavier | 92e3e00 | 2016-06-28 14:30:39 -0400 | [diff] [blame] | 112 | |
Scott Baker | 4683159 | 2016-06-20 17:32:04 -0700 | [diff] [blame] | 113 | # Is it a POST or DELETE? |
| 114 | |
Scott Baker | 4683159 | 2016-06-20 17:32:04 -0700 | [diff] [blame] | 115 | fields = { |
| 116 | 'rest_hostname': fos.rest_hostname, |
| 117 | 'rest_port': fos.rest_port, |
Scott Baker | 4683159 | 2016-06-20 17:32:04 -0700 | [diff] [blame] | 118 | 'rest_endpoint': "onos/v1/network/configuration/hosts", |
Andy Bavier | 82d8983 | 2016-06-28 15:31:06 -0400 | [diff] [blame] | 119 | 'rest_body': rest_body, |
Scott Baker | 4683159 | 2016-06-20 17:32:04 -0700 | [diff] [blame] | 120 | 'ansible_tag': '%s'%name, # name of ansible playbook |
| 121 | } |
| 122 | return fields |
| 123 | |
| 124 | def map_sync_outputs(self, controller_image, res): |
| 125 | pass |