procmon 0.0.3 → 0.0.4

data/README.md

@@ -8,45 +8,53 @@ It's hosted on [rubygems.org][rubygems].
 
     sudo gem install procmon
 
-## Example of how to use Procmon
-
-```
-require 'procmon'
-
-notifier = Proc.new do
-  puts "I NEED TO SEND OUT AN EMAIL"
-end
-
-notifier2 = Proc.new do
-  puts "I NEED TO SEND OUT AN SMS"
-end
-
-# action can be a Proc object. You can define whatever you want to do 
-Procmon.process("Mail") do |process|
-  process.checks :mem_usage, :above => 100.megabytes, :actions => [notifier, notifier2]
-end
-
-# You can specify a code block as well
-Procmon.process("Mail") do |process|
-  process.pid = 29124
-  process.checks :mem_usage, :above => 100.megabytes do
-    puts "I'm in your base"
-  end
-end
-
-# Using built-in email notifier
-NOTIFICATION_TARGET='ddao@example.com'
-Procmon.process("Mail") do |process|
-  process.checks :mem_usage, :above => 100.megabytes, :actions => [Procmon::Notifiers::Email]
-end
-
-# Would be nice if we can do this as well
-# Procmon.process("Mail") do |process|
-#  process.checks :mem_usage, :above => 100.megabytes do |action|
-#    action.perform(notifier)
-#    action.perform(Procmon::Notifiers::Email, 'someone@example.com')
-#  end
-#  process.checks :mem_usage, :above => 100.megabytes, :actions => [notifier, notifier2]
-#  process.checks :mem_usage, :above => 100.megabytes, :actions => [Procmon::Notifiers::Email]
-#end
+## Usage
+
+```ruby
+    require 'rubygems'
+    require 'procmon'
+
+    notifier = Proc.new do
+      puts "I NEED TO SEND OUT AN EMAIL"
+    end
+
+    notifier2 = Proc.new do
+      puts "I NEED TO SEND OUT AN SMS"
+    end
+ 
+    # action can be a Proc object. You can define whatever you want to do 
+    Procmon.process("Mail") do |process|
+      process.checks :mem_usage, :above => 100.megabytes, :actions => [notifier, notifier2]
+    end
+ 
+    # You can specify a code block as well
+    Procmon.process("Mail") do |process|
+      process.pid = 29124
+      process.checks :mem_usage, :above => 100.megabytes do
+        puts "I'm in your base"
+      end
+    end
+ 
+    # Using built-in email notifier
+    NOTIFICATION_TARGET='ddao@example.com'
+    Procmon.process("Mail") do |process|
+      process.checks :mem_usage, :above => 100.megabytes, :actions => [Procmon::Notifiers::Email]
+    end
+ 
+    # Check if a process is alive
+    Procmon.process("libvirtd") do |process|
+      process.checks :process_health, :status => "down" do
+        puts "Oh no! My process is not running"
+      end
+    end
+ 
+    # Would be nice if we can do this as well
+    # Procmon.process("Mail") do |process|
+    #  process.checks :mem_usage, :above => 100.megabytes do |action|
+    #    action.perform(notifier)
+    #    action.perform(Procmon::Notifiers::Email, 'someone@example.com')
+    #  end
+    #  process.checks :mem_usage, :above => 100.megabytes, :actions => [notifier, notifier2]
+    #  process.checks :mem_usage, :above => 100.megabytes, :actions => [Procmon::Notifiers::Email]
+    #end
 ```

data/lib/procmon/process_conditions/cpu_usage.rb

@@ -0,0 +1,38 @@
+module Procmon
+  module ProcessConditions
+    class CpuUsage < ProcessCondition
+      def initialize(options = {})
+        super(options)
+        @below = options[:below]
+        @above = options[:above]
+        if @below && @above && @below < @above
+          raise "Invalid range for mem check condition"
+        elsif @below.nil? && @above.nil?
+          raise "Invalid range for mem check condition"
+        end
+      end
+
+      def run(pid)
+        System.cpu_usage(pid).to_f
+      end
+
+      def check(value)
+        if @below && @above
+          value < @below && value > @above
+        elsif @below
+          value < @below 
+        elsif @above
+          value > @above
+        end
+      end
+
+      def description
+        ret = "Cpu usage is "
+        conditions = {}
+        conditions[:below] = @options[:below] if @options[:below]
+        conditions[:above] = @options[:above] if @options[:above]
+        ret += conditions.inspect
+      end
+    end
+  end
+end

data/lib/procmon/process_conditions/mem_usage.rb

@@ -6,11 +6,8 @@ module Procmon
       MB_LABEL = "MB"
       KB_LABEL = "KB"
 
-      attr_reader :actions
-
       def initialize(options = {})
-        @options = options
-        @actions = options[:actions]
+        super(options)
         @below = options[:below]
         @above = options[:above]
         if @below && @above && @below < @above

data/lib/procmon/process_conditions/process_condition.rb

@@ -1,8 +1,10 @@
 module Procmon
   module ProcessConditions
     class ProcessCondition
+      attr_reader :actions
       def initialize(options = {})
         @options = options
+        @actions = options[:actions]
       end
 
       def run(pid)

data/lib/procmon/process_conditions/process_health.rb

@@ -0,0 +1,26 @@
+module Procmon
+  module ProcessConditions
+    class ProcessHealth < ProcessCondition
+      def initialize(options = {})
+        super(options)
+        @status = options[:status]
+        if @status != "up" and @status != "down"
+          raise "Don't know what to do for process_health check. You need to specify :status => \"down\" or :status => \"up\""
+        end
+      end
+
+      def run(pid)
+        return false if pid.nil?
+        System.pid_alive?(pid)
+      end
+
+      def check(value)
+        !(value ^ (@status == "up"))
+      end
+
+      def description
+        "Checking to see if is_up = #{@is_up}"
+      end
+    end
+  end
+end

data/lib/procmon/system.rb

@@ -15,10 +15,23 @@ module Procmon
       @store ||= Hash.new
     end
 
+    def pid_alive?(pid)
+      begin
+        ::Process.kill(0, pid)
+        true
+      rescue Errno::ESRCH
+        false
+      end
+    end
+
     def memory_usage(pid)
       ps_axu[pid] && ps_axu[pid][IDX_MAP[:rss]].to_f
     end
 
+    def cpu_usage(pid)
+      ps_axu[pid] && ps_axu[pid][IDX_MAP[:pcpu]].to_f
+    end 
+
     def ps_axu
       # TODO: need a mutex here
       store[:ps_axu] ||= begin

data/lib/procmon/version.rb

@@ -1,3 +1,3 @@
 module Procmon
-  VERSION = "0.0.3".freeze
+  VERSION = "0.0.4".freeze
 end

data/lib/test.rb

@@ -1,6 +1,6 @@
 require 'procmon'
 
-NOTIFICATION_TARGET='ddao@example.com'
+NOTIFICATION_TARGET='ddao@eharmony.com'
 
 notifier = Proc.new do 
   puts "I NEED TO SEND OUT AN EMAIL"
@@ -11,7 +11,20 @@ notifier2 = Proc.new do
 end
 
 Procmon.process("Mail") do |process|
-  process.pid = 29124
+  process.pid = 8264
+  process.checks :cpu_usage, :above => 3 do
+    puts "OOOO SHOOT"
+  end
+end
+
+Procmon.process("Mailerious") do |process|
+  process.checks :process_health, :status => "up" do
+    puts "OH SHOOT PROCESS IS UP"
+  end
+end
+
+Procmon.process("Mail") do |process|
+#  process.pid = 29124
   process.checks :mem_usage, :above => 100.megabytes do
     puts "I'm in your base"
   end
@@ -21,8 +34,12 @@ Procmon.process("Mail") do |process|
   process.checks :mem_usage, :above => 100.megabytes, :actions => [notifier, notifier2]
 end
 
+#Procmon.process("Mail") do |process|
+#  process.checks :mem_usage, :above => 100.megabytes, :actions => [Procmon::Notifiers::Email]
+#end
+
 Procmon.process("Mail") do |process|
-  process.checks :mem_usage, :above => 100.megabytes, :actions => [Procmon::Notifiers::Email]
+  process.checks :mem_usage, :above => 100.megabytes, :actions => [notifier, notifier2]
 end
 
 # Would be nice if we can do this

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: procmon
 version: !ruby/object:Gem::Version 
-  hash: 25
+  hash: 23
   prerelease: 
   segments: 
   - 0
   - 0
-  - 3
-  version: 0.0.3
+  - 4
+  version: 0.0.4
 platform: ruby
 authors: 
 - Darren Dao
@@ -70,8 +70,10 @@ files:
 - lib/procmon/notifiers/notifier.rb
 - lib/procmon/process.rb
 - lib/procmon/process_conditions.rb
+- lib/procmon/process_conditions/cpu_usage.rb
 - lib/procmon/process_conditions/mem_usage.rb
 - lib/procmon/process_conditions/process_condition.rb
+- lib/procmon/process_conditions/process_health.rb
 - lib/procmon/process_notification.rb
 - lib/procmon/system.rb
 - lib/procmon/version.rb
@@ -112,3 +114,4 @@ specification_version: 3
 summary: A process monitor written in Ruby. Concepts and design are based on Bluepill and God, but with emphasis on simplicity and extensibility.
 test_files: []
 
+has_rdoc: