weasel-the-memory-watcher 1.0.4 → 1.1.0

data/VERSION

@@ -1 +1 @@
-1.0.4
+1.1.0

data/bin/weasel

@@ -13,6 +13,9 @@ OptionParser.new do |opts|
                           eval(v)
                         }
   end
+  opts.on("--callback_message string",String, "message to write when callback fired") do |v|
+    options[:callback_message] = v
+  end
 
   opts.on("--num_over_marks N",String, "how many times the proces memory must go over to be a candidate") do |v|
     options[:num_over_marks] = v.to_i

data/lib/memory_watch.rb

@@ -1,14 +1,20 @@
 
 class MemoryWatch
-  attr_accessor :watch_string,:high_water_pids,:callback,:high_water_mb,:delay,:num_cycles,:num_over_marks,:pids
+  attr_accessor :watch_string,:high_water_pids,:callback,:high_water_mb,
+                :delay,:num_cycles,:num_over_marks,:pids,:callback_message
+
   def initialize(options)
-    self.callback        = options[:callback]       || lambda {|pid| p pid}
-    self.watch_string    = (options[:watch]         || "this poem is a pomme").strip
-    self.delay           = options[:delay]          || 60
-    self.num_cycles      = options[:num_cycles]     || 3
-    self.num_over_marks  = options[:num_over_marks] || 2
-    self.high_water_mb   = options[:high_water_mb]  || 700
-    self.high_water_pids = {}
+    self.callback         = options[:callback]             || 
+                             lambda {|pid| p pid}
+    self.callback_message = (options[:callback_message]    || 
+                             "Callback Triggered").strip
+    self.watch_string     = (options[:watch]               || 
+                             "this poem is a pomme").strip
+    self.delay            = options[:delay]                || 60
+    self.num_cycles       = options[:num_cycles]           || 3
+    self.num_over_marks   = options[:num_over_marks]       || 2
+    self.high_water_mb    = options[:high_water_mb]        || 700
+    self.high_water_pids  = {}
   end
   public
   def cycle
@@ -23,18 +29,19 @@ class MemoryWatch
   def _trigger_callback
     self.high_water_pids.each do |pid,over_marks|
       if over_marks.size >= self.num_over_marks
+        puts "[#{Time.now}] #{pid} #{self.callback_message}"
         self.callback.call(pid)
       end
     end
   end
   def _run
-    p cmd = "ps x |grep '#{self.watch_string}' |grep -v grep | awk '{print $1}'"
+    cmd = "ps x |grep '#{self.watch_string}' |grep -v grep | awk '{print $1}'"
     self.pids = %x{#{cmd}}.split()
     self.pids
   end
   def _check_pids
     _run.each { |pid|
-      p memory_usage = %x{ps -o rss= -p #{pid}}.to_i # KB
+      memory_usage = %x{ps -o rss= -p #{pid}}.to_i # KB
       if memory_usage > self.high_water_mb * 1024 # 750MB
         puts "WARNING - Process #{pid} hit high water mark"
         self.high_water_pids[pid] ||= []

data/test/test_memory_watch.rb

@@ -1,7 +1,7 @@
 require 'helper'
 class TestMemoryWatch < Test::Unit::TestCase
   def test_watches_self
-    cmd = %q{ruby ./lib/memory_watch.rb --delay 999999 --num_cycles 9900000099}
+    cmd = %q{ruby ./bin/weasel --delay 999999 --num_cycles 9900000099}
     pid = fork { %x{#{cmd}} }
     Process.detach(pid)
     wm = MemoryWatch.new(:watch => cmd , :high_water_mb => 0.01,:delay => 0.1, :num_cycles => 1)
@@ -22,7 +22,8 @@ class TestMemoryWatch < Test::Unit::TestCase
                          :num_over_marks => 0,
                          :callback => lambda {|pid|
                             test_magiggy = pid
-                         })
+                         },
+                         :callback_message => 'Message')
     wm.cycle
     assert { 
       wm.high_water_pids.include? test_magiggy

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: weasel-the-memory-watcher
 version: !ruby/object:Gem::Version 
-  hash: 31
+  hash: 19
   prerelease: false
   segments: 
   - 1
+  - 1
   - 0
-  - 4
-  version: 1.0.4
+  version: 1.1.0
 platform: ruby
 authors: 
 - Curtis Schofield
@@ -16,7 +16,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-11-19 00:00:00 -08:00
+date: 2010-11-22 00:00:00 -08:00
 default_executable: weasel
 dependencies: 
 - !ruby/object:Gem::Dependency