add project pull mirroring logic

app/assets/javascripts/importer_status.js

@@ -56,6 +56,8 @@
               if (job.import_status === 'finished') {
                 job_item.removeClass("active").addClass("success");
                 return status_field.html('<span><i class="fa fa-check"></i> done</span>');
+              } else if (job.import_status === 'scheduled') {
+                return status_field.html("<i class='fa fa-spinner fa-spin'></i> scheduled");
               } else if (job.import_status === 'started') {
                 return status_field.html("<i class='fa fa-spinner fa-spin'></i> started");
               } else {

app/controllers/admin/application_settings_controller.rb

@@ -172,10 +172,12 @@ class Admin::ApplicationSettingsController < Admin::ApplicationController
       :elasticsearch_search,
       :repository_size_limit,
       :shared_runners_minutes,
-      :minimum_mirror_sync_time,
       :geo_status_timeout,
       :elasticsearch_experimental_indexer,
-      :check_namespace_plan
+      :check_namespace_plan,
+      :mirror_max_delay,
+      :mirror_max_capacity,
+      :mirror_capacity_threshold
     ]
   end
 end

app/controllers/projects/imports_controller.rb

@@ -12,14 +12,7 @@ class Projects::ImportsController < Projects::ApplicationController
 
   def create
     if @project.update_attributes(import_params)
-      @project.reload
-
-      if @project.import_failed?
-        @project.import_retry
-      else
-        @project.import_start
-        @project.add_import_job
-      end
+      @project.reload.import_schedule
     end
 
     redirect_to namespace_project_import_path(@project.namespace, @project)

app/controllers/projects/mirrors_controller.rb

@@ -14,10 +14,10 @@ class Projects::MirrorsController < Projects::ApplicationController
   def update
     if @project.update_attributes(mirror_params)
       if @project.mirror?
-        @project.update_mirror
+        @project.force_import_job!
 
         flash[:notice] = "Mirroring settings were successfully updated. The project is being updated."
-      elsif @project.mirror_changed?
+      elsif project.previous_changes.has_key?('mirror')
         flash[:notice] = "Mirroring was successfully disabled."
       else
         flash[:notice] = "Mirroring settings were successfully updated."
@@ -34,9 +34,10 @@ class Projects::MirrorsController < Projects::ApplicationController
       @project.update_remote_mirrors
       flash[:notice] = "The remote repository is being updated..."
     else
-      @project.update_mirror
+      @project.force_import_job!
       flash[:notice] = "The repository is being updated..."
     end
+
     redirect_to_repository_settings(@project)
   end
 
@@ -48,7 +49,6 @@ class Projects::MirrorsController < Projects::ApplicationController
 
   def mirror_params
     params.require(:project).permit(:mirror, :import_url, :mirror_user_id,
-                                    :mirror_trigger_builds, :sync_time,
-                                    remote_mirrors_attributes: [:url, :id, :enabled])
+                                    :mirror_trigger_builds, remote_mirrors_attributes: [:url, :id, :enabled])
   end
 end

app/helpers/mirror_helper.rb

@@ -4,10 +4,4 @@ module MirrorHelper
     message << "<br>To discard the local changes and overwrite the branch with the upstream version, delete it here and choose 'Update Now' above." if can?(current_user, :push_code, @project)
     message
   end
-
-  def mirror_sync_time_options
-    Gitlab::Mirror::SYNC_TIME_OPTIONS.select do |key, value|
-      value >= current_application_settings.minimum_mirror_sync_time
-    end
-  end
 end

app/helpers/projects_helper.rb

@@ -432,6 +432,12 @@ module ProjectsHelper
     end
   end
 
+  def can_force_update_mirror?(project)
+    return true unless project.mirror_last_update_at
+
+    Time.now - project.mirror_last_update_at >= 5.minutes
+  end
+
   def membership_locked?
     if @project.group && @project.group.membership_lock
       true

app/models/application_setting.rb

@@ -154,10 +154,6 @@ class ApplicationSetting < ActiveRecord::Base
             presence: true,
             numericality: { greater_than_or_equal_to: 0 }
 
-  validates :minimum_mirror_sync_time,
-            presence: true,
-            inclusion: { in: Gitlab::Mirror::SYNC_TIME_OPTIONS.values }
-
   validates_each :restricted_visibility_levels do |record, attr, value|
     value&.each do |level|
       unless Gitlab::VisibilityLevel.options.has_value?(level)
@@ -186,8 +182,6 @@ class ApplicationSetting < ActiveRecord::Base
   before_save :ensure_runners_registration_token
   before_save :ensure_health_check_access_token
 
-  after_update :update_mirror_cron_job, if: :minimum_mirror_sync_time_changed?
-
   after_commit do
     Rails.cache.write(CACHE_KEY, self)
   end
@@ -218,7 +212,7 @@ class ApplicationSetting < ActiveRecord::Base
     ApplicationSetting.define_attribute_methods
   end
 
-  def self.defaults_ce
+  def self.defaults
     {
       after_sign_up_text: nil,
       akismet_enabled: false,
@@ -269,20 +263,6 @@ class ApplicationSetting < ActiveRecord::Base
     }
   end
 
-  def self.defaults_ee
-    {
-      elasticsearch_url: ENV['ELASTIC_URL'] || 'http://localhost:9200',
-      elasticsearch_aws: false,
-      elasticsearch_aws_region: ENV['ELASTIC_REGION'] || 'us-east-1',
-      minimum_mirror_sync_time: Gitlab::Mirror::FIFTEEN,
-      repository_size_limit: 0
-    }
-  end
-
-  def self.defaults
-    defaults_ce.merge(defaults_ee)
-  end
-
   def self.create_from_defaults
     create(defaults)
   end
@@ -295,13 +275,6 @@ class ApplicationSetting < ActiveRecord::Base
     end
   end
 
-  def update_mirror_cron_job
-    Project.mirror.where('sync_time < ?', minimum_mirror_sync_time)
-      .update_all(sync_time: minimum_mirror_sync_time)
-
-    Gitlab::Mirror.configure_cron_job!
-  end
-
   def elasticsearch_url
     read_attribute(:elasticsearch_url).split(',').map(&:strip)
   end

app/models/ee/application_setting.rb

@@ -7,12 +7,54 @@ module EE
     extend ActiveSupport::Concern
 
     prepended do
+      include IgnorableColumn
+
+      ignore_column :minimum_mirror_sync_time
+
       validates :shared_runners_minutes,
                 numericality: { greater_than_or_equal_to: 0 }
+
+      validates :mirror_max_delay,
+                presence: true,
+                numericality: { allow_nil: true, only_integer: true, greater_than: 0 }
+
+      validates :mirror_max_capacity,
+                presence: true,
+                numericality: { allow_nil: true, only_integer: true, greater_than: 0 }
+
+      validates :mirror_capacity_threshold,
+                presence: true,
+                numericality: { allow_nil: true, only_integer: true, greater_than: 0 }
+
+      validate :mirror_capacity_threshold_less_than
+    end
+
+    module ClassMethods
+      def defaults
+        super.merge(
+          elasticsearch_url: ENV['ELASTIC_URL'] || 'http://localhost:9200',
+          elasticsearch_aws: false,
+          elasticsearch_aws_region: ENV['ELASTIC_REGION'] || 'us-east-1',
+          repository_size_limit: 0,
+          mirror_max_delay: Settings.gitlab['mirror_max_delay'],
+          mirror_max_capacity: Settings.gitlab['mirror_max_capacity'],
+          mirror_capacity_threshold: Settings.gitlab['mirror_capacity_threshold']
+        )
+      end
     end
 
     def should_check_namespace_plan?
       check_namespace_plan? && (::Gitlab.com? || Rails.env.development?)
     end
+
+    private
+
+    def mirror_capacity_threshold_less_than
+      return unless mirror_max_capacity && mirror_capacity_threshold
+
+      if mirror_capacity_threshold > mirror_max_capacity
+        errors.add(:mirror_capacity_threshold, "Project's mirror capacity threshold can't be higher than it's maximum capacity")
+      end
+    end
   end
 end

app/models/ee/project.rb

@@ -7,8 +7,28 @@ module EE
     extend ActiveSupport::Concern
 
     prepended do
+      include IgnorableColumn
+
+      ignore_column :sync_time
+
+      after_save :create_mirror_data, if: ->(project) { project.mirror? && project.mirror_changed? }
+      after_save :destroy_mirror_data, if: ->(project) { !project.mirror? && project.mirror_changed? }
+
+      has_one :mirror_data, dependent: :delete, autosave: true, class_name: 'ProjectMirrorData'
+
       scope :with_shared_runners_limit_enabled, -> { with_shared_runners.non_public_only }
 
+      scope :mirrors_to_sync, -> do
+        mirror.joins(:mirror_data).where("next_execution_timestamp <= ? AND import_status NOT IN ('scheduled', 'started')", Time.now).
+          order_by(:next_execution_timestamp).limit(::Gitlab::Mirror.available_capacity)
+      end
+
+      scope :stuck_mirrors, -> do
+        mirror.joins(:mirror_data).
+          where("(import_status = 'started' AND project_mirror_data.last_update_started_at < :limit) OR (import_status = 'scheduled' AND project_mirror_data.last_update_scheduled_at < :limit)",
+                { limit: 20.minutes.ago })
+      end
+
       delegate :shared_runners_minutes, :shared_runners_seconds, :shared_runners_seconds_last_reset,
         to: :statistics, allow_nil: true
 
@@ -44,6 +64,19 @@ module EE
       config.address&.gsub(wildcard, full_path)
     end
 
+    def force_import_job!
+      self.mirror_data.set_next_execution_to_now!
+      UpdateAllMirrorsWorker.perform_async
+    end
+
+    def add_import_job
+      if import? && !repository_exists?
+        super
+      elsif mirror?
+        RepositoryUpdateMirrorWorker.perform_async(self.id)
+      end
+    end
+
     private
 
     def licensed_feature_available?(feature)
@@ -57,6 +90,10 @@ module EE
       end
     end
 
+    def destroy_mirror_data
+      mirror_data.destroy
+    end
+
     def service_desk_available?
       return @service_desk_available if defined?(@service_desk_available)
 

app/models/project.rb

@@ -170,7 +170,7 @@ class Project < ActiveRecord::Base
   has_many :audit_events, as: :entity, dependent: :destroy
   has_many :notification_settings, dependent: :destroy, as: :source
 
-  has_one :import_data, dependent: :destroy, class_name: "ProjectImportData"
+  has_one :import_data, dependent: :delete, class_name: 'ProjectImportData'
   has_one :project_feature, dependent: :destroy
   has_one :statistics, class_name: 'ProjectStatistics', dependent: :delete
   has_many :container_repositories, dependent: :destroy
@@ -239,10 +239,6 @@ class Project < ActiveRecord::Base
   validates :repository_size_limit,
             numericality: { only_integer: true, greater_than_or_equal_to: 0, allow_nil: true }
 
-  validates :sync_time,
-    presence: true,
-    inclusion: { in: Gitlab::Mirror::SYNC_TIME_OPTIONS.values }
-
   with_options if: :mirror? do |project|
     project.validates :import_url, presence: true
     project.validates :mirror_user, presence: true
@@ -272,7 +268,6 @@ class Project < ActiveRecord::Base
   scope :for_milestones, ->(ids) { joins(:milestones).where('milestones.id' => ids).distinct }
   scope :with_push, -> { joins(:events).where('events.action = ?', Event::PUSHED) }
   scope :with_remote_mirrors, -> { joins(:remote_mirrors).where(remote_mirrors: { enabled: true }).distinct }
-
   scope :with_project_feature, -> { joins('LEFT JOIN project_features ON projects.id = project_features.project_id') }
   scope :with_statistics, -> { includes(:statistics) }
   scope :with_shared_runners, -> { where(shared_runners_enabled: true) }
@@ -323,8 +318,16 @@ class Project < ActiveRecord::Base
   scope :excluding_project, ->(project) { where.not(id: project) }
 
   state_machine :import_status, initial: :none do
+    event :import_schedule do
+      transition [:none, :finished, :failed] => :scheduled
+    end
+
+    event :force_import_start do
+      transition [:none, :finished, :failed] => :started
+    end
+
     event :import_start do
