dosire-god 0.7.10 → 0.7.12

data/lib/god.rb

@@ -46,7 +46,8 @@ require 'god/conditions/always'
 require 'god/conditions/lambda'
 require 'god/conditions/degrading_lambda'
 require 'god/conditions/flapping'
-require 'god/conditions/http_response_code'
+require 'god/conditions/http_response_code'
+require 'god/conditions/http_response_body'
 require 'god/conditions/disk_usage'
 require 'god/conditions/complex'
 

data/lib/god/conditions/http_response_body.rb

@@ -0,0 +1,165 @@
+require 'net/http'
+
+module God
+  module Conditions
+    
+    # Condition Symbol :http_response_code
+    # Type: Poll
+    # 
+    # Trigger based on the body of an HTTP response.
+    #
+    # Paramaters
+    #   Required
+    #     +host+ is the hostname to connect [required]
+    #     --one of code_is or code_is_not--
+    #     +code_is+ trigger if the response code matches this string
+    #               e.g. '500'
+    #     +code_is_not+ trigger if the response code does not match this string
+    #                   e.g. '200'
+    #  Optional
+    #     +port+ is the port to connect (default 80)
+    #     +path+ is the path to connect (default '/')
+    #     +headers+ is the hash of HTTP headers to send (default none)
+    #     +times+ is the number of times after which to trigger (default 1)
+    #             e.g. 3 (times in a row) or [3, 5] (three out of fives times)
+    #     +timeout+ is the time to wait for a connection (default 60.seconds)
+    #
+    # Examples
+    #
+    # Trigger if the response code from www.example.com/foo/bar
+    # is not a 200 (or if the connection is refused or times out:
+    #
+    #   on.condition(:http_response_code) do |c|
+    #     c.host = 'www.example.com'
+    #     c.path = '/foo/bar'
+    #     c.code_is_not = 200
+    #   end
+    #
+    # Trigger if the response code is a 404 or a 500 (will not
+    # be triggered by a connection refusal or timeout):
+    #
+    #   on.condition(:http_response_code) do |c|
+    #     c.host = 'www.example.com'
+    #     c.path = '/foo/bar'
+    #     c.code_is = [404, 500]
+    #   end
+    #
+    # Trigger if the response code is not a 200 five times in a row:
+    #
+    #   on.condition(:http_response_code) do |c|
+    #     c.host = 'www.example.com'
+    #     c.path = '/foo/bar'
+    #     c.code_is_not = 200
+    #     c.times = 5
+    #   end
+    #
+    # Trigger if the response code is not a 200 or does not respond
+    # within 10 seconds:
+    #
+    #   on.condition(:http_response_code) do |c|
+    #     c.host = 'www.example.com'
+    #     c.path = '/foo/bar'
+    #     c.code_is_not = 200
+    #     c.timeout = 10
+    #   end
+    class HttpResponseBody < PollCondition
+      attr_accessor :code_is,      # e.g. 500 or '500' or [404, 500] or %w{404 500}
+                    :code_is_not,  # e.g. 200 or '200' or [200, 302] or %w{200 302}
+                    :times,        # e.g. 3 or [3, 5]
+                    :host,         # e.g. www.example.com
+                    :port,         # e.g. 8080
+                    :timeout,      # e.g. 60.seconds
+                    :path,         # e.g. '/'
+                    :headers       # e.g. {'Host' => 'myvirtual.mydomain.com'}
+      
+      def initialize
+        super
+        self.port = 80
+        self.path = '/'
+        self.headers = {}
+        self.times = [1, 1]
+        self.timeout = 60.seconds
+      end
+      
+      def prepare
+        if self.times.kind_of?(Integer)
+          self.times = [self.times, self.times]
+        end
+        
+        @timeline = Timeline.new(self.times[1])
+        @history = Timeline.new(self.times[1])
+      end
+      
+      def reset
+        @timeline.clear
+        @history.clear
+      end
+      
+      def valid?
+        valid = true
+        valid &= complain("Attribute 'host' must be specified", self) if self.host.nil?
+        valid &= complain("One (and only one) of attributes 'code_is' and 'code_is_not' must be specified", self) if
+          (self.code_is.nil? && self.code_is_not.nil?) || (self.code_is && self.code_is_not)
+        valid
+      end
+      
+      def test
+        response = nil
+        
+        Net::HTTP.start(self.host, self.port) do |http|
+          http.read_timeout = self.timeout
+          response = http.get(self.path)
+        end
+        
+        actual_response_code = response.body[5,10].to_s
+        if self.code_is && actual_response_code.include?(self.code_is)
+          pass(actual_response_code)
+        elsif self.code_is_not && !actual_response_code.include?(self.code_is_not)
+          pass(actual_response_code)
+        else
+          fail(actual_response_code)
+        end
+      rescue Errno::ECONNREFUSED
+        self.code_is ? fail('Refused') : pass('Refused')
+      rescue Errno::ECONNRESET
+        self.code_is ? fail('Reset') : pass('Reset')
+      rescue EOFError
+        self.code_is ? fail('EOF') : pass('EOF')
+      rescue Timeout::Error
+        self.code_is ? fail('Timeout') : pass('Timeout')
+      rescue Errno::ETIMEDOUT
+        self.code_is ? fail('Timedout') : pass('Timedout')
+      rescue Exception => failure
+        self.code_is ? fail(failure.class.name) : pass(failure.class.name)
+      end
+      
+      private
+      
+      def pass(code)
+        @timeline << true
+        if @timeline.select { |x| x }.size >= self.times.first
+          self.info = "http response abnormal #{history(code, true)}"
+          true
+        else
+          self.info = "http response nominal #{history(code, true)}"
+          false
+        end
+      end
+      
+      def fail(code)
+        @timeline << false
+        self.info = "http response nominal #{history(code, false)}"
+        false
+      end
+      
+      def history(code, passed)
+        entry = code.to_s.dup
+        entry = '*' + entry if passed
+        @history << entry
+        '[' + @history.join(", ") + ']'
+      end
+      
+    end
+    
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: dosire-god
 version: !ruby/object:Gem::Version 
-  version: 0.7.10
+  version: 0.7.12
 platform: ruby
 authors: 
 - Tom Preston-Werner
@@ -52,6 +52,7 @@ files:
 - lib/god/conditions/disk_usage.rb
 - lib/god/conditions/flapping.rb
 - lib/god/conditions/http_response_code.rb
+- lib/god/conditions/http_response_body.rb
 - lib/god/conditions/lambda.rb
 - lib/god/conditions/memory_usage.rb
 - lib/god/conditions/process_exits.rb