1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
###############################################################################
# Copyright (c) 2017 Koren Lev (Cisco Systems), Yaron Yogev (Cisco Systems) #
# and others #
# #
# All rights reserved. This program and the accompanying materials #
# are made available under the terms of the Apache License, Version 2.0 #
# which accompanies this distribution, and is available at #
# http://www.apache.org/licenses/LICENSE-2.0 #
###############################################################################
from discover.events.event_base import EventResult
from discover.events.event_delete_base import EventDeleteBase
from discover.fetchers.api.api_fetch_host_instances import ApiFetchHostInstances
class EventPortDelete(EventDeleteBase):
def delete_port(self, env, port_id):
port_doc = self.inv.get_by_id(env, port_id)
if not port_doc:
self.log.info("Port document not found, aborting port deleting.")
return EventResult(result=False, retry=False)
# if port is binding to a instance, instance document needs to be updated.
if 'compute' in port_doc['device_owner']:
self.log.info("update instance document to which port is binding.")
self.update_instance(env, port_doc)
# delete port document
self.inv.delete('inventory', {'id': port_id})
# delete vnic and related document
vnic_doc = self.inv.get_by_field(env, 'vnic', 'mac_address', port_doc['mac_address'], get_single=True)
if not vnic_doc:
self.log.info("Vnic document not found, aborting vnic deleting.")
return EventResult(result=False, retry=False)
result = self.delete_handler(env, vnic_doc['id'], 'vnic')
result.related_object = port_id
result.display_context = port_doc.get('network_id')
self.log.info('Finished port deleting')
return result
def update_instance(self, env, port_doc):
# update instance document if port
network_id = port_doc['network_id']
instance_doc = self.inv.get_by_field(env, 'instance', 'network_info.id', port_doc['id'], get_single=True)
if instance_doc:
port_num = 0
for port in instance_doc['network_info']:
if port['network']['id'] == network_id:
port_num += 1
if port['id'] == port_doc['id']:
instance_doc['network_info'].remove(port)
self.log.info("update network information of instance document.")
if port_num == 1:
# remove network information only when last port in network will be deleted.
instance_doc['network'].remove(network_id)
# update instance mac address.
if port_doc['mac_address'] == instance_doc['mac_address']:
instance_fetcher = ApiFetchHostInstances()
instance_fetcher.setup(env=env, origin=self.origin)
host_id = port_doc['binding:host_id']
instance_id = port_doc['device_id']
instance_docs = instance_fetcher.get(host_id + '-')
instance = next(filter(lambda i: i['id'] == instance_id, instance_docs), None)
if instance:
if 'mac_address' not in instance:
instance_doc['mac_address'] = None
self.log.info("update mac_address:%s of instance document." % instance_doc['mac_address'])
self.inv.set(instance_doc)
else:
self.log.info("No instance document binding to network:%s." % network_id)
def handle(self, env, notification):
port_id = notification['payload']['port_id']
return self.delete_port(env, port_id)
|