-      transition [:none, :finished] => :started
+      transition scheduled: :started
     end
 
     event :import_finish do
@@ -332,24 +335,57 @@ class Project < ActiveRecord::Base
     end
 
     event :import_fail do
-      transition started: :failed
-    end
-
-    event :import_retry do
-      transition failed: :started
+      transition [:scheduled, :started] => :failed
     end
 
+    state :scheduled
     state :started
     state :finished
     state :failed
 
-    after_transition any => :finished, do: :reset_cache_and_import_attrs
+    before_transition [:none, :finished, :failed] => :scheduled do |project, _|
+      project.mirror_data&.last_update_scheduled_at = Time.now
+    end
+
+    after_transition [:none, :finished, :failed] => :scheduled do |project, _|
+      project.run_after_commit { add_import_job }
+    end
+
+    before_transition scheduled: :started do |project, _|
+      project.mirror_data&.last_update_started_at = Time.now
+    end
+
+    before_transition scheduled: :failed do |project, _|
+      if project.mirror?
+        timestamp = Time.now
+        project.mirror_last_update_at = timestamp
+        project.mirror_data.next_execution_timestamp = timestamp
+      end
+    end
+
+    after_transition [:scheduled, :started] => [:finished, :failed] do |project, _|
+      Gitlab::Mirror.decrement_capacity(project.id) if project.mirror?
+    end
+
+    before_transition started: :failed do |project, _|
+      if project.mirror?
+        project.mirror_last_update_at = Time.now
+
+        mirror_data = project.mirror_data
+        mirror_data.increment_retry_count!
+        mirror_data.set_next_execution_timestamp!
+      end
+    end
 
-    before_transition started: :finished do |project, transaction|
+    before_transition started: :finished do |project, _|
       if project.mirror?
-        timestamp = DateTime.now
+        timestamp = Time.now
         project.mirror_last_update_at = timestamp
         project.mirror_last_successful_update_at = timestamp
+
+        mirror_data = project.mirror_data
+        mirror_data.reset_retry_count!
+        mirror_data.set_next_execution_timestamp!
       end
 
       if current_application_settings.elasticsearch_indexing?
@@ -357,8 +393,10 @@ class Project < ActiveRecord::Base
       end
     end
 
-    before_transition started: :failed do |project, transaction|
-      project.mirror_last_update_at = DateTime.now if project.mirror?
+    after_transition started: :finished, do: :reset_cache_and_import_attrs
+
+    after_transition [:finished, :failed] => [:scheduled, :started] do |project, _|
+      Gitlab::Mirror.increment_capacity(project.id) if project.mirror?
     end
   end
 
@@ -577,9 +615,17 @@ class Project < ActiveRecord::Base
   end
 
   def import_in_progress?
+    import_started? || import_scheduled?
+  end
+
+  def import_started?
     import? && import_status == 'started'
   end
 
+  def import_scheduled?
+    import_status == 'scheduled'
+  end
+
   def import_failed?
     import_status == 'failed'
   end
@@ -597,7 +643,10 @@ class Project < ActiveRecord::Base
   end
 
   def updating_mirror?
-    mirror? && import_in_progress? && !empty_repo?
+    return false unless mirror? && !empty_repo?
+    return true if import_in_progress?
+
+    self.mirror_data.next_execution_timestamp < Time.now
   end
 
   def mirror_last_update_status
@@ -622,20 +671,6 @@ class Project < ActiveRecord::Base
     mirror_updated? && self.mirror_last_successful_update_at
   end
 
-  def update_mirror
-    return unless mirror? && repository_exists?
-
-    return if import_in_progress?
-
-    if import_failed?
-      import_retry
-    else
-      import_start
-    end
-
-    RepositoryUpdateMirrorWorker.perform_async(self.id)
-  end
-
   def has_remote_mirror?
     remote_mirrors.enabled.exists?
   end

app/models/project_mirror_data.rb

+class ProjectMirrorData < ActiveRecord::Base
+  include Gitlab::CurrentSettings
+
+  BACKOFF_PERIOD = 24.seconds
+  JITTER = 6.seconds
+
+  belongs_to :project
+
+  validates :project, presence: true
+  validates :next_execution_timestamp, presence: true
+
+  before_validation on: :create do
+    self.next_execution_timestamp = Time.now
+  end
+
+  def reset_retry_count!
+    self.retry_count = 0
+  end
+
+  def increment_retry_count!
+    self.retry_count += 1
+  end
+
+  # We schedule the next sync time based on the duration of the
+  # last mirroring period and add it a fixed backoff period with a random jitter
+  def set_next_execution_timestamp!
+    timestamp = Time.now
+    retry_factor = [1, self.retry_count].max
+    delay = [base_delay(timestamp) * retry_factor, Gitlab::Mirror.max_delay].min
+
+    self.next_execution_timestamp = timestamp + delay
+  end
+
+  def set_next_execution_to_now!
+    self.update_attributes(next_execution_timestamp: Time.now)
+  end
+
+  private
+
+  def base_delay(timestamp)
+    duration = timestamp - self.last_update_started_at
+
+    (BACKOFF_PERIOD + rand(JITTER)) * duration.seconds
+  end
+end

app/services/projects/create_service.rb

@@ -51,15 +51,14 @@ module Projects
 
       save_project_and_import_data(import_data)
 
-      @project.import_start if @project.import?
-
       after_create_actions if @project.persisted?
 
       if @project.errors.empty?
-        @project.add_import_job if @project.import?
+        @project.import_schedule if @project.import?
       else
         fail(error: @project.errors.full_messages.join(', '))
       end
+
       @project
     rescue ActiveRecord::RecordInvalid => e
       message = "Unable to save #{e.record.type}: #{e.record.errors.full_messages.join(", ")} "

app/views/admin/application_settings/_form.html.haml

@@ -71,10 +71,6 @@
         %span.help-block#repository_size_limit_help_block
           Includes LFS objects. It can be overridden per group, or per project. 0 for unlimited.
           = link_to icon('question-circle'), help_page_path("user/admin_area/settings/account_and_limit_settings")
-    .form-group
-      = f.label :minimum_mirror_sync_time, class: 'control-label col-sm-2'
-      .col-sm-10
-        = f.select :minimum_mirror_sync_time, options_for_select(Gitlab::Mirror::SYNC_TIME_OPTIONS, @application_setting.minimum_mirror_sync_time), {}, class: 'form-control'
     .form-group
       = f.label :session_expire_delay, 'Session duration (minutes)', class: 'control-label col-sm-2'
       .col-sm-10
@@ -105,6 +101,8 @@
               Enabling this will only make licensed EE features available to projects if the project namespace's plan
               includes the feature or if the project is public.
 
+  = render partial: 'repository_mirrors_form', locals: { f: f }
+
   %fieldset
     %legend Sign-up Restrictions
     .form-group

app/views/admin/application_settings/_repository_mirrors_form.html.haml

+%fieldset
+  %legend Repository mirror settings
+  .form-group
+    = f.label :mirror_max_delay, class: 'control-label col-sm-2' do
+      Maximum delay (Hours)
+    .col-sm-10
+      = f.number_field :mirror_max_delay, class: 'form-control', min: 0
+      %span.help-block#mirror_max_delay_help_block
+      Maximum time between updates that a mirror can have when scheduled to synchronize.
+  .form-group
+    = f.label :mirror_max_capacity, class: 'control-label col-sm-2' do
+      Maximum capacity
+    .col-sm-10
+      = f.number_field :mirror_max_capacity, class: 'form-control', min: 0
+      %span.help-block#mirror_max_capacity_help_block
+      Maximum number of mirrors that can be synchronizing at the same time.
+  .form-group
+    = f.label :mirror_capacity_threshold, class: 'control-label col-sm-2' do
+      Capacity threshold
+    .col-sm-10
+      = f.number_field :mirror_capacity_threshold, class: 'form-control', min: 0
+      %span.help-block#mirror_capacity_threshold
+      Minimum capacity to be available before we schedule more mirrors preemptively.

app/views/projects/mirrors/_show.html.haml

@@ -43,9 +43,6 @@
           They need to have at least master access to this project.
       - if @project.builds_enabled?
         = render "shared/mirror_trigger_builds_setting", f: f
-      .form-group
-        = f.label :sync_time, "Synchronization time", class: "label-light append-bottom-0"
-        = f.select :sync_time, options_for_select(mirror_sync_time_options, @project.sync_time), {}, class: 'form-control project-mirror-sync-time'
     .col-sm-12
       %hr
     .col-lg-3

app/views/shared/_mirror_update_button.html.haml

@@ -4,8 +4,13 @@
       %span.btn.disabled
         = icon("refresh spin")
         Updating…
+    - elsif !can_force_update_mirror?(@project)
+      %span.btn.disabled{ data: { toggle: 'tooltip', placement: 'auto top' }, style: 'cursor: default',
+      title: 'You can only force update once every five minutes.' }
+        = icon("refresh")
+        Update Now
     - else
-      = link_to update_now_namespace_project_mirror_path(@project.namespace, @project), method: :post, class: "btn" do
+      = link_to update_now_namespace_project_mirror_path(@project.namespace, @project), method: :post, class: 'btn' do
         = icon("refresh")
         Update Now
     - if @project.mirror_last_update_success?

app/workers/repository_fork_worker.rb

 class RepositoryForkWorker
+  ForkError = Class.new(StandardError)
+
   include Sidekiq::Worker
   include Gitlab::ShellAdapter
   include DedicatedSidekiqQueue
@@ -8,29 +10,31 @@ class RepositoryForkWorker
                               source_path: source_path,
                               target_path: target_path)
 
-    project = Project.find_by_id(project_id)
-
-    unless project.present?
-      logger.error("Project #{project_id} no longer exists!")
-      return
-    end
+    project = Project.find(project_id)
+    project.import_start
 
     result = gitlab_shell.fork_repository(forked_from_repository_storage_path, source_path,
                                           project.repository_storage_path, target_path)
-    unless result
-      logger.error("Unable to fork project #{project_id} for repository #{source_path} -> #{target_path}")
-      project.mark_import_as_failed('The project could not be forked.')
-      return
-    end
+    raise ForkError, "Unable to fork project #{project_id} for repository #{source_path} -> #{target_path}" unless result
 
     project.repository.after_import
-
-    unless project.valid_repo?
-      logger.error("Project #{project_id} had an invalid repository after fork")
-      project.mark_import_as_failed('The forked repository is invalid.')
-      return
-    end
+    raise ForkError, "Project #{project_id} had an invalid repository after fork" unless project.valid_repo?
 
     project.import_finish
+  rescue ForkError => ex
+    fail_fork(project, ex.message)
+    raise
+  rescue => ex
+    return unless project
+
+    fail_fork(project, ex.message)
+    raise ForkError, "#{ex.class} #{ex.message}"
+  end
+
+  private
+
+  def fail_fork(project, message)
+    Rails.logger.error(message)
+    project.mark_import_as_failed(message)
   end
 end

app/workers/repository_import_worker.rb

 class RepositoryImportWorker
+  ImportError = Class.new(StandardError)
+
   include Sidekiq::Worker
   include DedicatedSidekiqQueue
 
@@ -10,6 +12,8 @@ class RepositoryImportWorker
     @project = Project.find(project_id)
     @current_user = @project.creator
 
+    project.import_start
+
     Gitlab::Metrics.add_event(:import_repository,
                               import_url: @project.import_url,
                               path: @project.path_with_namespace)
@@ -17,16 +21,27 @@ class RepositoryImportWorker
     project.update_columns(import_jid: self.jid, import_error: nil)
 
     result = Projects::ImportService.new(project, current_user).execute
-
-    if result[:status] == :error
-      project.mark_import_as_failed(result[:message])
-      return
-    end
+    raise ImportError, result[:message] if result[:status] == :error
 
     project.repository.after_import
     project.import_finish
 
-    # Explicitly update mirror so that upstream remote is created and fetched
-    project.update_mirror
+    # Explicitly schedule mirror for update so
+    # that upstream remote is created and fetched
+    project.import_schedule if project.mirror?
+  rescue ImportError => ex
+    fail_import(project, ex.message)
+    raise
+  rescue => ex
+    return unless project
+
+    fail_import(project, ex.message)
+    raise ImportError, "#{ex.class} #{ex.message}"
+  end
+
+  private
+
+  def fail_import(project, message)
+    project.mark_import_as_failed(message)
   end
 end

