From b349e6ab6305a054bc1bf6b568a1e9fb7425f404 Mon Sep 17 00:00:00 2001 From: Steve Kemp Date: Thu, 5 Feb 2015 11:30:35 +0000 Subject: Simplify code by removing configuration. We now set the queue-address via $QUEUE_ADDRESS, otherwise we default to localhost. This works for both redis & beanstalkd. --- lib/custodian/queue.rb | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) (limited to 'lib/custodian/queue.rb') diff --git a/lib/custodian/queue.rb b/lib/custodian/queue.rb index d3513c3..67c5346 100644 --- a/lib/custodian/queue.rb +++ b/lib/custodian/queue.rb @@ -76,10 +76,10 @@ end # - # Connect to the server on localhost + # Connect to the server on localhost, unless QUEUE_ADDRESS is set. # def initialize - host = ENV["REDIS"] || "127.0.0.1" + host = ENV["QUEUE_ADDRESS"] || "127.0.0.1" @redis = Redis.new( :host => host ) end @@ -124,11 +124,11 @@ end class BeanstalkQueueType < QueueType # - # Connect to the server on localhost + # Connect to the server on localhost, unless QUEUE_ADDRESS is set. # def initialize - host = ENV["QUEUE"] || "127.0.0.1:11300" - @queue = Beanstalk::Pool.new([host] ) + host = ENV["QUEUE_ADDRESS"] || "127.0.0.1" + @queue = Beanstalk::Pool.new( ["#{host}:11300" ] ) end # @@ -168,8 +168,10 @@ end ( stats['current-jobs-ready'] || 0 ) end + # + # Flush the queue. + # def flush! - while( fetch(1) ) # nop end -- cgit v1.2.1