Moved the worker into the custodian namespace.

1 files changed, 162 insertions, 158 deletions

diff --git a/lib/custodian/worker.rb b/lib/custodian/worker.rb
index 0876f11..73cc97c 100644
--- a/lib/custodian/worker.rb
+++ b/lib/custodian/worker.rb
@@ -36,216 +36,220 @@ require 'custodian/protocol-tests/tcp.rb'
 # This class contains the code for connecting to a Beanstalk queue,
 # fetching tests from it, and executing them
 #
-class CustodianWorker
-
-  #
-  # The beanstalk queue.
-  #
-  attr_reader :queue
-
-  #
-  # How many times we re-test before we detect a failure
-  #
-  attr_reader :retry_count
-
-  #
-  # The log-file object
-  #
-  attr_reader :logger
-
-  #
-  # Constructor: Connect to the queue
-  #
-  def initialize( server, logfile )
-
-    # Connect to the queue
-    @queue = Beanstalk::Pool.new([server])
-
-    # Instantiate the logger.
-    @logger = Logger.new( logfile, "daily" )
-
-    if ( ENV['REPEAT'] )
-       @retry_count=ENV['REPEAT'].to_i
-    else
-       @retry_count=5
-    end
+module Custodian
 
-  end
+  class Worker
+
+
+    #
+    # The beanstalk queue.
+    #
+    attr_reader :queue
 
+    #
+    # How many times we re-test before we detect a failure
+    #
+    attr_reader :retry_count
 
+    #
+    # The log-file object
+    #
+    attr_reader :logger
 
-  #
-  # Write the given message to our logfile - and show it to the console
-  # if we're running with '--verbose' in play
-  #
-  def log_message( msg )
+    #
+    # Constructor: Connect to the queue
+    #
+    def initialize( server, logfile )
+
+      # Connect to the queue
+      @queue = Beanstalk::Pool.new([server])
+
+      # Instantiate the logger.
+      @logger = Logger.new( logfile, "daily" )
+
+      if ( ENV['REPEAT'] )
+        @retry_count=ENV['REPEAT'].to_i
+      else
+        @retry_count=5
+      end
+
+    end
+
+
+
+    #
+    # Write the given message to our logfile - and show it to the console
+    # if we're running with '--verbose' in play
+    #
+    def log_message( msg )
       @logger.info( msg )
       puts msg if ( ENV['VERBOSE'] )
-  end
+    end
 
 
 
-  #
-  # Process jobs from the queue - never return.
-  #
-  def run!
-    while( true )
-      log_message( "\n" )
-      log_message( "\n" )
-      log_message( "Waiting for job.." )
-      process_single_job()
+    #
+    # Process jobs from the queue - never return.
+    #
+    def run!
+      while( true )
+        log_message( "\n" )
+        log_message( "\n" )
+        log_message( "Waiting for job.." )
+        process_single_job()
+      end
     end
-  end
 
 
 
-  #
-  # Fetch a single job from the queue, and process it.
-  #
-  def process_single_job
+    #
+    # Fetch a single job from the queue, and process it.
+    #
+    def process_single_job
 
-    result = false
+      result = false
 
-    begin
-      job = @queue.reserve()
+      begin
+        job = @queue.reserve()
 
-      log_message( "Job aquired - Job ID : #{job.id}" )
+        log_message( "Job aquired - Job ID : #{job.id}" )
 
 
-      #
-      #  Get the job body
-      #
-      json = job.body
-      raise ArgumentError, "Body doesn't look like JSON" unless( json =~ /[{}]/ )
+        #
+        #  Get the job body
+        #
+        json = job.body
+        raise ArgumentError, "Body doesn't look like JSON" unless( json =~ /[{}]/ )
 
 
-      #
-      # Decode the JSON body - it should return a non-empty hash.
-      #
-      hash = JSON.parse( json )
+        #
+        # Decode the JSON body - it should return a non-empty hash.
+        #
+        hash = JSON.parse( json )
 
-      #
-      # Ensure we got a non-empty hash.
-      #
-      raise ArgumentError, "JSON didn't decode to a hash" unless hash.kind_of?(Hash)
-      raise ArgumentError, "JSON hash is empty" if (hash.empty?)
+        #
+        # Ensure we got a non-empty hash.
+        #
+        raise ArgumentError, "JSON didn't decode to a hash" unless hash.kind_of?(Hash)
+        raise ArgumentError, "JSON hash is empty" if (hash.empty?)
 
-      #
-      # Are we being verbose?
-      #
-      hash['verbose'] = 1 if ( ENV['VERBOSE'] )
+        #
+        # Are we being verbose?
+        #
+        hash['verbose'] = 1 if ( ENV['VERBOSE'] )
 
 
-      #
-      #  Output the details.
-      #
-      log_message( "Job body contains the following keys & values:")
-      hash.keys.each do |key|
-        log_message( "   #{key} => #{hash[key]}" )
-      end
+        #
+        #  Output the details.
+        #
+        log_message( "Job body contains the following keys & values:")
+        hash.keys.each do |key|
+          log_message( "   #{key} => #{hash[key]}" )
+        end
 
 
 
