2012-01-22 19:01:46 -05:00
|
|
|
require 'sidekiq/version'
|
|
|
|
require 'sidekiq/client'
|
2012-01-25 16:32:51 -05:00
|
|
|
require 'sidekiq/worker'
|
2012-03-10 15:30:15 -05:00
|
|
|
require 'sidekiq/rails'
|
2012-02-18 15:12:05 -05:00
|
|
|
require 'sidekiq/redis_connection'
|
2012-03-17 12:41:24 -04:00
|
|
|
require 'sidekiq/util'
|
2012-02-17 16:39:36 -05:00
|
|
|
|
2012-03-10 15:30:15 -05:00
|
|
|
require 'sidekiq/extensions/action_mailer'
|
|
|
|
require 'sidekiq/extensions/active_record'
|
2012-02-18 15:12:05 -05:00
|
|
|
|
2012-02-17 16:39:36 -05:00
|
|
|
module Sidekiq
|
2012-02-18 15:12:05 -05:00
|
|
|
|
2012-02-19 16:02:32 -05:00
|
|
|
DEFAULTS = {
|
|
|
|
:queues => [],
|
|
|
|
:concurrency => 25,
|
|
|
|
:require => '.',
|
|
|
|
:environment => nil,
|
2012-04-06 23:53:03 -04:00
|
|
|
:timeout => 8,
|
2012-03-26 19:05:33 -04:00
|
|
|
:enable_rails_extensions => true,
|
2012-02-19 16:02:32 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
def self.options
|
|
|
|
@options ||= DEFAULTS.dup
|
|
|
|
end
|
|
|
|
|
|
|
|
def self.options=(opts)
|
|
|
|
@options = opts
|
|
|
|
end
|
|
|
|
|
|
|
|
##
|
2012-02-20 12:46:28 -05:00
|
|
|
# Configuration for Sidekiq server, use like:
|
2012-02-19 16:02:32 -05:00
|
|
|
#
|
2012-02-20 12:46:28 -05:00
|
|
|
# Sidekiq.configure_server do |config|
|
2012-03-06 16:42:30 -05:00
|
|
|
# config.redis = { :namespace => 'myapp', :size => 25, :url => 'redis://myhost:8877/mydb' }
|
2012-02-19 16:02:32 -05:00
|
|
|
# config.server_middleware do |chain|
|
|
|
|
# chain.add MyServerHook
|
|
|
|
# end
|
|
|
|
# end
|
2012-02-20 12:46:28 -05:00
|
|
|
def self.configure_server
|
|
|
|
yield self if server?
|
|
|
|
end
|
|
|
|
|
|
|
|
##
|
|
|
|
# Configuration for Sidekiq client, use like:
|
|
|
|
#
|
|
|
|
# Sidekiq.configure_client do |config|
|
2012-03-06 16:42:30 -05:00
|
|
|
# config.redis = { :namespace => 'myapp', :size => 1, :url => 'redis://myhost:8877/mydb' }
|
2012-02-20 12:46:28 -05:00
|
|
|
# end
|
|
|
|
def self.configure_client
|
|
|
|
yield self unless server?
|
|
|
|
end
|
|
|
|
|
|
|
|
def self.server?
|
|
|
|
defined?(Sidekiq::CLI)
|
2012-02-19 16:02:32 -05:00
|
|
|
end
|
|
|
|
|
2012-03-14 12:56:13 -04:00
|
|
|
def self.redis(&block)
|
2012-02-17 16:51:46 -05:00
|
|
|
@redis ||= Sidekiq::RedisConnection.create
|
2012-03-28 22:16:54 -04:00
|
|
|
raise ArgumentError, "requires a block" if !block
|
|
|
|
@redis.with(&block)
|
2012-02-17 16:51:46 -05:00
|
|
|
end
|
2012-02-18 15:12:05 -05:00
|
|
|
|
2012-03-01 16:41:12 -05:00
|
|
|
def self.redis=(hash)
|
2012-03-28 22:16:54 -04:00
|
|
|
if hash.is_a?(Hash)
|
|
|
|
@redis = RedisConnection.create(hash)
|
2012-04-09 11:56:34 -04:00
|
|
|
elsif hash.is_a?(ConnectionPool)
|
2012-03-28 22:16:54 -04:00
|
|
|
@redis = hash
|
2012-04-09 11:56:34 -04:00
|
|
|
else
|
|
|
|
raise ArgumentError, "redis= requires a Hash or ConnectionPool"
|
2012-03-01 16:41:12 -05:00
|
|
|
end
|
2012-02-17 16:39:36 -05:00
|
|
|
end
|
2012-02-18 15:12:05 -05:00
|
|
|
|
|
|
|
def self.client_middleware
|
2012-02-19 16:02:32 -05:00
|
|
|
@client_chain ||= Client.default_middleware
|
2012-02-18 15:12:05 -05:00
|
|
|
yield @client_chain if block_given?
|
|
|
|
@client_chain
|
|
|
|
end
|
|
|
|
|
|
|
|
def self.server_middleware
|
2012-02-19 16:02:32 -05:00
|
|
|
@server_chain ||= Processor.default_middleware
|
2012-02-18 15:12:05 -05:00
|
|
|
yield @server_chain if block_given?
|
|
|
|
@server_chain
|
|
|
|
end
|
|
|
|
|
2012-02-17 16:39:36 -05:00
|
|
|
end
|