app/workers/repository_update_mirror_dispatch_worker.rb

-class RepositoryUpdateMirrorDispatchWorker
-  include Sidekiq::Worker
-
-  LEASE_TIMEOUT = 5.minutes
-
-  sidekiq_options queue: :project_mirror
-
-  attr_accessor :project, :repository, :current_user
-
-  def perform(project_id)
-    return unless try_obtain_lease(project_id)
-
-    @project = Project.find_by_id(project_id)
-    return unless project
-
-    project.update_mirror
-  end
-
-  private
-
-  def try_obtain_lease(project_id)
-    # Using 5 minutes timeout based on the 95th percent of timings (currently max of 25 minutes)
-    lease = ::Gitlab::ExclusiveLease.new("repository_update_mirror_dispatcher:#{project_id}", timeout: LEASE_TIMEOUT)
-    lease.try_obtain
-  end
-end

app/workers/repository_update_mirror_worker.rb

 class RepositoryUpdateMirrorWorker
-  UpdateMirrorError = Class.new(StandardError)
+  UpdateError = Class.new(StandardError)
+  UpdateAlreadyInProgressError = Class.new(StandardError)
 
   include Sidekiq::Worker
   include Gitlab::ShellAdapter
@@ -10,25 +11,35 @@ class RepositoryUpdateMirrorWorker
   attr_accessor :project, :repository, :current_user
 
   def perform(project_id)
-    begin
-      project = Project.find(project_id)
-
-      return unless project
-
-      @current_user = project.mirror_user || project.creator
-
-      result = Projects::UpdateMirrorService.new(project, @current_user).execute
-      if result[:status] == :error
-        project.mark_import_as_failed(result[:message])
-        return
-      end
-
-      project.import_finish
-    rescue => ex
-      if project
-        project.mark_import_as_failed("We're sorry, a temporary error occurred, please try again.")
-        raise UpdateMirrorError, "#{ex.class}: #{Gitlab::UrlSanitizer.sanitize(ex.message)}"
-      end
-    end
+    project = Project.find(project_id)
+
+    raise UpdateAlreadyInProgressError if project.import_started?
+    project.import_start
+
+    @current_user = project.mirror_user || project.creator
+
+    result = Projects::UpdateMirrorService.new(project, @current_user).execute
+    raise UpdateError, result[:message] if result[:status] == :error
+
+    project.import_finish
+  rescue UpdateAlreadyInProgressError
+    raise
+  rescue UpdateError => ex
+    fail_mirror(project, ex.message)
+    raise
+  rescue => ex
+    return unless project
+
+    fail_mirror(project, ex.message)
+    raise UpdateError, "#{ex.class}: #{ex.message}"
+  ensure
+    UpdateAllMirrorsWorker.perform_async if Gitlab::Mirror.threshold_reached?
+  end
+
+  private
+
+  def fail_mirror(project, message)
+    Rails.logger.error(message)
+    project.mark_import_as_failed(message)
   end
 end

app/workers/repository_update_remote_mirror_worker.rb

@@ -23,15 +23,25 @@ class RepositoryUpdateRemoteMirrorWorker
     project = remote_mirror.project
     current_user = project.creator
     result = Projects::UpdateRemoteMirrorService.new(project, current_user).execute(remote_mirror)
-
     raise UpdateError, result[:message] if result[:status] == :error
+
     remote_mirror.update_finish
   rescue UpdateAlreadyInProgressError
     raise
   rescue UpdateError => ex
-    remote_mirror.mark_as_failed(Gitlab::UrlSanitizer.sanitize(ex.message))
+    fail_remote_mirror(remote_mirror, ex.message)
     raise
   rescue => ex
+    return unless remote_mirror
+
+    fail_remote_mirror(remote_mirror, ex.message)
     raise UpdateError, "#{ex.class}: #{ex.message}"
   end
+
+  private
+
+  def fail_remote_mirror(remote_mirror, message)
+    Rails.logger.error(message)
+    remote_mirror.mark_as_failed(message)
+  end
 end

app/workers/update_all_mirrors_worker.rb

@@ -2,39 +2,38 @@ class UpdateAllMirrorsWorker
   include Sidekiq::Worker
   include CronjobQueue
 
-  LEASE_TIMEOUT = 840
+  LEASE_TIMEOUT = 5.minutes
+  LEASE_KEY = 'update_all_mirrors'.freeze
 
   def perform
     # This worker requires updating the database state, which we can't
     # do on a Geo secondary
     return if Gitlab::Geo.secondary?
-    return unless try_obtain_lease
+
+    lease_uuid = try_obtain_lease
+    return unless lease_uuid
 
     fail_stuck_mirrors!
 
-    mirrors_to_sync.find_each(batch_size: 200) do |project|
-      RepositoryUpdateMirrorDispatchWorker.perform_in(rand((project.sync_time / 2).minutes), project.id)
-    end
+    return if Gitlab::Mirror.max_mirror_capacity_reached?
+    Project.mirrors_to_sync.find_each(batch_size: 200, &:import_schedule)
+
+    cancel_lease(lease_uuid)
   end
 
   def fail_stuck_mirrors!
-    stuck = Project.mirror
-      .with_import_status(:started)
-      .where('mirror_last_update_at < ?', 2.hours.ago)
-
-    stuck.find_each(batch_size: 50) do |project|
+    Project.stuck_mirrors.find_each(batch_size: 50) do |project|
       project.mark_import_as_failed('The mirror update took too long to complete.')
     end
   end
 
   private
 
-  def mirrors_to_sync
-    Project.mirror.where("mirror_last_successful_update_at + #{Gitlab::Database.minute_interval('sync_time')} <= ? OR sync_time IN (?)", DateTime.now, Gitlab::Mirror.sync_times)
+  def try_obtain_lease
+    ::Gitlab::ExclusiveLease.new(LEASE_KEY, timeout: LEASE_TIMEOUT).try_obtain
   end
 
-  def try_obtain_lease
-    lease = ::Gitlab::ExclusiveLease.new("update_all_mirrors", timeout: LEASE_TIMEOUT)
-    lease.try_obtain
+  def cancel_lease(uuid)
+    ::Gitlab::ExclusiveLease.cancel(LEASE_KEY, uuid)
   end
 end

changelogs/unreleased-ee/move-to-adaptive-scheduling-in-project-mirror-pull.yml

+---
+title: Move pull mirroring to adaptive scheduling
+merge_request: 1853
+author:

config/initializers/1_settings.rb

@@ -264,6 +264,9 @@ Settings.gitlab['default_projects_features'] ||= {}
 Settings.gitlab['webhook_timeout'] ||= 10
 Settings.gitlab['max_attachment_size'] ||= 10
 Settings.gitlab['session_expire_delay'] ||= 10080
+Settings.gitlab['mirror_max_delay'] ||= 5
+Settings.gitlab['mirror_max_capacity'] ||= 30
+Settings.gitlab['mirror_capacity_threshold'] ||= 15
 Settings.gitlab.default_projects_features['issues']             = true if Settings.gitlab.default_projects_features['issues'].nil?
 Settings.gitlab.default_projects_features['merge_requests']     = true if Settings.gitlab.default_projects_features['merge_requests'].nil?
 Settings.gitlab.default_projects_features['wiki']               = true if Settings.gitlab.default_projects_features['wiki'].nil?

doc/workflow/repository_mirroring.md

@@ -51,8 +51,8 @@ whether you want to trigger builds for mirror updates.
 Since the repository on GitLab functions as a mirror of the upstream repository,
 you are advised not to push commits directly to the repository on GitLab.
 Instead, any commits should be pushed to the upstream repository, and will end
