defunkt · October 12, 2009 18:03 · Oct 12, 2009 · Oct 12, 2009
diff --git a/gistfile1.rb b/gistfile1.rb
@@ -7,6 +7,11 @@
   loop do
     begin
       # unicorn workers
+      #
+      # ps output line format:
+      # 31580 275444 unicorn_rails worker[15] -c /data/github/current/config/unicorn.rb -E production -D
+      # pid ram command
+
       lines = `ps -e -www -o pid,rss,command | grep '[u]nicorn_rails worker'`.split("\n")
       lines.each do |line|
         parts = line.split(' ')

diff --git a/gistfile1.rb b/gistfile1.rb
@@ -0,0 +1,25 @@
+# This will ride alongside god and kill any rogue memory-greedy
+# processes. Their sacrifice is for the greater good.
+
+unicorn_worker_memory_limit = 300_000
+
+Thread.new do
+  loop do
+    begin
+      # unicorn workers
+      lines = `ps -e -www -o pid,rss,command | grep '[u]nicorn_rails worker'`.split("\n")
+      lines.each do |line|
+        parts = line.split(' ')
+        if parts[1].to_i > unicorn_worker_memory_limit
+          # tell the worker to die after it finishes serving its request
+          ::Process.kill('QUIT', parts[0].to_i)
+        end
+      end
+    rescue Object
+      # don't die ever once we've tested this
+      nil
+    end
+
+    sleep 30
+  end
+end