Merge pull request #1132 from travis-ci/epic-fast-track-build

Fast track feature going live

Author: Shairyar Baig

.travis.yml

@@ -1,7 +1,7 @@
 language: ruby
 
 import:
-  - travis-ci/build-configs:db-setup.yml
+  - travis-ci/build-configs:db-setup.yml@bionic
 
 rvm: 2.6.5
 

lib/travis/api/v3/models/build.rb

@@ -3,6 +3,9 @@ class Models::BuildConfig < Model
   end
 
   class Models::Build < Model
+
+    HIGH_PRIORITY = 5
+    
     belongs_to :commit
     belongs_to :tag
     belongs_to :pull_request
@@ -30,6 +33,8 @@ class Models::Build < Model
       primary_key: [:repository_id, :branch],
       class_name:  'Travis::API::V3::Models::Branch'.freeze
 
+    scope :running_builds, -> { where.not(state: ['passed', 'failed', 'errored', 'cancelled']) }
+
     def created_by
       return unless sender
       sender.becomes(created_by_class)
@@ -88,5 +93,9 @@ def clear_debug_options!
     def log_complete
       jobs.all?(&:log_complete)
     end
+
+    def high_priority?
+      jobs.where(priority: HIGH_PRIORITY).present?
+    end
   end
 end

lib/travis/api/v3/models/organization.rb

@@ -21,6 +21,14 @@ def education
       Travis::Features.owner_active?(:educational_org, self)
     end
 
+    def builds
+      Models::Build.where(owner_type: 'Organization', owner_id: id)
+    end
+
+    def build_priorities_enabled?
+      Travis::Features.owner_active?(:build_priorities_org, self)
+    end
+
     alias members users
   end
 end

lib/travis/api/v3/permissions/build.rb

@@ -9,5 +9,9 @@ def cancel?
     def restart?
       restartable?
     end
+
+    def prioritize?
+      read? && build_priorities?
+    end
   end
 end

lib/travis/api/v3/permissions/generic.rb

@@ -75,5 +75,10 @@ def adminable?
       access_control.adminable? object
     end
 
+    def build_priorities?
+      return false if object.owner_type != 'Organization'
+      object.owner.build_priorities_enabled?
+    end
+
   end
 end

lib/travis/api/v3/permissions/job.rb

@@ -17,5 +17,9 @@ def debug?
     def delete_log?
       write?
     end
+
+    def prioritize?
+      read? && build_priorities?
+    end
   end
 end

lib/travis/api/v3/queries/build.rb

@@ -3,18 +3,20 @@
 
 module Travis::API::V3
   class Queries::Build < Query
-    params :id
+    params :id, :cancel_all
+
+    PRIORITY = { high: 5, low: -5, medium: nil }
 
     def find
       return Models::Build.find_by_id(id) if id
       raise WrongParams, 'missing build.id'.freeze
     end
 
-    def cancel(user)
+    def cancel(user, build_id)
       raise BuildNotCancelable if %w(passed failed canceled errored).include? find.state
 
-      payload = { id: id, user_id: user.id, source: 'api' }
-      service = Travis::Enqueue::Services::CancelModel.new(user, { build_id: id })
+      payload = { id: build_id, user_id: user.id, source: 'api' }
+      service = Travis::Enqueue::Services::CancelModel.new(user, { build_id: build_id })
       service.push("build:cancel", payload)
       payload
     end
@@ -33,5 +35,15 @@ def restart(user)
         payload
       end
     end
+
+    def prioritize_and_cancel(user)
+      raise NotFound, "Jobs are not found" if find.jobs.blank?
+      find.jobs.update_all(priority: PRIORITY[:high])
+      return if find.owner_type != "Organization"
+      if bool(cancel_all)
+        low_priority_builds = find.owner.builds.running_builds.select { |build| !build.high_priority? }
+        low_priority_builds.each { |build| cancel(user, build.id) } if low_priority_builds
+      end
+    end
   end
 end

lib/travis/api/v3/renderer/build.rb

@@ -1,11 +1,13 @@
 module Travis::API::V3
   class Renderer::Build < ModelRenderer
-    representation(:minimal,  :id, :number, :state, :duration, :event_type, :previous_state, :pull_request_title, :pull_request_number, :started_at, :finished_at, :private)
+    representation(:minimal,  :id, :number, :state, :duration, :event_type, :previous_state, :pull_request_title, :pull_request_number, :started_at, :finished_at, :private, :priority)
     representation(:standard, *representations[:minimal], :repository, :branch, :tag, :commit, :jobs, :stages, :created_by, :updated_at)
     representation(:active, *representations[:standard])
 
     hidden_representations(:active)
 
+    HIGH_PRIORITY = 5
+
     def self.available_attributes
       super + ['request', 'log_complete']
     end
@@ -44,6 +46,10 @@ def log_complete
       end
     end
 
+    def priority
+      model.jobs.where(priority: HIGH_PRIORITY).present?
+    end
+
     private def created_by_href(creator)
       case creator
       when V3::Models::Organization then Renderer.href(:organization, script_name: script_name, id: creator.id)

lib/travis/api/v3/routes.rb

@@ -15,6 +15,7 @@ module Routes
 
       post :cancel, '/cancel'
       post :restart, '/restart'
+      post :priority, '/priority'
 
       resource :jobs do
         route '/jobs'

lib/travis/api/v3/services/build/cancel.rb

@@ -5,7 +5,7 @@ def run
       build = check_login_and_find(:build)
       access_control.permissions(build).cancel!
 
-      query.cancel(access_control.user)
+      query.cancel(access_control.user, build.id)
       accepted(build: build, state_change: :cancel)
     end
   end