1 ##############################################################################
2 # Copyright (c) 2016 NEC Corporation and others.
4 # All rights reserved. This program and the accompanying materials
5 # are made available under the terms of the Apache License, Version 2.0
6 # which accompanies this distribution, and is available at
7 # http://www.apache.org/licenses/LICENSE-2.0
8 ##############################################################################
12 from flask import Flask
13 from flask import request
15 import logger as doctor_log
19 import novaclient.client as novaclient
21 import nova_force_down
23 LOG = doctor_log.Logger('doctor_inspector').getLogger()
26 class DoctorInspectorSample(object):
28 nova_api_version = '2.11'
31 self.servers = collections.defaultdict(list)
32 self.nova = novaclient.Client(self.nova_api_version,
33 os.environ['OS_USERNAME'],
34 os.environ['OS_PASSWORD'],
35 os.environ['OS_TENANT_NAME'],
36 os.environ['OS_AUTH_URL'],
38 # check nova is available
39 self.nova.servers.list(detailed=False)
40 self.init_servers_list()
42 def init_servers_list(self):
43 opts = {'all_tenants': True}
44 servers=self.nova.servers.list(search_opts=opts)
46 for server in servers:
48 host=server.__dict__.get('OS-EXT-SRV-ATTR:host')
49 self.servers[host].append(server)
50 LOG.debug('get hostname=%s from server=%s' % (host, server))
51 except Exception as e:
52 LOG.error('can not get hostname from server=%s' % server)
54 def disable_compute_host(self, hostname):
55 for server in self.servers[hostname]:
56 self.nova.servers.reset_state(server, 'error')
57 LOG.info('doctor mark vm(%s) error at %s' % (server, time.time()))
59 # NOTE: We use our own client here instead of this novaclient for a
60 # workaround. Once keystone provides v2.1 nova api endpoint
61 # in the service catalog which is configured by OpenStack
62 # installer, we can use this:
64 # self.nova.services.force_down(hostname, 'nova-compute', True)
66 nova_force_down.force_down(hostname)
67 LOG.info('doctor mark host(%s) down at %s' % (hostname, time.time()))
71 inspector = DoctorInspectorSample()
74 @app.route('/events', methods=['POST'])
76 LOG.info('event posted at %s' % time.time())
77 LOG.info('inspector = %s' % inspector)
78 LOG.info('received data = %s' % request.data)
79 d = json.loads(request.data)
81 hostname = event['details']['hostname']
82 event_type = event['type']
83 if event_type == 'compute.host.down':
84 inspector.disable_compute_host(hostname)
89 parser = argparse.ArgumentParser(description='Doctor Sample Inspector')
90 parser.add_argument('port', metavar='PORT', type=int, nargs='?',
91 help='a port for inspector')
92 return parser.parse_args()
97 app.run(port=args.port)
100 if __name__ == '__main__':