2016-10-20 06:59:39 -04:00
|
|
|
# Worker for updating any project specific caches.
|
|
|
|
#
|
|
|
|
# This worker runs at most once every 15 minutes per project. This is to ensure
|
|
|
|
# that multiple instances of jobs for this worker don't hammer the underlying
|
|
|
|
# storage engine as much.
|
2015-07-17 08:50:03 -04:00
|
|
|
class ProjectCacheWorker
|
|
|
|
include Sidekiq::Worker
|
2016-10-21 12:13:41 -04:00
|
|
|
include DedicatedSidekiqQueue
|
2015-07-17 08:50:03 -04:00
|
|
|
|
2016-10-20 06:59:39 -04:00
|
|
|
LEASE_TIMEOUT = 15.minutes.to_i
|
|
|
|
|
2016-10-25 10:01:24 -04:00
|
|
|
def self.lease_for(project_id)
|
|
|
|
Gitlab::ExclusiveLease.
|
|
|
|
new("project_cache_worker:#{project_id}", timeout: LEASE_TIMEOUT)
|
|
|
|
end
|
|
|
|
|
|
|
|
# Overwrite Sidekiq's implementation so we only schedule when actually needed.
|
|
|
|
def self.perform_async(project_id)
|
|
|
|
# If a lease for this project is still being held there's no point in
|
|
|
|
# scheduling a new job.
|
|
|
|
super unless lease_for(project_id).exists?
|
|
|
|
end
|
|
|
|
|
2015-07-17 08:50:03 -04:00
|
|
|
def perform(project_id)
|
2016-10-20 06:59:39 -04:00
|
|
|
if try_obtain_lease_for(project_id)
|
|
|
|
Rails.logger.
|
|
|
|
info("Obtained ProjectCacheWorker lease for project #{project_id}")
|
|
|
|
else
|
|
|
|
Rails.logger.
|
|
|
|
info("Could not obtain ProjectCacheWorker lease for project #{project_id}")
|
|
|
|
|
|
|
|
return
|
|
|
|
end
|
|
|
|
|
|
|
|
update_caches(project_id)
|
|
|
|
end
|
|
|
|
|
|
|
|
def update_caches(project_id)
|
2015-07-17 09:22:56 -04:00
|
|
|
project = Project.find(project_id)
|
2016-03-27 09:17:49 -04:00
|
|
|
|
|
|
|
return unless project.repository.exists?
|
|
|
|
|
2015-07-17 09:22:56 -04:00
|
|
|
project.update_repository_size
|
|
|
|
project.update_commit_count
|
2015-07-17 10:03:38 -04:00
|
|
|
|
|
|
|
if project.repository.root_ref
|
|
|
|
project.repository.build_cache
|
|
|
|
end
|
2015-07-17 08:50:03 -04:00
|
|
|
end
|
2016-10-20 06:59:39 -04:00
|
|
|
|
|
|
|
def try_obtain_lease_for(project_id)
|
2016-10-25 10:01:24 -04:00
|
|
|
self.class.lease_for(project_id).try_obtain
|
2016-10-20 06:59:39 -04:00
|
|
|
end
|
2015-07-17 08:50:03 -04:00
|
|
|
end
|