-up in the GitLab repository automatically within your project's configured
-synchronization time, or when a [forced update](#forcing-an-update) is initiated.
+up in the GitLab repository automatically within a certain period of time
+or when a [forced update](#forcing-an-update) is initiated.
 
 If you do manually update a branch in the GitLab repository, the branch will
 become diverged from upstream, and GitLab will no longer automatically update
@@ -83,7 +83,7 @@ In case of a diverged branch, you will see an error indicated at the
 
 ## Forcing an update
 
-While mirrors update at a pre-configured time (hourly by default), you can always force an update (either **push** or
+While mirrors are scheduled to update automatically, you can always force an update (either **push** or
 **pull**) by using the **Update now** button which is exposed in various places:
 
 - in the commits page
@@ -91,24 +91,13 @@ While mirrors update at a pre-configured time (hourly by default), you can alway
 - in the tags page
 - in the **Mirror repository** settings page
 
-## Adjusting synchronization times
-
-Your repository's default synchronization time is hourly.
-However, you can adjust it by visiting the **Mirror repository** page
-under the wheel icon in the upper right corner.
-Check the Synchronization time section where you can choose to have your mirror
-be updated once every fifteen minutes, hourly or daily and then hit **Save changes**
-at the bottom.
-
 ## Using both mirroring methods at the same time
 
 Currently there is no bidirectional support without conflicts. That means that
 if you configure a repository to both pull and push to a second one, there is
-no guarantee that it will update correctly on both remotes. You could
-adjust the synchronization times on the mirror settings page
-to a very low value and hope that no conflicts occur during
-the pull/push window time, but that is not a solution to consider on a
-production environment. Another thing you could try is [configuring custom Git hooks][hooks] on the GitLab server.
+no guarantee that it will update correctly on both remotes.
+You can try [configuring custom Git hooks][hooks] on the GitLab server in order
+to resolve this issue.
 
 
 [ee-51]: https://gitlab.com/gitlab-org/gitlab-ee/merge_requests/51

lib/gitlab/mirror.rb

@@ -2,85 +2,66 @@ module Gitlab
   module Mirror
     include Gitlab::CurrentSettings
 
-    FIFTEEN = 15
-    HOURLY  = 60
-    THREE   = 180
-    SIX     = 360
-    TWELVE  = 720
-    DAILY   = 1440
-
-    INTERVAL_BEFORE_FIFTEEN = 14.minutes
-
-    SYNC_TIME_TO_CRON = {
-      FIFTEEN => "*/15 * * * *",
-      HOURLY  => "0 * * * *",
-      THREE   => "0 */3 * * *",
-      SIX     => "0 */6 * * *",
-      TWELVE  => "0 */12 * * *",
-      DAILY   => "0 0 * * *"
-    }.freeze
-
-    SYNC_TIME_OPTIONS = {
-      "Update every 15 minutes"   => FIFTEEN,
-      "Update hourly"             => HOURLY,
-      "Update every three hours"  => THREE,
-      "Update every six hours"    => SIX,
-      "Update every twelve hours" => TWELVE,
-      "Update every day"          => DAILY
-    }.freeze
+    # Runs scheduler every minute
+    SCHEDULER_CRON = '* * * * *'.freeze
+    PULL_CAPACITY_KEY = 'MIRROR_PULL_CAPACITY'.freeze
+    UPPER_JITTER = 1.minute
 
     class << self
-      def sync_times
-        sync_times = [FIFTEEN]
-        sync_times << DAILY  if at_beginning_of_day?
-        sync_times << TWELVE if at_beginning_of_hour?(12)
-        sync_times << SIX    if at_beginning_of_hour?(6)
-        sync_times << THREE  if at_beginning_of_hour?(3)
-        sync_times << HOURLY if at_beginning_of_hour?
+      def configure_cron_job!
+        destroy_cron_job!
+        return if Gitlab::Geo.secondary?
 
-        sync_times
+        Sidekiq::Cron::Job.create(
+          name: 'update_all_mirrors_worker',
+          cron: SCHEDULER_CRON,
+          class: 'UpdateAllMirrorsWorker'
+        )
       end
 
-      def update_all_mirrors_cron_job
-        Sidekiq::Cron::Job.find("update_all_mirrors_worker")
+      def max_mirror_capacity_reached?
+        available_capacity <= 0
       end
 
-      def destroy_cron_job!
-        update_all_mirrors_cron_job&.destroy
+      def threshold_reached?
+        available_capacity >= capacity_threshold
       end
 
-      def configure_cron_job!
-        if Gitlab::Geo.secondary?
-          destroy_cron_job!
-          return
-        end
+      def available_capacity
+        current_capacity = Gitlab::Redis.with { |redis| redis.scard(PULL_CAPACITY_KEY) }
 
-        minimum_mirror_sync_time = current_application_settings.minimum_mirror_sync_time rescue FIFTEEN
-        sync_time = SYNC_TIME_TO_CRON[minimum_mirror_sync_time]
-        destroy_cron_job!
+        max_capacity - current_capacity.to_i
+      end
 
-        Sidekiq::Cron::Job.create(
-          name: 'update_all_mirrors_worker',
-          cron: sync_time,
-          class: 'UpdateAllMirrorsWorker'
-        )
+      def increment_capacity(project_id)
+        Gitlab::Redis.with { |redis| redis.sadd(PULL_CAPACITY_KEY, project_id) }
       end
 
-      def at_beginning_of_day?
-        start_at = DateTime.now.at_beginning_of_day
-        end_at = start_at + INTERVAL_BEFORE_FIFTEEN
+      # We do not want negative capacity
+      def decrement_capacity(project_id)
+        Gitlab::Redis.with { |redis| redis.srem(PULL_CAPACITY_KEY, project_id) }
+      end
 
-        DateTime.now.between?(start_at, end_at)
+      def max_delay
+        current_application_settings.mirror_max_delay.hours + rand(UPPER_JITTER)
       end
 
-      def at_beginning_of_hour?(hour_mark = nil)
-        start_at = DateTime.now.at_beginning_of_hour
-        end_at = start_at + INTERVAL_BEFORE_FIFTEEN
+      def max_capacity
+        current_application_settings.mirror_max_capacity
+      end
 
-        between_interval = DateTime.now.between?(start_at, end_at)
-        return between_interval unless hour_mark
+      def capacity_threshold
+        current_application_settings.mirror_capacity_threshold
+      end
 
-        between_interval && DateTime.now.hour % hour_mark == 0
+      private
+
+      def update_all_mirrors_cron_job
+        Sidekiq::Cron::Job.find("update_all_mirrors_worker")
+      end
+
+      def destroy_cron_job!
+        update_all_mirrors_cron_job&.destroy
       end
     end
   end

lib/tasks/import.rake

@@ -37,7 +37,7 @@ class GithubImport
   end
 
   def import!
-    @project.import_start
+    @project.force_import_start
 
     timings = Benchmark.measure do
       Github::Import.new(@project, @options).execute

spec/controllers/projects/mirrors_controller_spec.rb

 require 'spec_helper'
 
 describe Projects::MirrorsController do
-  let(:sync_times) { Gitlab::Mirror::SYNC_TIME_OPTIONS.values }
-
-  describe 'setting up a mirror' do
-    context 'when the current project is a mirror' do
-      before do
-        @project = create(:project, :mirror)
-        sign_in(@project.owner)
-      end
-
-      context 'sync_time update' do
-        it 'allows sync_time update with valid time' do
-          sync_times.each do |sync_time|
-            expect do
-              do_put(@project, sync_time: sync_time)
-            end.to change { Project.mirror.where(sync_time: sync_time).count }.by(1)
-          end
-        end
-
-        it 'fails to update sync_time with invalid time' do
-          expect do
-            do_put(@project, sync_time: 1000)
-          end.not_to change { @project.sync_time }
-        end
-      end
-    end
-  end
-
   describe 'setting up a remote mirror' do
     context 'when the current project is a mirror' do
+      let(:project) { create(:project, :mirror) }
+
       before do
-        @project = create(:project, :mirror)
-        sign_in(@project.owner)
+        sign_in(project.owner)
       end
 
       it 'allows to create a remote mirror' do
+        expect_any_instance_of(EE::Project).to receive(:force_import_job!)
+
         expect do
-          do_put(@project, remote_mirrors_attributes: { '0' => { 'enabled' => 1, 'url' => 'http://foo.com' } })
+          do_put(project, remote_mirrors_attributes: { '0' => { 'enabled' => 1, 'url' => 'http://foo.com' } })
         end.to change { RemoteMirror.count }.to(1)
       end
 
       context 'when remote mirror has the same URL' do
         it 'does not allow to create the remote mirror' do
           expect do
-            do_put(@project, remote_mirrors_attributes: { '0' => { 'enabled' => 1, 'url' => @project.import_url } })
+            do_put(project, remote_mirrors_attributes: { '0' => { 'enabled' => 1, 'url' => project.import_url } })
           end.not_to change { RemoteMirror.count }
         end
 
         context 'with disabled local mirror' do
           it 'allows to create a remote mirror' do
             expect do
-              do_put(@project, mirror: 0, remote_mirrors_attributes: { '0' => { 'enabled' => 1, 'url' => @project.import_url } })
+              do_put(project, mirror: 0, remote_mirrors_attributes: { '0' => { 'enabled' => 1, 'url' => project.import_url } })
             end.to change { RemoteMirror.count }.to(1)
           end
         end
@@ -70,32 +46,72 @@ describe Projects::MirrorsController do
     end
 
     context 'when the current project has a remote mirror' do
+      let(:project) { create(:project) }
+      let(:remote_mirror) { project.remote_mirrors.create!(enabled: 1, url: 'http://local.dev') }
+
       before do
-        @project = create(:project)
-        @remote_mirror = @project.remote_mirrors.create!(enabled: 1, url: 'http://local.dev')
-        sign_in(@project.owner)
+        sign_in(project.owner)
       end
 
       context 'when trying to create a mirror with the same URL' do
         it 'should not setup the mirror' do
-          do_put(@project, mirror: true, import_url: @remote_mirror.url)
+          do_put(project, mirror: true, import_url: remote_mirror.url)
 
-          expect(@project.reload.mirror).to be_falsey
-          expect(@project.reload.import_url).to be_blank
+          expect(project.reload.mirror).to be_falsey
+          expect(project.reload.import_url).to be_blank
         end
       end
 
       context 'when trying to create a mirror with a different URL' do
         it 'should setup the mirror' do
-          do_put(@project, mirror: true, mirror_user_id: @project.owner.id, import_url: 'http://test.com')
+          expect_any_instance_of(EE::Project).to receive(:force_import_job!)
+
+          do_put(project, mirror: true, mirror_user_id: project.owner.id, import_url: 'http://local.dev')
 
-          expect(@project.reload.mirror).to eq(true)
-          expect(@project.reload.import_url).to eq('http://test.com')
+          expect(project.reload.mirror).to eq(true)
+          expect(project.reload.import_url).to eq('http://local.dev')
         end
       end
     end
   end
 
+  describe 'setting up a mirror' do
+    before do
+      sign_in(project.owner)
+    end
+
+    context 'when project does not have a mirror' do
+      let(:project) { create(:project) }
+
+      it 'allows to create a mirror' do
+        expect_any_instance_of(EE::Project).to receive(:force_import_job!)
+
+        expect do
+          do_put(project, mirror: true, mirror_user_id: project.owner.id, import_url: 'http://foo.com')
+        end.to change { Project.mirror.count }.to(1)
+      end
+    end
+
+    context 'when project has a mirror' do
+      let(:project) { create(:project, :mirror, :import_finished) }
+
+      it 'is able to disable the mirror' do
+        expect { do_put(project, mirror: false) }.to change { Project.mirror.count }.to(0)
+      end
+    end
+  end
+
+  describe 'forcing an update' do
+    it 'forces update' do
+      expect_any_instance_of(EE::Project).to receive(:force_import_job!)
+
+      project = create(:project, :mirror)
+      sign_in(project.owner)
+
+      put :update_now, { namespace_id: project.namespace.to_param, project_id: project.to_param }
+    end
+  end
+
   def do_put(project, options)
     attrs = { namespace_id: project.namespace.to_param, project_id: project.to_param }
     attrs[:project] = options

spec/factories/forked_project_links.rb

@@ -7,5 +7,9 @@ FactoryGirl.define do
       link.forked_from_project.reload
       link.forked_to_project.reload
     end
+
+    trait :forked_to_empty_project do
+      association :forked_to_project, factory: :empty_project
+    end
   end
 end

spec/factories/projects.rb

@@ -24,20 +24,25 @@ FactoryGirl.define do
       visibility_level Gitlab::VisibilityLevel::PRIVATE
     end
 
+    trait :import_scheduled do
+      import_status :scheduled
+    end
+
     trait :import_started do
-      import_url { generate(:url) }
       import_status :started
     end
 
     trait :import_finished do
-      import_started
       import_status :finished
     end
 
-    trait :mirror do
-      import_started
+    trait :import_failed do
+      import_status :failed
+    end
 
+    trait :mirror do
       mirror true
+      import_url { generate(:url) }
       mirror_user_id { creator_id }
     end
 

spec/features/merge_requests/diffs_spec.rb

@@ -68,9 +68,14 @@ feature 'Diffs URL', js: true, feature: true do
     let(:merge_request) { create(:merge_request_with_diffs, source_project: forked_project, target_project: project, author: author_user) }
     let(:changelog_id) { Digest::SHA1.hexdigest("CHANGELOG") }
 
+    before do
+      forked_project.repository.after_import
+    end
+
     context 'as author' do
       it 'shows direct edit link' do
         login_as(author_user)
+
         visit diffs_namespace_project_merge_request_path(project.namespace, project, merge_request)
 
         # Throws `Capybara::Poltergeist::InvalidSelector` if we try to use `#hash` syntax
@@ -81,6 +86,7 @@ feature 'Diffs URL', js: true, feature: true do
     context 'as user who needs to fork' do
       it 'shows fork/cancel confirmation' do
         login_as(user)
+
         visit diffs_namespace_project_merge_request_path(project.namespace, project, merge_request)
 
         # Throws `Capybara::Poltergeist::InvalidSelector` if we try to use `#hash` syntax

spec/features/projects/mirror_spec.rb

@@ -10,27 +10,39 @@ feature 'Project mirror', feature: true do
       login_as user
     end
 
-    describe 'pressing "Update now"' do
-      before { visit namespace_project_mirror_path(project.namespace, project) }
+    context 'with Update now button' do
+      let(:timestamp) { Time.now }
 
-      it 'returns with the project updating (job enqueued)' do
-        Sidekiq::Testing.fake! { click_link('Update Now') }
-
-        expect(page).to have_content('Updating')
+      before do
+        project.mirror_data.update_attributes(next_execution_timestamp: timestamp + 10.minutes)
       end
-    end
 
-    describe 'synchronization times' do
-      Gitlab::Mirror::SYNC_TIME_TO_CRON.keys.reverse.each_with_index do |sync_time, index|
-        describe "#{sync_time} minimum mirror sync time" do
-          before do
-            stub_application_setting(minimum_mirror_sync_time: sync_time)
+      context 'when able to force update' do
+        it 'forces import' do
+          project.update_attributes(mirror_last_update_at: timestamp - 8.minutes)
+
+          expect_any_instance_of(EE::Project).to receive(:force_import_job!)
+
+          Timecop.freeze(timestamp) do
             visit namespace_project_mirror_path(project.namespace, project)
           end
 
-          it 'shows the correct selector options' do
-            expect(page).to have_selector('.project-mirror-sync-time > option', count: index + 1)
+          Sidekiq::Testing.fake! { click_link('Update Now') }
+        end
+      end
+
+      context 'when unable to force update' do
+        it 'does not force import' do
+          project.update_attributes(mirror_last_update_at: timestamp - 3.minutes)
+
+          expect_any_instance_of(EE::Project).not_to receive(:force_import_job!)
+
+          Timecop.freeze(timestamp) do
+            visit namespace_project_mirror_path(project.namespace, project)
           end
+
+          expect(page).to have_content('Update Now')
+          expect(page).to have_selector('.btn.disabled')
         end
       end
     end

spec/lib/gitlab/bitbucket_import/project_creator_spec.rb

@@ -25,7 +25,7 @@ describe Gitlab::BitbucketImport::ProjectCreator, lib: true do
   end
 
   it 'creates project' do
-    allow_any_instance_of(Project).to receive(:add_import_job)
+    allow_any_instance_of(EE::Project).to receive(:add_import_job)
 
     project_creator = Gitlab::BitbucketImport::ProjectCreator.new(repo, 'vim', namespace, user, access_params)
     project = project_creator.execute

spec/lib/gitlab/github_import/project_creator_spec.rb

@@ -17,7 +17,7 @@ describe Gitlab::GithubImport::ProjectCreator, lib: true do
 
   before do
     namespace.add_owner(user)
-    allow_any_instance_of(Project).to receive(:add_import_job)
+    allow_any_instance_of(EE::Project).to receive(:add_import_job)
   end
 
   describe '#execute' do

spec/lib/gitlab/gitlab_import/project_creator_spec.rb

@@ -21,7 +21,7 @@ describe Gitlab::GitlabImport::ProjectCreator, lib: true do
   end
 
   it 'creates project' do
-    allow_any_instance_of(Project).to receive(:add_import_job)
+    allow_any_instance_of(EE::Project).to receive(:add_import_job)
 
     project_creator = Gitlab::GitlabImport::ProjectCreator.new(repo, namespace, user, access_params)
     project = project_creator.execute

spec/lib/gitlab/google_code_import/project_creator_spec.rb

@@ -16,7 +16,7 @@ describe Gitlab::GoogleCodeImport::ProjectCreator, lib: true do
   end
 
   it 'creates project' do
-    allow_any_instance_of(Project).to receive(:add_import_job)
+    allow_any_instance_of(EE::Project).to receive(:add_import_job)
 
     project_creator = Gitlab::GoogleCodeImport::ProjectCreator.new(repo, namespace, user)
     project = project_creator.execute

spec/lib/gitlab/import_export/all_models.yml

@@ -260,6 +260,7 @@ project:
 - statistics
 - container_repositories
 - uploads
+- mirror_data
 award_emoji:
 - awardable
 - user

spec/lib/gitlab/mirror_spec.rb

 require 'spec_helper'
 
 describe Gitlab::Mirror do
-  before { Sidekiq::Logging.logger = nil }
-
-  describe '#sync_times' do
-    before { Timecop.freeze(DateTime.now.change(time)) }
-
-    describe 'every fifteen minutes' do
-      let!(:time) { { hour: 1, min: 15 } }
-
-      it 'returns only fifteen minutes' do
-        times = [Gitlab::Mirror::FIFTEEN]
-
-        expect(described_class.sync_times).to match_array(times)
-      end
-    end
-
-    describe 'at beginning of hour' do
-      let!(:time) { { hour: 1 } }
-
-      it 'returns only fifteen and hourly sync_times' do
-        times = [Gitlab::Mirror::FIFTEEN, Gitlab::Mirror::HOURLY]
-
-        expect(described_class.sync_times).to match_array(times)
-      end
-    end
-
-    describe 'at beginning of th hour' do
-      describe 'three' do
-        let!(:time) { { hour: 3 } }
-
-        it 'returns only fifteen hourly and three hour sync_times' do
-          times = [Gitlab::Mirror::FIFTEEN, Gitlab::Mirror::HOURLY, Gitlab::Mirror::THREE]
+  include Gitlab::CurrentSettings
 
-          expect(described_class.sync_times).to match_array(times)
-        end
-      end
-
-      describe 'six' do
-        let!(:time) { { hour: 6 } }
-
-        it 'returns only fifteen, hourly, three and six hour sync_times' do
-          times = [Gitlab::Mirror::FIFTEEN, Gitlab::Mirror::HOURLY, Gitlab::Mirror::THREE, Gitlab::Mirror::SIX]
-
-          expect(described_class.sync_times).to match_array(times)
-        end
-      end
-
-      describe 'twelve' do
-        let!(:time) { { hour: 12 } }
-
-        it 'returns only fifteen, hourly, three, six and twelve hour sync_times' do
-          times = [Gitlab::Mirror::FIFTEEN, Gitlab::Mirror::HOURLY, Gitlab::Mirror::THREE, Gitlab::Mirror::SIX, Gitlab::Mirror::TWELVE]
-
-          expect(described_class.sync_times).to match_array(times)
-        end
-      end
-    end
-
-    describe 'at beginning of day' do
-      let!(:time) { { hour: 0 } }
-
-      it 'returns daily hourly and fifteen sync_times' do
-        times = [Gitlab::Mirror::FIFTEEN, Gitlab::Mirror::HOURLY, Gitlab::Mirror::THREE, Gitlab::Mirror::SIX, Gitlab::Mirror::TWELVE, Gitlab::Mirror::DAILY]
-
-        expect(described_class.sync_times).to match_array(times)
-      end
-    end
-
-    after { Timecop.return }
-  end
+  before { Sidekiq::Logging.logger = nil }
 
   describe '#configure_cron_job!' do
-    let(:daily_cron)   { Gitlab::Mirror::SYNC_TIME_TO_CRON[Gitlab::Mirror::DAILY] }
-    let(:twelve_cron)  { Gitlab::Mirror::SYNC_TIME_TO_CRON[Gitlab::Mirror::TWELVE] }
-    let(:six_cron)     { Gitlab::Mirror::SYNC_TIME_TO_CRON[Gitlab::Mirror::SIX] }
-    let(:three_cron)   { Gitlab::Mirror::SYNC_TIME_TO_CRON[Gitlab::Mirror::THREE] }
-    let(:hourly_cron)  { Gitlab::Mirror::SYNC_TIME_TO_CRON[Gitlab::Mirror::HOURLY] }
-    let(:fifteen_cron) { Gitlab::Mirror::SYNC_TIME_TO_CRON[Gitlab::Mirror::FIFTEEN] }
+    let(:cron) { Gitlab::Mirror::SCHEDULER_CRON }
 
     describe 'with jobs already running' do
-      def setup_mirrors_cron_job(current, updated_time)
-        allow_any_instance_of(ApplicationSetting).to receive(:minimum_mirror_sync_time).and_return(current)
-        Gitlab::Mirror.configure_cron_job!
-        allow_any_instance_of(ApplicationSetting).to receive(:minimum_mirror_sync_time).and_return(updated_time)
-      end
-
-      describe 'with daily minimum_mirror_sync_time' do
-        before { setup_mirrors_cron_job(Gitlab::Mirror::HOURLY, Gitlab::Mirror::DAILY) }
-
-        it 'changes cron of update_all_mirrors_worker to daily' do
-          expect { described_class.configure_cron_job! }.to change { Sidekiq::Cron::Job.find("update_all_mirrors_worker").cron }.from(hourly_cron).to(daily_cron)
-        end
-      end
-
-      describe 'with twelve hour minimum_mirror_sync_time' do
-        before { setup_mirrors_cron_job(Gitlab::Mirror::DAILY, Gitlab::Mirror::TWELVE) }
-
-        it 'changes cron of update_all_mirrors_worker to every twelve hours' do
-          expect { described_class.configure_cron_job! }.to change { Sidekiq::Cron::Job.find("update_all_mirrors_worker").cron }.from(daily_cron).to(twelve_cron)
-        end
-      end
-
-      describe 'with six hour minimum_mirror_sync_time' do
-        before { setup_mirrors_cron_job(Gitlab::Mirror::DAILY, Gitlab::Mirror::SIX) }
-
-        it 'changes cron of update_all_mirrors_worker to every six hours' do
-          expect { described_class.configure_cron_job! }.to change { Sidekiq::Cron::Job.find("update_all_mirrors_worker").cron }.from(daily_cron).to(six_cron)
-        end
-      end
-
-      describe 'with three hour minimum_mirror_sync_time' do
-        before { setup_mirrors_cron_job(Gitlab::Mirror::DAILY, Gitlab::Mirror::THREE) }
-
-        it 'changes cron of update_all_mirrors_worker to every three hours' do
-          expect { described_class.configure_cron_job! }.to change { Sidekiq::Cron::Job.find("update_all_mirrors_worker").cron }.from(daily_cron).to(three_cron)
-        end
-      end
-
-      describe 'with hourly minimum_mirror_sync_time' do
-        before { setup_mirrors_cron_job(Gitlab::Mirror::DAILY, Gitlab::Mirror::HOURLY) }
+      it 'creates a new cron job' do
+        described_class.configure_cron_job!
 
-        it 'changes cron of update_all_mirrors_worker to hourly' do
-          expect { described_class.configure_cron_job! }.to change { Sidekiq::Cron::Job.find("update_all_mirrors_worker").cron }.from(daily_cron).to(hourly_cron)
-        end
-      end
+        expect(subject).to receive(:destroy_cron_job!)
+        expect(Sidekiq::Cron::Job).to receive(:create)
 
-      describe 'with fifteen minimum_mirror_sync_time' do
-        before { setup_mirrors_cron_job(Gitlab::Mirror::DAILY, Gitlab::Mirror::FIFTEEN) }
-
-        it 'changes cron of update_all_mirrors_worker to fifteen' do
-          expect { described_class.configure_cron_job! }.to change { Sidekiq::Cron::Job.find("update_all_mirrors_worker").cron }.from(daily_cron).to(fifteen_cron)
-        end
+        expect { subject.configure_cron_job! }.to change { Sidekiq::Cron::Job.find("update_all_mirrors_worker") }
       end
     end
 
     describe 'without jobs already running' do
       before do
-        Sidekiq::Cron::Job.find("update_all_mirrors_worker").destroy
-      end
-
-      describe 'with daily minimum_mirror_sync_time' do
-        before { allow_any_instance_of(ApplicationSetting).to receive(:minimum_mirror_sync_time).and_return(Gitlab::Mirror::DAILY) }
-
-        it 'creates update_all_mirrors_worker with cron of daily sync_time' do
-          expect { described_class.configure_cron_job! }.to change { Sidekiq::Cron::Job.find("update_all_mirrors_worker") }.from(nil).to(Sidekiq::Cron::Job)
-          expect(Sidekiq::Cron::Job.find("update_all_mirrors_worker").cron).to eq(daily_cron)
-        end
+        Sidekiq::Cron::Job.find("update_all_mirrors_worker")&.destroy
       end
 
-      describe 'with twelve hours minimum_mirror_sync_time' do
-        before { allow_any_instance_of(ApplicationSetting).to receive(:minimum_mirror_sync_time).and_return(Gitlab::Mirror::TWELVE) }
-
-        it 'creates update_all_mirrors_worker with cron of every twelve hours sync_time' do
-          expect { described_class.configure_cron_job! }.to change { Sidekiq::Cron::Job.find("update_all_mirrors_worker") }.from(nil).to(Sidekiq::Cron::Job)
-          expect(Sidekiq::Cron::Job.find("update_all_mirrors_worker").cron).to eq(twelve_cron)
-        end
-      end
-
-      describe 'with six hours minimum_mirror_sync_time' do
-        before { allow_any_instance_of(ApplicationSetting).to receive(:minimum_mirror_sync_time).and_return(Gitlab::Mirror::SIX) }
-
-        it 'creates update_all_mirrors_worker with cron of every six hours sync_time' do
-          expect { described_class.configure_cron_job! }.to change { Sidekiq::Cron::Job.find("update_all_mirrors_worker") }.from(nil).to(Sidekiq::Cron::Job)
-          expect(Sidekiq::Cron::Job.find("update_all_mirrors_worker").cron).to eq(six_cron)
-        end
-      end
-
-      describe 'with three hours minimum_mirror_sync_time' do
-        before { allow_any_instance_of(ApplicationSetting).to receive(:minimum_mirror_sync_time).and_return(Gitlab::Mirror::THREE) }
-
-        it 'creates update_all_mirrors_worker with cron of every three hours sync_time' do
-          expect { described_class.configure_cron_job! }.to change { Sidekiq::Cron::Job.find("update_all_mirrors_worker") }.from(nil).to(Sidekiq::Cron::Job)
-          expect(Sidekiq::Cron::Job.find("update_all_mirrors_worker").cron).to eq(three_cron)
-        end
-      end
-
-      describe 'with hourly minimum_mirror_sync_time' do
-        before { allow_any_instance_of(ApplicationSetting).to receive(:minimum_mirror_sync_time).and_return(Gitlab::Mirror::HOURLY) }
-
-        it 'creates update_all_mirrors_worker with cron of hourly sync_time' do
-          expect { described_class.configure_cron_job! }.to change { Sidekiq::Cron::Job.find("update_all_mirrors_worker") }.from(nil).to(Sidekiq::Cron::Job)
-          expect(Sidekiq::Cron::Job.find("update_all_mirrors_worker").cron).to eq(hourly_cron)
-        end
-      end
-
-      describe 'with fifteen minimum_mirror_sync_time' do
-        it 'creates update_all_mirrors_worker with cron of fifteen sync_time' do
-          expect { described_class.configure_cron_job! }.to change { Sidekiq::Cron::Job.find("update_all_mirrors_worker") }.from(nil).to(Sidekiq::Cron::Job)
-          expect(Sidekiq::Cron::Job.find("update_all_mirrors_worker").cron).to eq(fifteen_cron)
-        end
+      it 'creates update_all_mirrors_worker' do
+        expect { described_class.configure_cron_job! }.to change { Sidekiq::Cron::Job.find("update_all_mirrors_worker") }.from(nil).to(Sidekiq::Cron::Job)
+        expect(Sidekiq::Cron::Job.find("update_all_mirrors_worker").cron).to eq(cron)
       end
 
       describe 'when Geo is enabled' do
         it 'disables mirror cron job' do
           described_class.configure_cron_job!
 
-          expect(described_class.update_all_mirrors_cron_job).to be_enabled
+          expect(Sidekiq::Cron::Job.find("update_all_mirrors_worker")).to be_enabled
 
           allow(Gitlab::Geo).to receive(:secondary?).and_return(true)
           described_class.configure_cron_job!
 
-          expect(described_class.update_all_mirrors_cron_job).to be_nil
+          expect(Sidekiq::Cron::Job.find("update_all_mirrors_worker")).to be_nil
         end
       end
     end
   end
 
-  describe '#at_beginning_of_day?' do
-    it 'returns true if at beginning_of_day' do
-      Timecop.freeze(DateTime.now.beginning_of_day)
+  describe '#max_mirror_capacity_reached?' do
+    it 'returns true if available capacity is 0' do
+      expect(described_class).to receive(:available_capacity).and_return(0)
 
-      expect(described_class.at_beginning_of_day?).to be true
+      expect(described_class.max_mirror_capacity_reached?).to eq(true)
     end
 
-    it 'returns true during the first 14 minutes of the day' do
-      Timecop.freeze(DateTime.now.beginning_of_day + 14.minutes)
+    it 'returns false if available capacity is not 0' do
+      expect(described_class).to receive(:available_capacity).and_return(1)
 
-      expect(described_class.at_beginning_of_day?).to be true
+      expect(described_class.max_mirror_capacity_reached?).to eq(false)
     end
 
-    it 'returns false if some time after the day started' do
-      Timecop.freeze(DateTime.now.midday)
-
-      expect(described_class.at_beginning_of_day?).to be false
+    after do
+      Gitlab::Redis.with { |redis| redis.del(Gitlab::Mirror::PULL_CAPACITY_KEY) }
     end
-
-    after { Timecop.return }
   end
 
-  describe '#at_beginning_of_hour?' do
-    before { Timecop.freeze(DateTime.now.change(time)) }
+  describe '#threshold_reached?' do
+    let(:mirror_capacity_threshold) { current_application_settings.mirror_capacity_threshold }
 
-    describe 'without hour mark' do
-      describe 'at beginning of day' do
-        let!(:time) { { hour: 0 } }
+    it 'returns true if available capacity surpassed defined threshold' do
+      expect(described_class).to receive(:available_capacity).and_return(mirror_capacity_threshold + 1)
 
-        it { expect(described_class.at_beginning_of_hour?).to be true }
-      end
+      expect(described_class.threshold_reached?).to eq(true)
+    end
 
-      describe 'at beginning of hour' do
-        let!(:time) { { hour: 1 } }
+    it 'returns true if available capacity is equal to the defined threshold' do
+      expect(described_class).to receive(:available_capacity).and_return(mirror_capacity_threshold)
 
-        it { expect(described_class.at_beginning_of_hour?).to be true }
-      end
+      expect(described_class.threshold_reached?).to eq(true)
+    end
 
-      describe 'at beginning of hour' do
-        let!(:time) { { hour: 1, min: 15 } }
+    it 'returns false if available capacity is below the defined threshold' do
+      expect(described_class).to receive(:available_capacity).and_return(mirror_capacity_threshold - 1)
 
-        it { expect(described_class.at_beginning_of_hour?).to be false }
-      end
+      expect(described_class.threshold_reached?).to eq(false)
     end
 
-    describe 'with hour mark' do
-      describe 'three' do
-        let!(:time) { { hour: 3 } }
+    after do
+      Gitlab::Redis.with { |redis| redis.del(Gitlab::Mirror::PULL_CAPACITY_KEY) }
+    end
+  end
 
-        it { expect(described_class.at_beginning_of_hour?(3)).to be true }
+  describe '#available_capacity' do
+    context 'when redis key does not exist' do
+      it 'returns mirror_max_capacity' do
+        expect(described_class.available_capacity).to eq(current_application_settings.mirror_max_capacity)
+      end
+    end
 
-        describe 'with another hour' do
-          let!(:time) { { hour: 4 } }
+    context 'when redis key exists' do
+      it 'returns available capacity' do
+        current_capacity = 10
 
-          it { expect(described_class.at_beginning_of_hour?(3)).to be false }
+        Gitlab::Redis.with do |redis|
+          (1..10).to_a.each do |id|
+            redis.sadd(Gitlab::Mirror::PULL_CAPACITY_KEY, id)
+          end
         end
+
+        expect(described_class.available_capacity).to eq(current_application_settings.mirror_max_capacity - current_capacity)
       end
+    end
+
+    after do
+      Gitlab::Redis.with { |redis| redis.del(Gitlab::Mirror::PULL_CAPACITY_KEY) }
+    end
+  end
 
-      describe 'six' do
-        let!(:time) { { hour: 6 } }
+  describe '#increment_capacity' do
+    it 'increments capacity' do
+      max_capacity = current_application_settings.mirror_max_capacity
 
-        it { expect(described_class.at_beginning_of_hour?(6)).to be true }
+      expect { described_class.increment_capacity(1) }.to change { described_class.available_capacity }.from(max_capacity).to(max_capacity - 1)
+    end
 
-        describe 'with another hour' do
-          let!(:time) { { hour: 4 } }
+    after do
+      Gitlab::Redis.with { |redis| redis.del(Gitlab::Mirror::PULL_CAPACITY_KEY) }
+    end
+  end
 
-          it { expect(described_class.at_beginning_of_hour?(6)).to be false }
-        end
-      end
+  describe '#decrement_capacity' do
+    let!(:id) { 1 }
 
-      describe 'twelve' do
-        let!(:time) { { hour: 12 } }
+    context 'with capacity above 0' do
+      it 'decrements capacity' do
+        max_capacity = current_application_settings.mirror_max_capacity
 
-        it { expect(described_class.at_beginning_of_hour?(12)).to be true }
+        described_class.increment_capacity(id)
 
-        describe 'with another hour' do
-          let!(:time) { { hour: 4 } }
+        expect { described_class.decrement_capacity(id) }.to change { described_class.available_capacity }.from(max_capacity - 1).to(max_capacity)
+      end
+    end
 
-          it { expect(described_class.at_beginning_of_hour?(12)).to be false }
-        end
+    context 'with non-existent id' do
+      it 'does not decrement capacity' do
+        expect { described_class.decrement_capacity(id) }.not_to change { described_class.available_capacity }
       end
     end
 
-    after { Timecop.return }
+    after do
+      Gitlab::Redis.with { |redis| redis.del(Gitlab::Mirror::PULL_CAPACITY_KEY) }
+    end
+  end
+
+  describe '#max_delay' do
+    it 'returns max delay with some jitter' do
+      expect(described_class.max_delay).to be_within(1.minute).of(5.hours)
+    end
   end
 end

spec/models/application_setting_spec.rb

@@ -21,11 +21,24 @@ describe ApplicationSetting, models: true do
     it { is_expected.to allow_value(https).for(:after_sign_out_path) }
     it { is_expected.not_to allow_value(ftp).for(:after_sign_out_path) }
 
-    it { is_expected.to allow_value(Gitlab::Mirror::FIFTEEN).for(:minimum_mirror_sync_time) }
-    it { is_expected.to allow_value(Gitlab::Mirror::HOURLY).for(:minimum_mirror_sync_time) }
-    it { is_expected.to allow_value(Gitlab::Mirror::DAILY).for(:minimum_mirror_sync_time) }
-    it { is_expected.not_to allow_value(nil).for(:minimum_mirror_sync_time) }
-    it { is_expected.not_to allow_value(61).for(:minimum_mirror_sync_time) }
+    it { is_expected.to allow_value(10).for(:mirror_max_delay) }
+    it { is_expected.not_to allow_value(nil).for(:mirror_max_delay) }
+    it { is_expected.not_to allow_value(0).for(:mirror_max_delay) }
+    it { is_expected.not_to allow_value(1.0).for(:mirror_max_delay) }
+    it { is_expected.not_to allow_value(-1).for(:mirror_max_delay) }
+
+    it { is_expected.to allow_value(10).for(:mirror_max_capacity) }
+    it { is_expected.not_to allow_value(nil).for(:mirror_max_capacity) }
+    it { is_expected.not_to allow_value(0).for(:mirror_max_capacity) }
+    it { is_expected.not_to allow_value(1.0).for(:mirror_max_capacity) }
+    it { is_expected.not_to allow_value(-1).for(:mirror_max_capacity) }
+
+    it { is_expected.to allow_value(10).for(:mirror_capacity_threshold) }
+    it { is_expected.not_to allow_value(nil).for(:mirror_capacity_threshold) }
+    it { is_expected.not_to allow_value(0).for(:mirror_capacity_threshold) }
+    it { is_expected.not_to allow_value(1.0).for(:mirror_capacity_threshold) }
+    it { is_expected.not_to allow_value(-1).for(:mirror_capacity_threshold) }
+    it { is_expected.not_to allow_value(subject.mirror_max_capacity + 1).for(:mirror_capacity_threshold) }
 
     describe 'disabled_oauth_sign_in_sources validations' do
       before do
@@ -82,51 +95,6 @@ describe ApplicationSetting, models: true do
       subject { setting }
     end
 
-    context "update minimum_mirror_sync_time" do
-      sync_times = Gitlab::Mirror::SYNC_TIME_TO_CRON.keys
-
-      before do
-        Sidekiq::Logging.logger = nil
-        sync_times.each do |sync_time|
-          create(:project, :mirror, sync_time: sync_time)
-        end
-      end
-
-      sync_times.drop(1).each_with_index do |sync_time, index|
-        context "with #{sync_time} sync_time" do
-          subject { setting.update_attributes(minimum_mirror_sync_time: sync_time) }
-
-          it "updates minimum mirror sync time to #{sync_time}" do
-            expect_any_instance_of(ApplicationSetting).to receive(:update_mirror_cron_job).and_call_original
-            expect(Gitlab::Mirror).to receive(:configure_cron_job!)
-
-            subject
-          end
-
-          it 'updates every mirror to the current minimum_mirror_sync_time' do
-            expect { subject }.to change { Project.mirror.where('sync_time < ?', sync_time).count }.from(index + 1).to(0)
-          end
-        end
-      end
-
-      # fifteen is a special case so we isolate it
-      context 'with default fifteen sync time' do
-        let(:sync_time) { Gitlab::Mirror::FIFTEEN }
-
-        it 'does not update minimum_mirror_sync_time' do
-          expect_any_instance_of(ApplicationSetting).not_to receive(:update_mirror_cron_job)
-          expect(Gitlab::Mirror).not_to receive(:configure_cron_job!)
-          expect(setting.minimum_mirror_sync_time).to eq(Gitlab::Mirror::FIFTEEN)
-
-          setting.update_attributes(minimum_mirror_sync_time: sync_time)
-        end
-
-        it 'updates every mirror to the current minimum_mirror_sync_time' do
-          expect { setting.update_attributes(minimum_mirror_sync_time: sync_time) }.not_to change { Project.mirror.where('sync_time < ?', sync_time).count }
-        end
-      end
-    end
-
     # Upgraded databases will have this sort of content
     context 'repository_storages is a String, not an Array' do
       before { setting.__send__(:raw_write_attribute, :repository_storages, 'default') }

spec/models/project_mirror_data_spec.rb

+require 'rails_helper'
+
+describe ProjectMirrorData, type: :model do
+  include Gitlab::CurrentSettings
+
+  describe 'associations' do
+    it { is_expected.to belong_to(:project) }
+  end
+
+  describe 'modules' do
+    it { is_expected.to include_module(Gitlab::CurrentSettings) }
+  end
+
+  describe 'validations' do
+    it { is_expected.to validate_presence_of(:project) }
+  end
+
+  describe 'when create' do
+    it 'sets next execution timestamp to now' do
+      project = create(:empty_project)
+
+      Timecop.freeze(Time.now) do
+        project.create_mirror_data
+
+        expect(project.mirror_data.next_execution_timestamp).to eq(Time.now)
+      end
+    end
+  end
+
+  describe '#reset_retry_count!' do
+    let(:mirror_data) { create(:project, :mirror, :import_finished).mirror_data }
+
+    it 'resets retry_count to 0' do
+      mirror_data.retry_count = 3
+
+      expect { mirror_data.reset_retry_count! }.to change { mirror_data.retry_count }.from(3).to(0)
+    end
+  end
+
+  describe '#increment_retry_count!' do
+    let(:mirror_data) { create(:project, :mirror, :import_finished).mirror_data }
+
+    it 'increments retry_count' do
+      expect { mirror_data.increment_retry_count! }.to change { mirror_data.retry_count }.from(0).to(1)
+    end
+  end
+
+  describe '#set_next_execution_timestamp!' do
+    let(:mirror_data) { create(:project, :mirror, :import_finished).mirror_data }
+    let!(:timestamp) { Time.now }
+    let!(:jitter) { 2.seconds }
+    let(:interval) { 2.minutes }
+
+    before do
+      allow_any_instance_of(ProjectMirrorData).to receive(:rand).and_return(jitter)
+    end
+
+    context 'when base delay is lower than mirror_max_delay' do
+      before do
+        mirror_data.last_update_started_at = timestamp - 1.minute
+      end
+
+      context 'when retry count is 0' do
+        it 'applies transition successfully' do
+          expect do
+            mirror_data.set_next_execution_timestamp!
+          end.to change { mirror_data.next_execution_timestamp }.to be_within(interval).of(timestamp + 26.minutes)
+        end
+      end
+
+      context 'when incrementing retry count' do
+        it 'applies transition successfully' do
+          mirror_data.retry_count = 2
+          mirror_data.increment_retry_count!
+
+          expect do
+            mirror_data.set_next_execution_timestamp!
+          end.to change { mirror_data.next_execution_timestamp }.to be_within(interval).of(timestamp + 79.minutes)
+        end
+      end
+    end
+
+    context 'when base delay is higher than mirror_max_delay' do
+      let!(:upper_jitter) { 30.seconds }
+      let(:max_timestamp) { timestamp + current_application_settings.mirror_max_delay.hours }
+
+      before do
+        allow_any_instance_of(Gitlab::Mirror).to receive(:rand).and_return(upper_jitter)
+        mirror_data.last_update_started_at = timestamp - 1.hour
+      end
+
+      context 'when reseting retry count' do
+        it 'applies transition successfully' do
+          expect do
+            mirror_data.set_next_execution_timestamp!
+          end.to change { mirror_data.next_execution_timestamp }.to be_within(interval).of(max_timestamp + upper_jitter)
+        end
+      end
+
+      context 'when incrementing retry count' do
+        it 'applies transition successfully' do
+          mirror_data.retry_count = 2
+          mirror_data.increment_retry_count!
+
+          expect do
+            mirror_data.set_next_execution_timestamp!
+          end.to change { mirror_data.next_execution_timestamp }.to be_within(interval).of(max_timestamp + upper_jitter)
+        end
+      end
+    end
+  end
+end

spec/models/project_spec.rb

@@ -50,7 +50,8 @@ describe Project, models: true do
     it { is_expected.to have_one(:external_wiki_service).dependent(:destroy) }
     it { is_expected.to have_one(:project_feature).dependent(:destroy) }
     it { is_expected.to have_one(:statistics).class_name('ProjectStatistics').dependent(:delete) }
-    it { is_expected.to have_one(:import_data).class_name('ProjectImportData').dependent(:destroy) }
+    it { is_expected.to have_one(:import_data).class_name('ProjectImportData').dependent(:delete) }
+    it { is_expected.to have_one(:mirror_data).class_name('ProjectMirrorData').dependent(:delete) }
     it { is_expected.to have_one(:last_event).class_name('Event') }
     it { is_expected.to have_one(:forked_from_project).through(:forked_project_link) }
     it { is_expected.to have_many(:commit_statuses) }
@@ -262,6 +263,18 @@ describe Project, models: true do
       expect(project2.errors[:import_url]).to include('imports are not allowed from that URL')
     end
 
+    it 'creates mirror data when enabled' do
+      project2 = create(:empty_project, :mirror, mirror: false)
+
+      expect { project2.update_attributes(mirror: true) }.to change { ProjectMirrorData.count }.from(0).to(1)
+    end
+
+    it 'destroys mirror data when disabled' do
+      project2 = create(:empty_project, :mirror)
+
+      expect { project2.update_attributes(mirror: false) }.to change { ProjectMirrorData.count }.from(1).to(0)
+    end
+
     describe 'project pending deletion' do
       let!(:project_pending_deletion) do
         create(:empty_project,
@@ -1681,16 +1694,13 @@ describe Project, models: true do
   end
 
   describe 'Project import job' do
-    let(:project) { create(:empty_project) }
-    let(:mirror) { false }
+    let(:project) { create(:empty_project, import_url: generate(:url)) }
 
     before do
       allow_any_instance_of(Gitlab::Shell).to receive(:import_repository)
         .with(project.repository_storage_path, project.path_with_namespace, project.import_url)
         .and_return(true)
 
-      allow(project).to receive(:repository_exists?).and_return(true)
-
       expect_any_instance_of(Repository).to receive(:after_import)
         .and_call_original
     end
@@ -1698,23 +1708,21 @@ describe Project, models: true do
     it 'imports a project' do
       expect_any_instance_of(RepositoryImportWorker).to receive(:perform).and_call_original
 
-      project.import_start
-      project.add_import_job
+      project.import_schedule
 
       expect(project.reload.import_status).to eq('finished')
     end
 
-    it 'imports a mirrored project' do
-      allow_any_instance_of(RepositoryUpdateMirrorWorker).to receive(:perform)
-      expect_any_instance_of(RepositoryImportWorker).to receive(:perform).and_call_original
-
-      project.import_start
+    context 'with a mirrored project' do
+      let(:project) { create(:empty_project, :mirror) }
 
-      project.mirror = true
+      it 'first calls RepositoryImportWorker and RepositoryUpdateMirrorWorker after' do
+        allow_any_instance_of(Project).to receive(:repository_exists?).and_return(false, true)
+        expect_any_instance_of(RepositoryUpdateMirrorWorker).to receive(:perform).with(project.id)
+        expect_any_instance_of(RepositoryImportWorker).to receive(:perform).with(project.id).and_call_original
 
-      project.add_import_job
-
-      expect(project.reload.import_status).to eq('finished')
+        project.import_schedule
+      end
     end
   end
 
@@ -1797,9 +1805,61 @@ describe Project, models: true do
     end
   end
 
+  describe  '#updating_mirror?' do
+    context 'when repository is empty' do
+      it 'returns false' do
+        project = create(:empty_project, :mirror, :import_started)
+
+        expect(project.updating_mirror?).to be false
+      end
+    end
+
+    context 'when project is not a mirror' do
+      it 'returns false' do
+        project = create(:project, :import_started)
+
+        expect(project.updating_mirror?).to be false
+      end
+    end
+
+    context 'when project is in progress' do
+      it 'returns true' do
+        project = create(:project, :mirror, :import_started)
+
+        expect(project.updating_mirror?).to be true
+      end
+    end
+
+    context 'when project is expected to run soon' do
+      it 'returns true' do
+        timestamp = Time.now
+        project = create(:project, :mirror, :import_finished)
+
+        project.mirror_last_update_at = timestamp - 3.minutes
+        project.mirror_data.next_execution_timestamp = timestamp - 2.minutes
+
+        expect(project.updating_mirror?).to be true
+      end
+    end
+  end
+
+  describe '#force_import_job!' do
+    it 'sets next execution timestamp to now and schedules UpdateAllMirrorsWorker' do
+      timestamp = Time.now
+      project = create(:project, :mirror)
+
+      project.mirror_data.update_attributes(next_execution_timestamp: timestamp - 3.minutes)
+
+      expect(UpdateAllMirrorsWorker).to receive(:perform_async)
+      Timecop.freeze(timestamp) do
+        expect { project.force_import_job! }.to change { project.mirror_data.reload.next_execution_timestamp }.to be_within(1.second).of(timestamp)
+      end
+    end
+  end
+
   describe '#add_import_job' do
     context 'forked' do
-      let(:forked_project_link) { create(:forked_project_link) }
+      let(:forked_project_link) { create(:forked_project_link, :forked_to_empty_project) }
       let(:forked_from_project) { forked_project_link.forked_from_project }
       let(:project) { forked_project_link.forked_to_project }
 
@@ -1810,15 +1870,33 @@ describe Project, models: true do
 
         project.add_import_job
       end
-    end
 
-    context 'not forked' do
-      let(:project) { create(:empty_project) }
+      context 'without mirror' do
+        it 'returns nil' do
+          project = create(:project)
+
+          expect(project.add_import_job).to be_nil
+        end
+      end
 
-      it 'schedules a RepositoryImportWorker job' do
-        expect(RepositoryImportWorker).to receive(:perform_async).with(project.id)
+      context 'without repository' do
+        it 'schedules RepositoryImportWorker' do
+          project = create(:empty_project, import_url: generate(:url))
 
-        project.add_import_job
+          expect(RepositoryImportWorker).to receive(:perform_async).with(project.id)
+
+          project.add_import_job
+        end
+      end
+
+      context 'with mirror' do
+        it 'schedules RepositoryUpdateMirrorWorker' do
+          project = create(:project, :mirror)
+
+          expect(RepositoryUpdateMirrorWorker).to receive(:perform_async).with(project.id)
+
+          project.add_import_job
+        end
       end
     end
   end
@@ -2143,6 +2221,16 @@ describe Project, models: true do
     end
   end
 
+  describe '#create_mirror_data' do
+    it 'it is called after save' do
+      project = create(:project)
+
+      expect(project).to receive(:create_mirror_data)
+
+      project.update(mirror: true, mirror_user: project.owner, import_url: 'http://foo.com')
+    end
+  end
+
   describe 'inside_path' do
     let!(:project1) { create(:empty_project, namespace: create(:namespace, path: 'name_pace')) }
     let!(:project2) { create(:empty_project) }

spec/services/projects/create_service_spec.rb

@@ -197,15 +197,13 @@ describe Projects::CreateService, '#execute', services: true do
   end
 
   context 'when a bad service template is created' do
-    before do
-      create(:service, type: 'DroneCiService', project: nil, template: true, active: true)
-    end
-
     it 'reports an error in the imported project' do
       opts[:import_url] = 'http://www.gitlab.com/gitlab-org/gitlab-ce'
+      create(:service, type: 'DroneCiService', project: nil, template: true, active: true)
+
       project = create_project(user, opts)
 
-      expect(project.errors.full_messages_for(:base).first).to match /Unable to save project. Error: Unable to save DroneCiService/
+      expect(project.errors.full_messages_for(:base).first).to match(/Unable to save project. Error: Unable to save DroneCiService/)
       expect(project.services.count).to eq 0
     end
   end

spec/workers/repository_fork_worker_spec.rb

 require 'spec_helper'
 
 describe RepositoryForkWorker do
-  let(:project) { create(:project, :repository) }
+  let(:project) { create(:project, :repository, :import_scheduled) }
   let(:fork_project) { create(:project, :repository, forked_from_project: project) }
   let(:shell) { Gitlab::Shell.new }
 
@@ -46,15 +46,27 @@ describe RepositoryForkWorker do
     end
 
     it "handles bad fork" do
+      source_path = project.full_path
+      target_path = fork_project.namespace.full_path
+      error_message = "Unable to fork project #{project.id} for repository #{source_path} -> #{target_path}"
+
       expect(shell).to receive(:fork_repository).and_return(false)
 
-      expect(subject.logger).to receive(:error)
+      expect do
+        subject.perform(project.id, '/test/path', source_path, target_path)
+      end.to raise_error(RepositoryForkWorker::ForkError, error_message)
+    end
 
-      subject.perform(
-        project.id,
-        '/test/path',
-        project.full_path,
-        fork_project.namespace.full_path)
+    it 'handles unexpected error' do
+      source_path = project.full_path
+      target_path = fork_project.namespace.full_path
+
+      allow_any_instance_of(Gitlab::Shell).to receive(:fork_repository).and_raise(RuntimeError)
+
+      expect do
+        subject.perform(project.id, '/test/path', source_path, target_path)
+      end.to raise_error(RepositoryForkWorker::ForkError)
+      expect(project.reload.import_status).to eq('failed')
     end
   end
 end

spec/workers/repository_import_worker_spec.rb

 require 'spec_helper'
 
 describe RepositoryImportWorker do
-  let(:project) { create(:empty_project) }
+  let(:project) { create(:empty_project, :import_scheduled) }
 
   subject { described_class.new }
 
@@ -21,15 +21,26 @@ describe RepositoryImportWorker do
     context 'when the import has failed' do
       it 'hide the credentials that were used in the import URL' do
         error = %q{remote: Not Found fatal: repository 'https://user:pass@test.com/root/repoC.git/' not found }
-        expect_any_instance_of(Projects::ImportService).to receive(:execute).
-          and_return({ status: :error, message: error })
-        allow(subject).to receive(:jid).and_return('123')
 
-        subject.perform(project.id)
+        expect_any_instance_of(Projects::ImportService).to receive(:execute).and_return({ status: :error, message: error })
+        allow(subject).to receive(:jid).and_return('123')
 
-        expect(project.reload.import_error).to include("https://*****:*****@test.com/root/repoC.git/")
+        expect do
+          subject.perform(project.id)
+        end.to raise_error(RepositoryImportWorker::ImportError, error)
         expect(project.reload.import_jid).not_to be_nil
       end
     end
+
+    context 'with unexpected error' do
+      it 'marks import as failed' do
+        allow_any_instance_of(Projects::ImportService).to receive(:execute).and_raise(RuntimeError)
+
+        expect do
+          subject.perform(project.id)
+        end.to raise_error(RepositoryImportWorker::ImportError)
+        expect(project.reload.import_status).to eq('failed')
+      end
+    end
   end
 end

spec/workers/repository_update_mirror_dispatch_worker_spec.rb

-require 'rails_helper'
-
-describe RepositoryUpdateMirrorDispatchWorker do
-  describe '#perform' do
-    it 'executes project#update_mirror if can obtain a lease' do
-      allow_any_instance_of(Gitlab::ExclusiveLease)
-        .to receive(:try_obtain).and_return(true)
-      expect_any_instance_of(Project).to receive(:update_mirror)
-
-      project = create(:empty_project, :mirror)
-      described_class.new.perform(project.id)
-    end
-
-    it 'just returns if cannot obtain a lease' do
-      allow_any_instance_of(Gitlab::ExclusiveLease)
-        .to receive(:try_obtain).and_return(false)
-      expect_any_instance_of(Project).not_to receive(:update_mirror)
-
-      project = create(:empty_project, :mirror)
-      described_class.new.perform(project.id)
-    end
-  end
-end

spec/workers/repository_update_mirror_worker_spec.rb

 require 'rails_helper'
 
 describe RepositoryUpdateMirrorWorker do
+  subject { described_class.new }
+
   describe '#perform' do
-    it 'sets import as finished when update mirror service executes successfully' do
-      project = create(:empty_project, :mirror)
+    context 'with status none' do
+      let(:project) { create(:empty_project, :mirror, :import_scheduled) }
+
+      it 'sets status as finished when update mirror service executes successfully' do
+        expect_any_instance_of(Projects::UpdateMirrorService).to receive(:execute).and_return(status: :success)
+
+        expect { subject.perform(project.id) }.to change { project.reload.import_status }.to('finished')
+      end
+
+      it 'sets status as failed when update mirror service executes with errors' do
+        error_message = 'fail!'
+
+        expect_any_instance_of(Projects::UpdateMirrorService).to receive(:execute).and_return(status: :error, message: error_message)
+
+        expect do
+          subject.perform(project.id)
+        end.to raise_error(RepositoryUpdateMirrorWorker::UpdateError, error_message)
+        expect(project.reload.import_status).to eq('failed')
+      end
+    end
 
-      expect_any_instance_of(Projects::UpdateMirrorService).to receive(:execute).and_return(status: :success)
+    context 'with another worker already running' do
+      it 'raises UpdateAlreadyInProgressError' do
+        mirror = create(:project, :mirror, :import_started)
 
-      expect { described_class.new.perform(project.id) }
-        .to change { project.reload.import_status }.to('finished')
+        expect do
+          subject.perform(mirror.id)
+        end.to raise_error(RepositoryUpdateMirrorWorker::UpdateAlreadyInProgressError)
+      end
     end
 
-    it 'sets import as failed when update mirror service executes with errors' do
-      project = create(:empty_project, :mirror)
+    context 'with unexpected error' do
+      it 'marks mirror as failed' do
+        mirror = create(:project, :mirror, :import_scheduled)
 
-      expect_any_instance_of(Projects::UpdateMirrorService).to receive(:execute).and_return(status: :error, message: 'fail!')
+        allow_any_instance_of(Projects::UpdateMirrorService).to receive(:execute).and_raise(RuntimeError)
 
-      expect { described_class.new.perform(project.id) }
-        .to change { project.reload.import_status }.to('failed')
+        expect do
+          subject.perform(mirror.id)
+        end.to raise_error(RepositoryUpdateMirrorWorker::UpdateError)
+        expect(mirror.reload.import_status).to eq('failed')
+      end
     end
 
-    it 'does nothing if project does not exist' do
-      expect(described_class.new.perform(1000000)).to be_nil
+    context 'threshold_reached?' do
+      let(:mirror) { create(:project, :mirror) }
+
+      before do
+        expect_any_instance_of(Projects::UpdateMirrorService).to receive(:execute).and_return(status: :success)
+      end
+
+      context 'with threshold_reached? true' do
+        it 'schedules UpdateAllMirrorsWorker' do
+          expect(Gitlab::Mirror).to receive(:threshold_reached?).and_return(true)
+
+          expect(UpdateAllMirrorsWorker).to receive(:perform_async)
+
+          subject.perform(mirror.id)
+        end
+      end
+
+      context 'with threshold_reached? false' do
+        it 'does not schedule UpdateAllMirrorsWorker' do
+          expect(Gitlab::Mirror).to receive(:threshold_reached?).and_return(false)
+
+          expect(UpdateAllMirrorsWorker).not_to receive(:perform_async)
+
+          subject.perform(mirror.id)
+        end
+      end
     end
   end
 end

spec/workers/repository_update_remote_mirror_worker_spec.rb

@@ -47,6 +47,17 @@ describe RepositoryUpdateRemoteMirrorWorker do
       end
     end
 
+    context 'with unexpected error' do
+      it 'marks mirror as failed' do
+        allow_any_instance_of(Projects::UpdateRemoteMirrorService).to receive(:execute).with(remote_mirror).and_raise(RuntimeError)
+
+        expect do
+          subject.perform(remote_mirror.id, Time.now)
+        end.to raise_error(RepositoryUpdateRemoteMirrorWorker::UpdateError)
+        expect(remote_mirror.reload.update_status).to eq('failed')
+      end
+    end
+
     context 'with another worker already running' do
       before do
         remote_mirror.update_attributes(update_status: 'started')

spec/workers/update_all_mirrors_worker_spec.rb

@@ -6,14 +6,6 @@ describe UpdateAllMirrorsWorker do
   before { allow_any_instance_of(Gitlab::ExclusiveLease).to receive(:try_obtain).and_return(true) }
 
   describe '#perform' do
-    let!(:fifteen_mirror)  { create(:empty_project, :mirror, sync_time: Gitlab::Mirror::FIFTEEN) }
-    let!(:hourly_mirror)   { create(:empty_project, :mirror, sync_time: Gitlab::Mirror::HOURLY) }
-    let!(:three_mirror)    { create(:empty_project, :mirror, sync_time: Gitlab::Mirror::THREE) }
-    let!(:six_mirror)      { create(:empty_project, :mirror, sync_time: Gitlab::Mirror::SIX) }
-    let!(:twelve_mirror)   { create(:empty_project, :mirror, sync_time: Gitlab::Mirror::TWELVE) }
-    let!(:daily_mirror)    { create(:empty_project, :mirror, sync_time: Gitlab::Mirror::DAILY) }
-    let!(:outdated_mirror) { create(:empty_project, :mirror) }
-
     it 'fails stuck mirrors' do
       expect(worker).to receive(:fail_stuck_mirrors!)
 
@@ -29,68 +21,6 @@ describe UpdateAllMirrorsWorker do
 
       worker.perform
     end
-
-    describe 'sync_time' do
-      def expect_worker_to_enqueue_mirrors(mirrors)
-        mirrors.each do |mirror|
-          expect(worker).to receive(:rand).with((mirror.sync_time / 2).minutes).and_return(mirror.sync_time / 2)
-          expect(RepositoryUpdateMirrorDispatchWorker).to receive(:perform_in).with(mirror.sync_time / 2, mirror.id)
-        end
-
-        worker.perform
-      end
-
-      before do
-        time = DateTime.now.change(time_params)
-
-        Timecop.freeze(time)
-        outdated_mirror.update_attributes(mirror_last_successful_update_at: time - (Gitlab::Mirror::DAILY + 5).minutes)
-      end
-
-      describe 'fifteen' do
-        let!(:time_params) { { hour: 1, min: 15 } }
-        let(:mirrors) { [fifteen_mirror, outdated_mirror] }
-
-        it { expect_worker_to_enqueue_mirrors(mirrors) }
-      end
-
-      describe 'hourly' do
-        let!(:time_params) { { hour: 1 } }
-        let(:mirrors) { [fifteen_mirror, hourly_mirror, outdated_mirror] }
-
-        it { expect_worker_to_enqueue_mirrors(mirrors) }
-      end
-
-      describe 'three' do
-        let!(:time_params) { { hour: 3 } }
-        let(:mirrors) { [fifteen_mirror, hourly_mirror, three_mirror, outdated_mirror] }
-
-        it { expect_worker_to_enqueue_mirrors(mirrors) }
-      end
-
-      describe 'six' do
-        let!(:time_params) { { hour: 6 } }
-        let(:mirrors) { [fifteen_mirror, hourly_mirror, three_mirror, six_mirror, outdated_mirror] }
-
-        it { expect_worker_to_enqueue_mirrors(mirrors) }
-      end
-
-      describe 'twelve' do
-        let!(:time_params) { { hour: 12 } }
-        let(:mirrors) { [fifteen_mirror, hourly_mirror, three_mirror, six_mirror, twelve_mirror, outdated_mirror] }
-
-        it { expect_worker_to_enqueue_mirrors(mirrors) }
-      end
-
-      describe 'daily' do
-        let!(:time_params) { { hour: 0 } }
-        let(:mirrors) { [fifteen_mirror, hourly_mirror, three_mirror, six_mirror, twelve_mirror, daily_mirror, outdated_mirror] }
-
-        it { expect_worker_to_enqueue_mirrors(mirrors) }
-      end
-
-      after { Timecop.return }
-    end
   end
 
   describe '#fail_stuck_mirrors!' do
@@ -120,22 +50,15 @@ describe UpdateAllMirrorsWorker do
       fail_stuck_mirrors!
     end
 
-    it 'transitions stuck mirrors to a failed state' do
-      project = create(:empty_project, :mirror, mirror_last_update_at: 12.hours.ago)
+    it 'transitions stuck mirrors to a failed state and updates import_error message' do
+      project = create(:empty_project, :mirror, :import_started)
+      project.mirror_data.update_attributes(last_update_started_at: 25.minutes.ago)
 
       fail_stuck_mirrors!
       project.reload
 
       expect(project).to be_import_failed
-    end
-
-    it 'updates the import_error message' do
-      project = create(:empty_project, :mirror, mirror_last_update_at: 12.hours.ago)
-
-      fail_stuck_mirrors!
-      project.reload
-
-      expect(project.import_error).to eq 'The mirror update took too long to complete.'
+      expect(project.reload.import_error).to eq 'The mirror update took too long to complete.'
     end
   end
 end