add proc count module

This commit is contained in:
Dave Pedu 2015-12-05 16:46:45 -08:00
parent 30f6a51f3b
commit d1b1b09e5e
3 changed files with 86 additions and 1 deletions

View File

@ -17,6 +17,11 @@
"type":"meminfo",
"freq":"30",
"args":{}
},
{
"type":"procs",
"freq":"30",
"args":{}
}
]
}

View File

@ -106,7 +106,7 @@ class Backend:
}
}
mapping["mappings"].update(self.mapping)
self.logger.info("creating index %s with mapping %s" % (indexName, mapping))
self.logger.info("creating index %s with mapping %s" % (indexName, json.dumps(mapping, indent=4)))
self.es.indices.create(index=indexName, ignore=400, body=mapping)# ignore already exists error
self.current_index = indexName

View File

@ -0,0 +1,80 @@
from glob import glob
import re
KTHREADD_PID = 2
PAT_UID = re.compile(r'Uid:\s+(?P<uid>[0-9]+)\s')
PAT_GID = re.compile(r'Gid:\s+(?P<gid>[0-9]+)\s')
def procs():
# Get uid->name mapping
users = {}
with open('/etc/passwd', 'r') as passwd:
while True:
line = passwd.readline()
if not line:
break
uname,x,uid,gid,opts,home,shell = line.split(":")
users[int(uid)]=uname
# Get gid->groupname mapping
groups = {}
with open('/etc/group', 'r') as group:
while True:
line = group.readline()
if not line:
break
gname,x,gid,y = line.split(":")
groups[int(gid)]=gname
num_procs = 0
num_threads = 0
num_kthreads = 0
for f in glob('/proc/[0-9]*/stat'):
try:
with open(f, "r") as statfile:
with open(f+"us", "r") as statusfile:
# Read stat info
stat = statfile.read().strip().split(" ")
# Read uid/gid from status
status = statusfile.read()
proc_uid = PAT_UID.findall(status)
proc_gid = PAT_GID.findall(status)
proc_id = int(stat[0])
proc_parent = int(stat[3])
if proc_parent == KTHREADD_PID:
num_kthreads+=1
else:
num_procs+=1
num_threads += int(stat[19])
except Exception as e:
print(e)
print("Failed to open %s" % f)
return {"procs": num_procs, "threads":num_threads, "kthreads": num_kthreads}
mapping = {
"procs": {
"properties": {
"procs": {
"type": "integer"
},
"threads": {
"type": "integer"
},
"kthreads": {
"type": "integer"
}
}
}
}
if __name__ == '__main__':
stats = procs()
print("%s procs %s kthreads %s threads" % (stats["procs"], stats["kthreads"], stats["threads"]))