Class: Sidekiq::Config

Inherits:
Object
  • Object
show all
Extended by:
Forwardable
Defined in:
lib/sidekiq/config.rb

Overview

Sidekiq::Config represents the global configuration for an instance of Sidekiq.

Constant Summary collapse

DEFAULTS =
{
  labels: Set.new,
  require: ".",
  environment: nil,
  concurrency: 5,
  timeout: 25,
  poll_interval_average: nil,
  average_scheduled_poll_interval: 5,
  on_complex_arguments: :raise,
  iteration: {
    max_job_runtime: nil,
    retry_backoff: 0
  },
  error_handlers: [],
  death_handlers: [],
  lifecycle_events: {
    startup: [],
    quiet: [],
    shutdown: [],
    # triggers when we fire the first heartbeat on startup OR repairing a network partition
    heartbeat: [],
    # triggers on EVERY heartbeat call, every 10 seconds
    beat: []
  },
  dead_max_jobs: 10_000,
  dead_timeout_in_seconds: 180 * 24 * 60 * 60, # 6 months
  reloader: proc { |&block| block.call },
  backtrace_cleaner: ->(backtrace) { backtrace }
}
ERROR_HANDLER =
->(ex, ctx, cfg = Sidekiq.default_configuration) {
  l = cfg.logger
  l.warn(Sidekiq.dump_json(ctx)) unless ctx.empty?
  l.warn("#{ex.class.name}: #{ex.message}")
  unless ex.backtrace.nil?
    backtrace = cfg[:backtrace_cleaner].call(ex.backtrace)
    l.warn(backtrace.join("\n"))
  end
}

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(options = {}) ⇒ Config

Returns a new instance of Config.



53
54
55
56
57
58
59
# File 'lib/sidekiq/config.rb', line 53

def initialize(options = {})
  @options = DEFAULTS.merge(options)
  @options[:error_handlers] << ERROR_HANDLER if @options[:error_handlers].empty?
  @directory = {}
  @redis_config = {}
  @capsules = {}
end

Instance Attribute Details

#capsulesObject (readonly)

Returns the value of attribute capsules.



62
63
64
# File 'lib/sidekiq/config.rb', line 62

def capsules
  @capsules
end

Instance Method Details

#average_scheduled_poll_interval=(interval) ⇒ Object

How frequently Redis should be checked by a random Sidekiq process for scheduled and retriable jobs. Each individual process will take turns by waiting some multiple of this value.

See sidekiq/scheduled.rb for an in-depth explanation of this value



236
237
238
# File 'lib/sidekiq/config.rb', line 236

def average_scheduled_poll_interval=(interval)
  @options[:average_scheduled_poll_interval] = interval
end

#capsule(name) {|cap| ... } ⇒ Object

register a new queue processing subsystem

Yields:

  • (cap)


123
124
125
126
127
128
129
130
131
# File 'lib/sidekiq/config.rb', line 123

def capsule(name)
  nm = name.to_s
  cap = @capsules.fetch(nm) do
    cap = Sidekiq::Capsule.new(nm, self)
    @capsules[nm] = cap
  end
  yield cap if block_given?
  cap
end

#client_middleware {|@client_chain| ... } ⇒ Object

Yields:

  • (@client_chain)


106
107
108
109
110
# File 'lib/sidekiq/config.rb', line 106

def client_middleware
  @client_chain ||= Sidekiq::Middleware::Chain.new(self)
  yield @client_chain if block_given?
  @client_chain
end

#concurrencyObject



80
81
82
# File 'lib/sidekiq/config.rb', line 80

def concurrency
  default_capsule.concurrency
end

#concurrency=(val) ⇒ Object

LEGACY: edits the default capsule config.concurrency = 5



76
77
78
# File 'lib/sidekiq/config.rb', line 76

def concurrency=(val)
  default_capsule.concurrency = Integer(val)
end

#death_handlersObject

Death handlers are called when all retries for a job have been exhausted and the job dies. It’s the notification to your application that this job will not succeed without manual intervention.

Sidekiq.configure_server do |config|

config.death_handlers << ->(job, ex) do
end

end



227
228
229
# File 'lib/sidekiq/config.rb', line 227

def death_handlers
  @options[:death_handlers]
end

#default_capsule(&block) ⇒ Object



118
119
120
# File 'lib/sidekiq/config.rb', line 118

def default_capsule(&block)
  capsule("default", &block)
end

#error_handlersObject

Register a proc to handle any error which occurs within the Sidekiq process.

Sidekiq.configure_server do |config|
  config.error_handlers << proc {|ex,ctx_hash| MyErrorService.notify(ex, ctx_hash) }
end

The default error handler logs errors to @logger.



247
248
249
# File 'lib/sidekiq/config.rb', line 247

def error_handlers
  @options[:error_handlers]
end

#freeze!Object



212
213
214
215
216
# File 'lib/sidekiq/config.rb', line 212

def freeze!
  @directory.freeze
  @options.freeze
  true
end

#handle_exception(ex, ctx = {}) ⇒ Object

INTERNAL USE ONLY



291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
# File 'lib/sidekiq/config.rb', line 291

def handle_exception(ex, ctx = {})
  if @options[:error_handlers].size == 0
    p ["!!!!!", ex]
  end
  @options[:error_handlers].each do |handler|
    if parameter_size(handler) == 2
      # TODO Remove in 8.0
      logger.info { "DEPRECATION: Sidekiq exception handlers now take three arguments, see #{handler}" }
      handler.call(ex, {_config: self}.merge(ctx))
    else
      handler.call(ex, ctx, self)
    end
  rescue Exception => e
    l = logger
    l.error "!!! ERROR HANDLER THREW AN ERROR !!!"
    l.error e
    l.error e.backtrace.join("\n") unless e.backtrace.nil?
  end
end

#inspectObject



64
65
66
67
68
# File 'lib/sidekiq/config.rb', line 64

def inspect
  "#<#{self.class.name} @options=#{
    @options.except(:lifecycle_events, :reloader, :death_handlers, :error_handlers).inspect
  }>"
end

#loggerObject



265
266
267
268
269
270
271
272
273
274
# File 'lib/sidekiq/config.rb', line 265

def logger
  @logger ||= Sidekiq::Logger.new($stdout, level: :info).tap do |log|
    log.level = Logger::INFO
    log.formatter = if ENV["DYNO"]
      Sidekiq::Logger::Formatters::WithoutTimestamp.new
    else
      Sidekiq::Logger::Formatters::Pretty.new
    end
  end
end

#logger=(logger) ⇒ Object



276
277
278
279
280
281
282
283
# File 'lib/sidekiq/config.rb', line 276

def logger=(logger)
  if logger.nil?
    self.logger.level = Logger::FATAL
    return
  end

  @logger = logger
end

#lookup(name, default_class = nil) ⇒ Object

find a singleton



204
205
206
207
208
209
210
# File 'lib/sidekiq/config.rb', line 204

def lookup(name, default_class = nil)
  # JNDI is just a fancy name for a hash lookup
  @directory.fetch(name) do |key|
    return nil unless default_class
    register(key, default_class.new(self))
  end
end

#new_redis_pool(size, name = "unset") ⇒ Object



148
149
150
151
152
# File 'lib/sidekiq/config.rb', line 148

def new_redis_pool(size, name = "unset")
  # connection pool is lazy, it will not create connections unless you actually need them
  # so don't be skimpy!
  RedisConnection.create({size: size, logger: logger, pool_name: name}.merge(@redis_config))
end

#on(event, &block) ⇒ Object

Register a block to run at a point in the Sidekiq lifecycle. :startup, :quiet or :shutdown are valid events.

Sidekiq.configure_server do |config|
  config.on(:shutdown) do
    puts "Goodbye cruel world!"
  end
end

Raises:

  • (ArgumentError)


259
260
261
262
263
# File 'lib/sidekiq/config.rb', line 259

