1 files changed, 14 insertions, 0 deletions
diff --git a/app/models/deployment.rb b/app/models/deployment.rb
index 811e623b7f7..9e46b3938cf 100644
--- a/app/models/deployment.rb
+++ b/app/models/deployment.rb
@@ -16,6 +16,11 @@ class Deployment < ActiveRecord::Base
 
   validates :sha, presence: true
   validates :ref, presence: true
+  validates :track, presence: true
+  validates :rollout, presence: true, if: :rollout?,
+    numericality: { greater_than: 0, less_than: 100 }
+  validates :rollout, presence: true, unless: :rollout?,
+    numericality: { equal_to: 100 }
 
   delegate :name, to: :environment, prefix: true
 
@@ -66,6 +71,15 @@ class Deployment < ActiveRecord::Base
     find(ids)
   end
 
+  enum track: {
+    stable: 0,
+    rollout: 1
+  }
+
+  def incremental_rollout?
+    rollout? && rollout.to_i < 100
+  end
+
   def commit
     project.commit(sha)
   end