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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
|
##############################################################################
# Copyright (c) 2017 NEC 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
##############################################################################
import os
import socket
import getpass
import sys
from doctor_tests.monitor.base import BaseMonitor
class CollectdMonitor(BaseMonitor):
def __init__(self, conf, inspector_url, log):
super(CollectdMonitor, self).__init__(conf, inspector_url, log)
monitor_dir = os.path.split(os.path.realpath(__file__))[0]
self.top_dir = os.path.dirname(monitor_dir)
tmp_sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
tmp_sock.connect(("8.8.8.8", 80))
## control_ip is the IP of primary interface of control node i.e.
## eth0, eno1. It is used by collectd monitor to communicate with
## sample inspector.
## TODO (umar) see if mgmt IP of control is a better option. Also
## primary interface may not be the right option
self.control_ip = tmp_sock.getsockname()[0]
self.compute_user = getpass.getuser()
self.interface_name = os.environ.get('INTERFACE_NAME') or ''
self.inspector_type = os.environ.get('INSPECTOR_TYPE', 'sample')
self.auth_url = os.environ.get('OS_AUTH_URL')
self.username = os.environ.get('OS_USERNAME')
self.password = os.environ.get('OS_PASSWORD')
self.project_name = os.environ.get('OS_PROJECT_NAME')
self.user_domain_name = os.environ.get('OS_USER_DOMAIN_NAME') or 'default'
self.user_domain_id = os.environ.get('OS_USER_DOMAIN_ID')
self.project_domain_name = os.environ.get('OS_PROJECT_DOMAIN_NAME') or 'default'
self.project_domain_id = os.environ.get('OS_PROJECT_DOMAIN_ID')
self.ssh_opts_cpu = '-o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no'
def start(self, host):
self.log.info("Collectd monitor start.........")
self.compute_host = host.name
self.compute_ip = host.ip
f = open("%s/collectd.conf" % self.top_dir, 'w')
collectd_conf_file = """
Hostname %s
FQDNLookup false
Interval 1
MaxReadInterval 2
<LoadPlugin python>
Globals true
</LoadPlugin>
LoadPlugin ovs_events
LoadPlugin logfile
<Plugin logfile>
File \"/var/log/collectd.log\"
Timestamp true
LogLevel \"info\"
</Plugin>
<Plugin python>
ModulePath \"/home/%s\"
LogTraces true
Interactive false
Import \"collectd_plugin\"
<Module \"collectd_plugin\">
control_ip \"%s\"
compute_ip \"%s\"
compute_host \"%s\"
compute_user \"%s\"
inspector_type \"%s\"
os_auth_url \"%s\"
os_username \"%s\"
os_password \"%s\"
os_project_name \"%s\"
os_user_domain_name \"%s\"
os_user_domain_id \"%s\"
os_project_domain_name \"%s\"
os_project_domain_id \"%s\"
</Module>
</Plugin>
<Plugin ovs_events>
Port 6640
Socket \"/var/run/openvswitch/db.sock\"
Interfaces \"@INTERFACE_NAME@\"
SendNotification true
DispatchValues false
</Plugin>
""" % (self.compute_host, self.compute_user, self.control_ip, self.compute_ip, self.compute_host, self.compute_user,
self.inspector_type, self.auth_url, self.username, self.password, self.project_name, self.user_domain_name,
self.user_domain_id, self.project_domain_name, self.project_domain_id)
f.write(collectd_conf_file)
f.close()
os.system(" scp %s %s/collectd.conf %s@%s: " % (self.ssh_opts_cpu, self.top_dir, self.compute_user, self.compute_ip))
self.log.info("after first scp")
## @TODO (umar) Always assuming that the interface is assigned an IP if
## interface name is not provided. See if there is a better approach
os.system(""" ssh %s %s@%s \"if [ -n \"%s\" ]; then
dev=%s
else
dev=\$(sudo ip a | awk '/ %s\//{print \$NF}')
fi
sed -i -e \"s/@INTERFACE_NAME@/\$dev/\" collectd.conf
collectd_conf=/opt/collectd/etc/collectd.conf
if [ -e \$collectd_conf ]; then
sudo cp \$collectd_conf \${collectd_conf}-doctor-saved
else
sudo touch \${collectd_conf}-doctor-created
fi
sudo mv collectd.conf /opt/collectd/etc/collectd.conf\" """ % (self.ssh_opts_cpu, self.compute_user, self.compute_ip, self.interface_name, self.interface_name, self.compute_ip))
self.log.info("after first ssh")
os.system(" scp %s %s/monitor/collectd_plugin.py %s@%s:collectd_plugin.py " % (self.ssh_opts_cpu, self.top_dir, self.compute_user, self.compute_ip))
self.log.info("after sec scp")
os.system(" ssh %s %s@%s \"sudo pkill collectd; sudo /opt/collectd/sbin/collectd\" " % (self.ssh_opts_cpu, self.compute_user, self.compute_ip))
self.log.info("after sec ssh")
def stop(self):
os.system(" ssh %s %s@%s \"sudo pkill collectd\" " % (self.ssh_opts_cpu, self.compute_user, self.compute_ip))
def cleanup(self):
os.system(""" ssh %s %s@%s \"
collectd_conf=/opt/collectd/etc/collectd.conf
if [ -e \"\${collectd_conf}-doctor-created\" ]; then
sudo rm \"\${collectd_conf}-doctor-created\"
sudo rm \$collectd_conf
elif [ -e \"\${collectd_conf}-doctor-saved\" ]; then
sudo cp -f \"\${collectd_conf}-doctor-saved\" \$collectd_conf
sudo rm \"\${collectd_conf}-doctor-saved\"
fi\" """ % (self.ssh_opts_cpu, self.compute_user, self.compute_ip))
os.remove("%s/collectd.conf" % self.top_dir)
|