-      #
-      # Did the test succeed?  If not count the number of times it failed in
-      # a row.  We'll repeat several times
-      #
-      success = false
-      count   = 0
+        #
+        # Did the test succeed?  If not count the number of times it failed in
+        # a row.  We'll repeat several times
+        #
+        success = false
+        count   = 0
 
-      #
-      # As a result of this test we'll either raise/clear with mauve.
-      #
-      # This helper will do that job.
-      #
-      alert = Alerter.new( hash )
+        #
+        # As a result of this test we'll either raise/clear with mauve.
+        #
+        # This helper will do that job.
+        #
+        alert = Alerter.new( hash )
 
 
-      #
-      # Convert the test-type to a class name, to do the protocol test.
-      #
-      # Given a test-type "foo" we'll attempt to instantiate a class called FOOTest.
-      #
-      test  = hash['test_type']
-      test  = "http" if ( test == "https" )
-      clazz = test.upcase
-      clazz = "#{clazz}Test"
+        #
+        # Convert the test-type to a class name, to do the protocol test.
+        #
+        # Given a test-type "foo" we'll attempt to instantiate a class called FOOTest.
+        #
+        test  = hash['test_type']
+        test  = "http" if ( test == "https" )
+        clazz = test.upcase
+        clazz = "#{clazz}Test"
 
 
-      #
-      # Create the test object.
-      #
-      obj = eval(clazz).new( hash )
+        #
+        # Create the test object.
+        #
+        obj = eval(clazz).new( hash )
 
 
-      #
-      # Ensure that the object we load implements the two methods
-      # we expect.
-      #
-      if ( ( ! obj.respond_to?( "error") ) ||
-           ( ! obj.respond_to?( "run_test" ) ) )
-        puts "Class #{clazz} doesn't implement the full protocol-test API"
-      end
+        #
+        # Ensure that the object we load implements the two methods
+        # we expect.
+        #
+        if ( ( ! obj.respond_to?( "error") ) ||
+             ( ! obj.respond_to?( "run_test" ) ) )
+          puts "Class #{clazz} doesn't implement the full protocol-test API"
+        end
 
 
 
-      #
-      #  We'll run no more than MAX times.
-      #
-      #  We stop the execution on a single success.
-      #
-      while ( ( count < @retry_count ) && ( success == false ) )
+        #
+        #  We'll run no more than MAX times.
+        #
+        #  We stop the execution on a single success.
+        #
+        while ( ( count < @retry_count ) && ( success == false ) )
 
-        log_message( "Running test - [#{count}/#{@retry_count}]" )
+          log_message( "Running test - [#{count}/#{@retry_count}]" )
 
-        if ( obj.run_test() )
-          log_message( "Test succeeed - clearing alert" )
-          success = true
-          alert.clear()
-          result = true
+          if ( obj.run_test() )
+            log_message( "Test succeeed - clearing alert" )
+            success = true
+            alert.clear()
+            result = true
+          end
+          count += 1
+        end
+
+        #
+        #  If we didn't succeed on any of the attempts raise the alert.
+        #
+        if ( ! success )
+
+          #
+          # Raise the alert, passing the error message.
+          #
+          log_message( "Test failed - alerting with #{obj.error()}" )
+          alert.raise( obj.error() )
         end
-        count += 1
-      end
 
-      #
-      #  If we didn't succeed on any of the attempts raise the alert.
-      #
-      if ( ! success )
+      rescue => ex
+        puts "Exception raised processing job: #{ex}"
 
+      ensure
         #
-        # Raise the alert, passing the error message.
+        #  Delete the job - either we received an error, in which case
+        # we should remove it to avoid picking it up again, or we handled
+        # it successfully so it should be removed.
         #
-        log_message( "Test failed - alerting with #{obj.error()}" )
-        alert.raise( obj.error() )
+        log_message( "Job ID : #{job.id} - Removed" )
+        job.delete if ( job )
       end
 
-    rescue => ex
-      puts "Exception raised processing job: #{ex}"
-
-    ensure
-      #
-      #  Delete the job - either we received an error, in which case
-      # we should remove it to avoid picking it up again, or we handled
-      # it successfully so it should be removed.
-      #
-      log_message( "Job ID : #{job.id} - Removed" )
-      job.delete if ( job )
+      return result
     end
 
-    return result
-  end
-
 
-  #
-  #  Process jobs until we see a failure - stop then.
-  #
-  def process_until_fail
+    #
+    #  Process jobs until we see a failure - stop then.
+    #
+    def process_until_fail
       while( process_single_job() )
-          # nop
+        # nop
       end
-  end
+    end
 
-end
+  end
 
 
 
+end