summaryrefslogtreecommitdiffstats
path: root/doctor_tests/inspector/congress.py
blob: 7f918fb28dfec147625f0c600d6eb195c14a542e (plain)
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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
##############################################################################
# Copyright (c) 2017 ZTE Corporation 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 doctor_tests.identity_auth import get_identity_auth
from doctor_tests.identity_auth import get_session
from doctor_tests.os_clients import congress_client

from doctor_tests.inspector.base import BaseInspector


class CongressInspector(BaseInspector):
    nova_api_min_version = '2.11'
    doctor_driver = 'doctor'
    doctor_datasource = 'doctor'
    policy = 'classification'
    rules = {
        'host_down':
            'host_down(host) :- doctor:events(hostname=host, type="compute.host.down", status="down")',       # noqa
        'active_instance_in_host':
            'active_instance_in_host(vmid, host) :- nova:servers(id=vmid, host_name=host, status="ACTIVE")',  # noqa
        'host_force_down':
            'execute[nova:services.force_down(host, "nova-compute", "True")] :- host_down(host)',             # noqa
        'error_vm_states':
            'execute[nova:servers.reset_state(vmid, "error")] :- host_down(host), active_instance_in_host(vmid, host)'    # noqa
    }

    def __init__(self, conf, log):
        super(CongressInspector, self).__init__(conf, log)
        self.is_create_doctor_datasource = False
        self.doctor_datasource_id = None
        self.auth = get_identity_auth()
        self.congress = congress_client(get_session(auth=self.auth))
        self._init_driver_and_ds()
        self.inspector_url = self.get_inspector_url()

    def _init_driver_and_ds(self):
        datasources = \
            {ds['name']: ds for ds in
             self.congress.list_datasources()['results']}

        # check nova_api version
        nova_api_version = datasources['nova']['config'].get('api_version')
        if nova_api_version and nova_api_version < self.nova_api_min_version:
            raise Exception('Congress Nova datasource API '
                            'version < nova_api_min_version(%s)'
                            % self.nova_api_min_version)

        # check whether doctor driver exist
        drivers = \
            {driver['id']: driver for driver in
             self.congress.list_drivers()['results']}
        if self.doctor_driver not in drivers:
            raise Exception('Do not support doctor driver in congress')

        # create doctor datasource if it's not exist
        if self.doctor_datasource not in datasources:
            response = self.congress.create_datasource(
                body={'driver': self.doctor_driver,
                      'name': self.doctor_datasource})
            self.doctor_datasource_id = response['id']
            self.is_create_doctor_datasource = True

        self.policy_rules = \
            {rule['name']: rule for rule in
             self.congress.list_policy_rules(self.policy)['results']}

    def get_inspector_url(self):
        ds = self.congress.list_datasources()['results']
        doctor_ds = next((item for item in ds if item['driver'] == 'doctor'),
                         None)
        congress_endpoint = \
            self.congress.httpclient.get_endpoint(auth=self.auth)
        return ('%s/v1/data-sources/%s/tables/events/rows' %
                (congress_endpoint, doctor_ds['id']))

    def start(self):
        self.log.info('congress inspector start......')

        for rule_name, rule in self.rules.items():
            self._add_rule(rule_name, rule)

    def stop(self):
        self.log.info('congress inspector stop......')

        for rule_name in self.rules.keys():
            self._del_rule(rule_name)

        if self.is_create_doctor_datasource:
            self.congress.delete_datasource(self.doctor_datasource_id)

    def _add_rule(self, rule_name, rule):
        if rule_name not in self.policy_rules:
            self.congress.create_policy_rule(self.policy,
                                             body={'name': rule_name,
                                                   'rule': rule})

    def _del_rule(self, rule_name):
        if rule_name in self.policy_rules:
            rule_id = self.policy_rules[rule_name]['id']
            self.congress.delete_policy_rule(self.policy, rule_id)