commit efe7adfb0e15324ac241957e3addef2cfe3173d1
author Allen Li <ayatane@chromium.org> Thu Oct 27 11:36:04 2016 -0700
committer chrome-bot <chrome-bot@chromium.org> Mon Oct 31 22:40:10 2016 -0700
tree 6a0b0afc06c0cbe02eaa1720f824d749cbc63ccb
parent 340d0d268ff9f320e3b40710c692bb54efd99fcb

[sysmon] Add autoserv process count

BUG=chromium:659308
TEST=Run sysmon on test drone

Change-Id: I5228d90f40ba3785e7b1f8049aa86f39581d7502
Reviewed-on: https://chromium-review.googlesource.com/404450
Commit-Ready: Allen Li <ayatane@chromium.org>
Tested-by: Allen Li <ayatane@chromium.org>
Reviewed-by: Aviv Keshet <akeshet@chromium.org>

scripts/sysmon/system_metrics.py

diff --git a/scripts/sysmon/system_metrics.py b/scripts/sysmon/system_metrics.py
index b56c171..4caf6da 100644
--- a/scripts/sysmon/system_metrics.py
+++ b/scripts/sysmon/system_metrics.py
@@ -105,6 +105,9 @@
 _proc_count_metric = ts_mon.GaugeMetric(
     'dev/proc/count',
     description='Number of processes currently running.')
+_autoserv_proc_count_metric = ts_mon.GaugeMetric(
+    'dev/proc/autoserv_count',
+    description='Number of autoserv processes currently running.')
 _load_average_metric = ts_mon.FloatMetric(
     'dev/proc/load_average',
     description='Number of processes currently '
@@ -301,8 +304,20 @@
 
 
 def get_proc_info():
-  procs = psutil.pids()
-  _proc_count_metric.set(len(procs))
+  autoserv_count = 0
+  total = 0
+  for proc in psutil.process_iter():
+    if _is_autoserv_proc(proc):
+      autoserv_count += 1
+    total += 1
+  _autoserv_proc_count_metric.set(autoserv_count)
+  _proc_count_metric.set(total)
+
+
+def _is_autoserv_proc(proc):
+  return (
+      proc.name == 'python'
+      and '/usr/local/autotest/server/autoserv' in proc.cmdline)
 
 
 def get_load_avg():