summaryrefslogtreecommitdiff
path: root/lib
diff options
context:
space:
mode:
authorSteve Kemp <steve@steve.org.uk>2015-02-05 11:30:35 +0000
committerSteve Kemp <steve@steve.org.uk>2015-02-05 11:30:35 +0000
commitb349e6ab6305a054bc1bf6b568a1e9fb7425f404 (patch)
treef79725efd627778e3e1e8e6abff0b0235519991e /lib
parentf853516b77dd58e65f7dfb08b32b93936d7567ec (diff)
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.
Diffstat (limited to 'lib')
-rw-r--r--lib/custodian/queue.rb14
-rw-r--r--lib/custodian/settings.rb13
2 files changed, 11 insertions, 16 deletions
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
diff --git a/lib/custodian/settings.rb b/lib/custodian/settings.rb
index 897c099..bef2977 100644
--- a/lib/custodian/settings.rb
+++ b/lib/custodian/settings.rb
@@ -140,18 +140,11 @@ module Custodian
#
- # The address of the queue.
- #
- def queue_server
- _load() unless( _loaded? )
-
- @settings['queue_server'] || "127.0.0.1:11300"
- end
-
-
- #
# The type of queue to use.
#
+ # The hostname/IP address of the queue should be set in the
+ # environmental variable 'QUEUE_ADDRESS'.
+ #
def queue_type
_load() unless( _loaded? )