From c2bdb3edd65b0fbbe27b927d70a0fe3969fc7e71 Mon Sep 17 00:00:00 2001
From: watsonian <watsonian@gmail.com>
Date: Mon, 3 Sep 2012 21:43:43 -0700
Subject: [PATCH] Make log directory and thresholds configurable.

---
 lib/mem_watcher.rb | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/lib/mem_watcher.rb b/lib/mem_watcher.rb
index 6108990..c0d6ffb 100644
--- a/lib/mem_watcher.rb
+++ b/lib/mem_watcher.rb
@@ -8,12 +8,13 @@
 class MemWatcher < Sinatra::Base
   set :views, File.join(File.dirname(settings.root), "/views")
   set :public_folder, File.join(File.dirname(settings.root), "/public")
+  set :log_dir, ENV['MEMWATCHER_LOG_DIR'] || File.join(File.dirname(settings.root), "/log/procs")
 
-  @@memory_threshold = 50000
+  @@memory_threshold = (ENV['MEMWATCHER_MEMORY_THRESHOLD'] || 50000).to_i
   @@max_memused = @@memory_threshold
   @@num_checks = 0
-  @@max_checks = 6
-  @@logs_to_keep = 25
+  @@max_checks = (ENV['MEMWATCHER_MAX_CHECKS'] || 6).to_i
+  @@logs_to_keep = (ENV['MEMWATCHER_LOGS_TO_KEEP'] || 25).to_i
 
   #####
   # Actions
@@ -72,7 +73,7 @@ def update_max_memused!(cur_meminfo)
     end
 
     def save_process_list!(cur_meminfo)
-      FileUtils.mkdir_p("./log/procs")
+      FileUtils.mkdir_p(settings.log_dir)
       system("ps auxf > log/procs/`date \"+%Y%m%d_%H%M%S-#{@@max_memused}-#{cur_meminfo.memused}\"`.snapshot.out")
     end