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
|
import os
from utils.logging import log
from utils.misc import run_cmd
COLLECTD_CONFIG = '/tmp/.collectd.conf'
COLLECTD_PIDFILE = '/tmp/.collectd.pid'
class CollectdCollector(object):
"""
Collect basic system and database statistics using collectd.
"""
def __init__(self, outdir, dbname, bin_path):
self._bin_path = bin_path
# Hard code all possible places a packager might install collectd.
self._env = os.environ
self._env['PATH'] = ':'.join(['/usr/sbin/', '/sbin/', self._env['PATH']])
# Assume collectd.conf.in file to be in the same directory as this
# file.
cwd = os.path.dirname(os.path.realpath(__file__))
modules = (
'LoadPlugin aggregation\n'
'LoadPlugin contextswitch\n'
'LoadPlugin cpu\n'
'LoadPlugin csv\n'
'LoadPlugin disk\n'
'LoadPlugin interface\n'
'LoadPlugin memory\n'
'LoadPlugin postgresql\n'
'LoadPlugin processes\n'
'LoadPlugin swap\n'
)
system = os.popen("uname").readlines()[0].split()[0]
if system == 'Linux':
modules += (
'LoadPlugin ipc\n'
'LoadPlugin vmem\n'
)
outdir = '%s/stats' % outdir
config_template = open('%s/collectd.conf.in' % cwd, 'r')
config = open(COLLECTD_CONFIG, 'w')
config.write(config_template.read() % {'database': dbname,
'datadir': outdir,
'modules': modules,
'pguser': self._env['USER']})
config.close()
config_template.close()
# TODO: Use collectd to test config act exit appropriately.
def start(self):
log("starting collectd")
cmd = 'collectd -C %s -P %s' % (COLLECTD_CONFIG, COLLECTD_PIDFILE)
run_cmd(cmd.split(' '), env=self._env)
def stop(self):
log("stopping collectd")
pidfile = open(COLLECTD_PIDFILE, 'r')
pid = pidfile.read().strip()
run_cmd(['kill', pid])
def result(self):
return {}
def run_collector(in_queue, out_queue, dbname, bin_path, outdir, interval=1.0):
pass
|
