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
|
### Author: <gianfranco@mongodb.com>
global mongodb_user
mongodb_user = os.getenv('DSTAT_MONGODB_USER') or os.getenv('USER')
global mongodb_pwd
mongodb_pwd = os.getenv('DSTAT_MONGODB_PWD')
global mongodb_host
mongodb_host = os.getenv('DSTAT_MONGODB_HOST') or '127.0.0.1:27017'
class dstat_plugin(dstat):
"""
Plugin for MongoDB.
"""
def __init__(self):
global pymongo
import pymongo
try:
self.m = pymongo.MongoClient(mongodb_host)
if mongodb_pwd:
self.m.admin.authenticate(mongodb_user, mongodb_pwd)
self.db = self.m.admin
except Exception as e:
raise Exception('Cannot interface with MongoDB server: %s' % e)
line = self.db.command("serverStatus")
if 'storageEngine' in line:
self.storageEngine = line.get('storageEngine').get('name')
else:
self.storageEngine = 'mmapv1'
self.name = 'mongodb mem'
self.nick = ('res', 'virt')
self.vars = ('mem.resident', 'mem.virtual')
self.type = 'd'
self.width = 5
self.scale = 2
self.lastVal = {}
if self.storageEngine == 'mmapv1':
self.nick = self.nick + ('map', 'mapj', 'flt')
self.vars = self.vars + ('mem.mapped', 'mem.mappedWithJournal', 'extra_info.page_faults')
def extract(self):
status = self.db.command("serverStatus")
for name in self.vars:
if name in ('extra_info.page_faults'):
if not name in self.lastVal:
self.lastVal[name] = int(self.getDoc(status, name))
self.val[name] = (int(self.getDoc(status, name)) - self.lastVal[name])
self.lastVal[name] = self.getDoc(status, name)
else:
self.val[name] = (int(self.getDoc(status, name)))
def getDoc(self, dic, doc):
par = doc.split('.')
sdic = dic
for p in par:
sdic = sdic.get(p)
return sdic
|
