Merge "update the docs of testing"
[doctor.git] / tests / inspector.py
1 ##############################################################################
2 # Copyright (c) 2016 NEC Corporation and others.
3 #
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 ##############################################################################
9
10 import argparse
11 import collections
12 from flask import Flask
13 from flask import request
14 import json
15 import logger as doctor_log
16 import threading
17 import time
18
19 from keystoneauth1 import session
20 import novaclient.client as novaclient
21
22 import doctor_tests.identity_auth
23
24 LOG = doctor_log.Logger('doctor_inspector').getLogger()
25
26
27 class ThreadedResetState(threading.Thread):
28
29     def __init__(self, nova, state, server):
30         threading.Thread.__init__(self)
31         self.nova = nova
32         self.state = state
33         self.server = server
34
35     def run(self):
36         self.nova.servers.reset_state(self.server, self.state)
37         LOG.info('doctor mark vm(%s) error at %s' % (self.server, time.time()))
38
39
40 class DoctorInspectorSample(object):
41
42     NOVA_API_VERSION = '2.34'
43     NUMBER_OF_CLIENTS = 50
44     # TODO(tojuvone): This could be enhanced in future with dynamic
45     # reuse of self.novaclients when all threads in use and
46     # self.NUMBER_OF_CLIENTS based on amount of cores or overriden by input
47     # argument
48
49     def __init__(self):
50         self.servers = collections.defaultdict(list)
51         self.novaclients = list()
52         auth=identity_auth.get_identity_auth()
53         sess=session.Session(auth=auth)
54         # Pool of novaclients for redundant usage
55         for i in range(self.NUMBER_OF_CLIENTS):
56             self.novaclients.append(
57                 novaclient.Client(self.NOVA_API_VERSION, session=sess))
58         # Normally we use this client for non redundant API calls
59         self.nova=self.novaclients[0]
60         self.nova.servers.list(detailed=False)
61         self.init_servers_list()
62
63     def init_servers_list(self):
64         opts = {'all_tenants': True}
65         servers=self.nova.servers.list(search_opts=opts)
66         self.servers.clear()
67         for server in servers:
68             try:
69                 host=server.__dict__.get('OS-EXT-SRV-ATTR:host')
70                 self.servers[host].append(server)
71                 LOG.debug('get hostname=%s from server=%s' % (host, server))
72             except Exception as e:
73                 LOG.error('can not get hostname from server=%s' % server)
74
75     def disable_compute_host(self, hostname):
76         threads = []
77         if len(self.servers[hostname]) > self.NUMBER_OF_CLIENTS:
78             # TODO(tojuvone): This could be enhanced in future with dynamic
79             # reuse of self.novaclients when all threads in use
80             LOG.error('%d servers in %s. Can handle only %d'%(
81                       self.servers[hostname], hostname, self.NUMBER_OF_CLIENTS))
82         for nova, server in zip(self.novaclients, self.servers[hostname]):
83             t = ThreadedResetState(nova, "error", server)
84             t.start()
85             threads.append(t)
86         for t in threads:
87             t.join()
88         self.nova.services.force_down(hostname, 'nova-compute', True)
89         LOG.info('doctor mark host(%s) down at %s' % (hostname, time.time()))
90
91
92 app = Flask(__name__)
93 inspector = DoctorInspectorSample()
94
95
96 @app.route('/events', methods=['POST'])
97 def event_posted():
98     LOG.info('event posted at %s' % time.time())
99     LOG.info('inspector = %s' % inspector)
100     LOG.info('received data = %s' % request.data)
101     d = json.loads(request.data)
102     for event in d:
103         hostname = event['details']['hostname']
104         event_type = event['type']
105         if event_type == 'compute.host.down':
106             inspector.disable_compute_host(hostname)
107     return "OK"
108
109
110 def get_args():
111     parser = argparse.ArgumentParser(description='Doctor Sample Inspector')
112     parser.add_argument('port', metavar='PORT', type=int, nargs='?',
113                         help='a port for inspector')
114     return parser.parse_args()
115
116
117 def main():
118     args = get_args()
119     app.run(host='0.0.0.0', port=args.port)
120
121 if __name__ == '__main__':
122     main()