From b9d5cfd40bfd794a8f66ed443f13259934b43cbc Mon Sep 17 00:00:00 2001
From: Siraaj Khandkar <siraaj@khandkar.net>
Date: Wed, 22 Aug 2018 10:34:39 -0400
Subject: [PATCH] Monitor process states totals

---
 bin/khatus              |  2 ++
 bin/khatus_bar          | 18 +++++++++++++++++-
 bin/khatus_parse_ps     | 27 +++++++++++++++++++++++++++
 bin/khatus_sensor_procs |  7 +++++++
 sanity_check            |  2 +-
 5 files changed, 54 insertions(+), 2 deletions(-)
 create mode 100755 bin/khatus_parse_ps
 create mode 100755 bin/khatus_sensor_procs

diff --git a/bin/khatus b/bin/khatus
index 95bd517..747fd32 100755
--- a/bin/khatus
+++ b/bin/khatus
@@ -99,6 +99,7 @@ main() {
         ["--fan_path"]='/proc/acpi/ibm/fan'
         ["--pulseaudio_sink"]='0'
         ["--interval_datetime"]=1
+        ["--interval_procs"]=1
         ["--interval_brightness"]=1
         ["--interval_weather"]=$(( 30 * 60))  # 30 minutes
         ["--interval_mpd"]=1
@@ -189,6 +190,7 @@ main() {
     fork_watcher                                           "$pipe" "$bin" "khatus_sensor_energy $bin"
     fork_watcher                                           "$pipe" "$bin" "khatus_sensor_devices $bin"
     fork_poller "${opts['--interval_datetime']}"   "$perf" "$pipe" "$bin" khatus_sensor_datetime
+    fork_poller "${opts['--interval_procs']}"      "$perf" "$pipe" "$bin" "khatus_sensor_procs $bin"
     fork_poller "${opts['--interval_brightness']}" "$perf" "$pipe" "$bin" "$cmd_sens_screen_brightness"
     fork_poller "${opts['--interval_weather']}"    "$perf" "$pipe" "$bin" "$cmd_sens_weather"
     fork_poller "${opts['--interval_mpd']}"        "$perf" "$pipe" "$bin" "$cmd_sens_mpd"
diff --git a/bin/khatus_bar b/bin/khatus_bar
index 2c7ee32..6fa05ba 100755
--- a/bin/khatus_bar
+++ b/bin/khatus_bar
@@ -102,11 +102,15 @@ function cache_gc(    src_and_key, parts, src, key, unused_for) {
         split(src_and_key, parts, SUBSEP)
         src = parts[1]
         key = parts[2]
+        val = _cache[src, key]
         unused_for = cache_get_time() - _cache_atime[src, key]
         if (unused_for > GC_Interval) {
             print_msg_info(\
                 "cache_gc",
-                sprintf("Deleting unused data SRC=%s KEY=%s", src, key) \
+                sprintf(\
+                    "Deleting unused data SRC=%s KEY=%s VAL=%s",
+                    src, key, val\
+                ) \
             )
             delete _cache[src, key]
         }
@@ -141,6 +145,7 @@ function make_status_bar(    position, bar, sep, i, j) {
     position[++i] = ""
     position[++i] = make_status_energy()
     position[++i] = make_status_mem()
+    position[++i] = make_status_procs()
     position[++i] = make_status_cpu()
     position[++i] = make_status_disk()
     position[++i] = make_status_net()
@@ -191,6 +196,17 @@ function make_status_mem(    total, used, percent, status) {
     return sprintf("M=%s", status)
 }
 
+function make_status_procs() {
+    src = "khatus_sensor_procs"
+    all = cache_get_fmt_def(src, "total_procs"            , 15, "%d")
+    r   = cache_get_fmt_def(src, "total_per_state" Kfs "R", 15, "%d", "0")
+    d   = cache_get_fmt_def(src, "total_per_state" Kfs "D", 15, "%d", "0")
+    t   = cache_get_fmt_def(src, "total_per_state" Kfs "T", 15, "%d", "0")
+    i   = cache_get_fmt_def(src, "total_per_state" Kfs "I", 15, "%d", "0")
+    z   = cache_get_fmt_def(src, "total_per_state" Kfs "Z", 15, "%d", "0")
+    return sprintf("P=[%s %sr %sd %st %si %sz]", all, r, d, t, i, z)
+}
+
 function make_status_cpu(    l, t, f) {
     l_src = "khatus_sensor_loadavg"
     t_src = "khatus_sensor_temperature"
diff --git a/bin/khatus_parse_ps b/bin/khatus_parse_ps
new file mode 100755
index 0000000..3b8417e
--- /dev/null
+++ b/bin/khatus_parse_ps
@@ -0,0 +1,27 @@
+#! /usr/bin/awk -f
+
+BEGIN {
+    OFS = "|"
+    Kfs = ":"
+}
+
+{
+    pid       = $1
+    state     = $2
+    rss       = $3
+    command_i = index($0, $4)
+    command   = substr($0, command_i, length($0) - (command_i - 1))
+
+    Pids[pid] = 1
+    Total_Per_State[state]++
+    print("state"   Kfs pid, state)
+    print("rss"     Kfs pid, rss)
+    print("command" Kfs pid, command)
+}
+
+END {
+    print("total_procs", length(Pids))
+    for (state in Total_Per_State) {
+        print("total_per_state" Kfs state, Total_Per_State[state])
+    }
+}
diff --git a/bin/khatus_sensor_procs b/bin/khatus_sensor_procs
new file mode 100755
index 0000000..7083d81
--- /dev/null
+++ b/bin/khatus_sensor_procs
@@ -0,0 +1,7 @@
+#! /bin/sh
+
+set -e
+
+dir_bin="$1"
+
+ps -eo pid,state,rss,command ww --no-headers | "$dir_bin"/khatus_parse_ps
diff --git a/sanity_check b/sanity_check
index c10dcaf..fdcb394 100755
--- a/sanity_check
+++ b/sanity_check
@@ -8,7 +8,7 @@
         -v Opt_Mpd_Song_Max_Chars=10 \
         -v Opt_Net_Interfaces_To_Show=wlp3s0 \
         -v Opt_Pulseaudio_Sink=0 \
-        -v GC_Interval=5 \
+        -v GC_Interval=10 \
     ) \
     >(stdbuf -o L ./bin/khatus_monitor_energy) \
     >(stdbuf -o L ./bin/khatus_monitor_errors) \
-- 
2.20.1