initial commit

lib/meminfo.rb

@@ -0,0 +1,23 @@
+require 'objspace'
+require 'meminfo/ext'
+
+# MemInfo gives a handful of methods allowing inspection of process memory use.
+module MemInfo
+  module_function
+
+  class ProcessNotFound < StandardError; end
+
+  def rss(pid = Process.pid)
+    Integer(ps_rss(pid)) * 1024
+  rescue TypeError
+    raise ProcessNotFound, pid
+  end
+
+  def memsize
+    ObjectSpace.each_object.reduce(0) { |a, e| a + ObjectSpace.memsize_of(e) }
+  end
+
+  private def ps_rss(pid)
+    `ps ax -o rss,pid`.each_line.grep(/^\s*(\d+)\s+#{pid}$/) { $1 }.first
+  end
+end