sidekiq.rb 1.89 KB
Newer Older
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
1 2
Sidekiq.configure_server do |config|
  config.redis = {
3
    url: Gitlab::Redis.url,
4
    namespace: Gitlab::Redis::SIDEKIQ_NAMESPACE
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
5
  }
Jacob Vosmaer's avatar
Jacob Vosmaer committed
6 7

  config.server_middleware do |chain|
8
    chain.add Gitlab::SidekiqMiddleware::ArgumentsLogger if ENV['SIDEKIQ_LOG_ARGUMENTS']
9
    chain.add Gitlab::SidekiqMiddleware::MemoryKiller if ENV['SIDEKIQ_MEMORY_KILLER_MAX_RSS']
10
    chain.add Gitlab::SidekiqMiddleware::RequestStoreMiddleware unless ENV['SIDEKIQ_REQUEST_STORE'] == '0'
Jacob Vosmaer's avatar
Jacob Vosmaer committed
11
  end
12

13
  # Sidekiq-cron: load recurring jobs from gitlab.yml
14 15 16
  # UGLY Hack to get nested hash from settingslogic
  cron_jobs = JSON.parse(Gitlab.config.cron_jobs.to_json)
  # UGLY hack: Settingslogic doesn't allow 'class' key
17 18 19 20 21
  cron_jobs_required_keys = %w(job_class cron)
  cron_jobs.each do |k, v|
    if cron_jobs[k] && cron_jobs_required_keys.all? { |s| cron_jobs[k].key?(s) }
      cron_jobs[k]['class'] = cron_jobs[k].delete('job_class')
    else
22 23
      cron_jobs.delete(k)
      Rails.logger.error("Invalid cron_jobs config key: '#{k}'. Check your gitlab config file.")
24 25
    end
  end
26
  Sidekiq::Cron::Job.load_from_hash! cron_jobs
Gabriel Mazetto's avatar
Gabriel Mazetto committed
27 28 29 30 31 32 33 34

  # Database pool should be at least `sidekiq_concurrency` + 2
  # For more info, see: https://github.com/mperham/sidekiq/blob/master/4.0-Upgrade.md
  config = ActiveRecord::Base.configurations[Rails.env] ||
                Rails.application.config.database_configuration[Rails.env]
  config['pool'] = Sidekiq.options[:concurrency] + 2
  ActiveRecord::Base.establish_connection(config)
  Rails.logger.debug("Connection Pool size for Sidekiq Server is now: #{ActiveRecord::Base.connection.pool.instance_variable_get('@size')}")
35 36 37 38

  # Avoid autoload issue such as 'Mail::Parsers::AddressStruct'
  # https://github.com/mikel/mail/issues/912#issuecomment-214850355
  Mail.eager_autoload!
39 40
end

Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
41 42
Sidekiq.configure_client do |config|
  config.redis = {
43
    url: Gitlab::Redis.url,
44
    namespace: Gitlab::Redis::SIDEKIQ_NAMESPACE
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
45 46
  }
end