def on(event, &block)
  raise ArgumentError, "Symbols only please: #{event}" unless event.is_a?(Symbol)
  raise ArgumentError, "Invalid event name: #{event}" unless @options[:lifecycle_events].key?(event)
  @options[:lifecycle_events][event] << block
end

#queuesObject



102
103
104
# File 'lib/sidekiq/config.rb', line 102

def queues
  default_capsule.queues
end

#queues=(val) ⇒ Object

Edit the default capsule. config.queues = %w( high default low ) # strict config.queues = %w( high,3 default,2 low,1 ) # weighted config.queues = %w( feature1,1 feature2,1 feature3,1 ) # random

With weighted priority, queue will be checked first (weight / total) of the time. high will be checked first (3/6) or 50% of the time. I’d recommend setting weights between 1-10. Weights in the hundreds or thousands are ridiculous and unnecessarily expensive. You can get random queue ordering by explicitly setting all weights to 1.



98
99
100
# File 'lib/sidekiq/config.rb', line 98

def queues=(val)
  default_capsule.queues = val
end

#redisObject

Raises:

  • (ArgumentError)


170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
# File 'lib/sidekiq/config.rb', line 170

def redis
  raise ArgumentError, "requires a block" unless block_given?
  redis_pool.with do |conn|
    retryable = true
    begin
      yield conn
    rescue RedisClientAdapter::BaseError => ex
      # 2550 Failover can cause the server to become a replica, need
      # to disconnect and reopen the socket to get back to the primary.
      # 4495 Use the same logic if we have a "Not enough replicas" error from the primary
      # 4985 Use the same logic when a blocking command is force-unblocked
      # The same retry logic is also used in client.rb
      if retryable && ex.message =~ /READONLY|NOREPLICAS|UNBLOCKED/
        conn.close
        retryable = false
        retry
      end
      raise
    end
  end
end

#redis=(hash) ⇒ Object

All capsules must use the same Redis configuration



134
135
136
# File 'lib/sidekiq/config.rb', line 134

def redis=(hash)
  @redis_config = @redis_config.merge(hash)
end

#redis_infoObject



154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
# File 'lib/sidekiq/config.rb', line 154

def redis_info
  redis do |conn|
    conn.call("INFO") { |i| i.lines(chomp: true).map { |l| l.split(":", 2) }.select { |l| l.size == 2 }.to_h }
  rescue RedisClientAdapter::CommandError => ex
    # 2850 return fake version when INFO command has (probably) been renamed
    raise unless /unknown command/.match?(ex.message)
    {
      "redis_version" => "9.9.9",
      "uptime_in_days" => "9999",
      "connected_clients" => "9999",
      "used_memory_human" => "9P",
      "used_memory_peak_human" => "9P"
    }.freeze
  end
end

#redis_poolObject



138
139
140
# File 'lib/sidekiq/config.rb', line 138

def redis_pool
  Thread.current[:sidekiq_redis_pool] || Thread.current[:sidekiq_capsule]&.redis_pool || local_redis_pool
end

#register(name, instance) ⇒ Object

register global singletons which can be accessed elsewhere



193
194
195
196
197
198
199
200
201
# File 'lib/sidekiq/config.rb', line 193

def register(name, instance)
  # logger.debug("register[#{name}] = #{instance}")
  # Sidekiq Enterprise lazy registers a few services so we
  # can't lock down this hash completely.
  hash = @directory.dup
  hash[name] = instance
  @directory = hash.freeze
  instance
end

#server_middleware {|@server_chain| ... } ⇒ Object

Yields:

  • (@server_chain)


112
113
114
115
116
# File 'lib/sidekiq/config.rb', line 112

def server_middleware
  @server_chain ||= Sidekiq::Middleware::Chain.new(self)
  yield @server_chain if block_given?
  @server_chain
end

#to_jsonObject



70
71
72
# File 'lib/sidekiq/config.rb', line 70

def to_json(*)
  Sidekiq.dump_json(@options)
end

#total_concurrencyObject



84
85
86
# File 'lib/sidekiq/config.rb', line 84

def total_concurrency
  capsules.each_value.sum(&:concurrency)
end