Process metrics are added to report total CPU and Memory usage of a specific process.a

Author: Caglar Anar

cloudwatchmon/cli/put_instance_stats.py

@@ -29,6 +29,7 @@
 import re
 import sys
 import time
+import subprocess
 
 CLIENT_NAME = 'CloudWatch-PutInstanceData'
 FileCache.CLIENT_NAME = CLIENT_NAME
@@ -302,11 +303,19 @@ def config_parser():
                             type=to_lower,
                             choices=size_units,
                             help='Specifies units for disk space metrics.')
-     disk_group.add_argument('--disk-inode-util',
+    disk_group.add_argument('--disk-inode-util',
                             action='store_true',
                             help='Reports disk inode utilization in percentages.')
 
 
+    process_group = parser.add_argument_group('process metrics')
+    process_group.add_argument('--process-name',
+                               metavar='PROCNAME',
+                               default='init',
+                               type=to_lower,
+                               help='Report CPU and Memory metrics of comma-sperated processes.')
+
+
     exclusive_group = parser.add_mutually_exclusive_group()
     exclusive_group.add_argument('--from-cron',
                                  action='store_true',
@@ -422,6 +431,20 @@ def add_disk_metrics(args, metrics):
                                disk.inode_util, disk.mount, disk.file_system)
 
 
+def add_process_metrics(args, metrics):
+    process_names = args.process_name.split(',')
+    for process_name in process_names:
+        processes = subprocess.Popen(["ps", "axco", "command,pcpu,pmem"], stdout=subprocess.PIPE)
+        total_cpu = 0.0
+        total_mem = 0.0
+        for line in processes.stdout:
+            if re.search(process_name, line):
+                out = line.split()
+                total_cpu += float(out[1])
+                total_mem += float(out[2])
+        metrics.add_metric(process_name+'-CpuUtilization', 'Percent', total_cpu)
+        metrics.add_metric(process_name+'-MemoryUtilization', 'Percent', total_mem)
+
 
 def add_static_file_metrics(args, metrics):
     with open(args.from_file[0]) as f:
@@ -457,6 +480,7 @@ def validate_args(args):
         args.swap_util or args.swap_used
     report_disk_data = args.disk_path is not None
     report_loadavg_data = args.loadavg or args.loadavg_percpu
+    report_process_data = args.process_name is not None
 
     if report_disk_data:
         if not args.disk_space_util and not args.disk_space_used and \
@@ -478,7 +502,7 @@ def validate_args(args):
         raise ValueError('No metrics specified for collection and '
                          'submission to CloudWatch.')
 
-    return report_disk_data, report_mem_data, report_loadavg_data
+    return report_disk_data, report_mem_data, report_loadavg_data, report_process_data
 
 
 def main():
@@ -496,7 +520,7 @@ def main():
         return 0
 
     try:
-        report_disk_data, report_mem_data, report_loadavg_data = validate_args(args)
+        report_disk_data, report_mem_data, report_loadavg_data, report_process_data = validate_args(args)
 
         # avoid a storm of calls at the beginning of a minute
         if args.from_cron:
@@ -541,6 +565,9 @@ def main():
         if report_disk_data:
             add_disk_metrics(args, metrics)
 
+        if report_process_data:
+            add_process_metrics(args, metrics)
+
         if args.verbose:
             print 'Request:\n' + str(metrics)