Add a python program to record memory usage during installation.

This program only runs during installation if the "debug" command line option
is provided.  It writes out a /tmp/memory.dat file which can be copied off the
system and processed with the provided gnuplot script to display a graph.

3 files changed, 77 insertions, 1 deletions

diff --git a/scripts/Makefile.am b/scripts/Makefile.am
index 329a2fc6f..7b843a066 100644
--- a/scripts/Makefile.am
+++ b/scripts/Makefile.am
@@ -25,9 +25,12 @@ dist_scripts_DATA    = mk-images.* pyrc.py
 dist_noinst_SCRIPTS  = getlangnames.py upd-bootimage upd-initrd upd-kernel \
                        makeupdates
 
-dist_bin_SCRIPTS = analog anaconda-cleanup
+dist_bin_SCRIPTS = analog anaconda-cleanup instperf
 
 stage2scriptsdir = $(datadir)/$(PACKAGE_NAME)
 dist_stage2scripts_SCRIPTS = restart-anaconda
 
+miscdir = $(datadir)/$(PACKAGE_NAME)
+dist_misc_DATA = instperf.p
+
 MAINTAINERCLEANFILES = Makefile.in
diff --git a/scripts/instperf b/scripts/instperf
new file mode 100755
index 000000000..f0f53cc9e
--- /dev/null
+++ b/scripts/instperf
@@ -0,0 +1,58 @@
+#!/usr/bin/python
+
+import logging
+from string import split
+from subprocess import Popen, PIPE
+import time
+
+# grab the top five memory consuming processes, returning them in a string of
+# of the format:
+#
+# name1:kB1,name2:kB2,...,name5:kB5
+def topProcesses():
+    output = Popen(["ps", "-eo", "comm,rss", "--sort", "-rss", "--no-headers"], stdout=PIPE).communicate()[0]
+    top5 = output.split("\n")[:5]
+    return ",".join(map(lambda (a,b): a+":"+b, map(split, top5)))
+
+def logit():
+    buffers = 0
+    cached = 0
+    memTotal = 0
+    memFree = 0
+    swapTotal = 0
+    swapFree = 0
+
+    global mem_logger
+
+    with open("/proc/meminfo") as f:
+        for line in f.readlines():
+            if line.startswith("Buffers:"):
+                buffers = line.split()[1]
+            elif line.startswith("Cached:"):
+                cached = line.split()[1]
+            elif line.startswith("MemTotal:"):
+                memTotal = line.split()[1]
+            elif line.startswith("MemFree:"):
+                memFree = line.split()[1]
+            elif line.startswith("SwapTotal:"):
+                swapTotal = line.split()[1]
+            elif line.startswith("SwapFree:"):
+                swapFree = line.split()[1]
+
+    d = {"memUsed": int(memTotal)-int(memFree)-int(cached)-int(buffers),
+         "swapUsed": int(swapTotal)-int(swapFree),
+         "procs": topProcesses()}
+
+    mem_logger.debug("%(memUsed)d %(swapUsed)d %(procs)s" % d)
+
+mem_logger = logging.getLogger("memLog")
+mem_logger.setLevel(logging.DEBUG)
+
+handler = logging.FileHandler("/tmp/memory.dat")
+handler.setLevel(logging.DEBUG)
+handler.setFormatter(logging.Formatter("%(asctime)s %(message)s", "%H:%M:%S"))
+mem_logger.addHandler(handler)
+
+while True:
+    logit()
+    time.sleep(1)
diff --git a/scripts/instperf.p b/scripts/instperf.p
new file mode 100644
index 000000000..078a5a140
--- /dev/null
+++ b/scripts/instperf.p
@@ -0,0 +1,15 @@
+# This script processes a memory.dat file as generated by instperf during
+# installation and writes out a graph to memusage.png.
+set terminal png size 1024,768
+set output "memusage.png"
+set title "anaconda Memory Usage"
+set xlabel "Time"
+set xtics rotate
+set xdata time
+set timefmt "%H:%M:%S"
+set ylabel "Memory Used (in KB)\n(Total-Free-Buffers-Cached)"
+set format y "%.0f"
+set grid
+
+plot "memory.dat" using 1:2 title "Memory" with lines, \
+     "memory.dat" using 1:3 title "Swap" with lines