diff options
author | James Lopez <james@jameslopez.es> | 2017-05-03 18:20:12 +0300 |
---|---|---|
committer | James Lopez <james@jameslopez.es> | 2017-05-03 18:20:12 +0300 |
commit | 264bf229277caf1c1eaca4e83921ca1b305d5401 (patch) | |
tree | 610c4b0427f8303a5484c06e6f4779b7c9e1669d /app/workers | |
parent | e81ea165ba738fedab07d5e20423856e004e2594 (diff) |
add propagate service worker and updated spec and controller
Diffstat (limited to 'app/workers')
-rw-r--r-- | app/workers/propagate_project_service_worker.rb | 49 |
1 files changed, 49 insertions, 0 deletions
diff --git a/app/workers/propagate_project_service_worker.rb b/app/workers/propagate_project_service_worker.rb new file mode 100644 index 00000000000..53551770968 --- /dev/null +++ b/app/workers/propagate_project_service_worker.rb @@ -0,0 +1,49 @@ +# Worker for updating any project specific caches. +class PropagateProjectServiceWorker + include Sidekiq::Worker + include DedicatedSidekiqQueue + + LEASE_TIMEOUT = 30.minutes.to_i + + def perform(template_id) + template = Service.find_by(id: template_id) + + return unless template&.active + return unless try_obtain_lease_for(template.id) + + Rails.logger.info("Propagating services for template #{template.id}") + + project_ids_for_template(template) do |project_id| + Service.build_from_template(project_id, template).save! + end + end + + private + + def project_ids_for_template(template) + limit = 100 + offset = 0 + + loop do + batch = project_ids_batch(limit, offset, template.type) + + batch.each { |project_id| yield(project_id) } + + break if batch.count < limit + + offset += limit + end + end + + def project_ids_batch(limit, offset, template_type) + Project.joins('LEFT JOIN services ON services.project_id = projects.id'). + where('services.type != ? OR services.id IS NULL', template_type). + limit(limit).offset(offset).pluck(:id) + end + + def try_obtain_lease_for(template_id) + Gitlab::ExclusiveLease. + new("propagate_project_service_worker:#{template_id}", timeout: LEASE_TIMEOUT). + try_obtain + end +end |