From b6a2ab804e1dfdc30c0489603d9f2190a885ff63 Mon Sep 17 00:00:00 2001
From: wyike <77846369+wangyikewxgm@users.noreply.github.com>
Date: Mon, 16 Aug 2021 20:30:56 +0800
Subject: [PATCH] add batch partition (#2097)

---
 charts/vela-core/templates/defwithtemplate/rollout.yaml | 2 ++
 vela-templates/definitions/internal/rollout.cue         | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/charts/vela-core/templates/defwithtemplate/rollout.yaml b/charts/vela-core/templates/defwithtemplate/rollout.yaml
index d9e77b57e..07ae4fca4 100644
--- a/charts/vela-core/templates/defwithtemplate/rollout.yaml
+++ b/charts/vela-core/templates/defwithtemplate/rollout.yaml
@@ -26,6 +26,7 @@ spec:
         			rolloutStrategy: "IncreaseFirst"
         			rolloutBatches:  parameter.rolloutBatches
         			targetSize:      parameter.targetSize
+        			batchPartition:  parameter.batchPartition
         		}
         	}
         }
@@ -33,6 +34,7 @@ spec:
         	targetRevision: *context.revision | string
         	targetSize:     int
         	rolloutBatches: [...rolloutBatch]
+        	batchPartition?: int
         }
         rolloutBatch: replicas: int
   skipRevisionAffect: true
diff --git a/vela-templates/definitions/internal/rollout.cue b/vela-templates/definitions/internal/rollout.cue
index 8ad5f7947..521660987 100644
--- a/vela-templates/definitions/internal/rollout.cue
+++ b/vela-templates/definitions/internal/rollout.cue
@@ -23,6 +23,7 @@ template: {
 				rolloutStrategy: "IncreaseFirst"
 				rolloutBatches:  parameter.rolloutBatches
 				targetSize:      parameter.targetSize
+				batchPartition:  parameter.batchPartition
 			}
 		}
 	}
@@ -31,6 +32,7 @@ template: {
 		targetRevision: *context.revision | string
 		targetSize:     int
 		rolloutBatches: [...rolloutBatch]
+		batchPartition?: int
 	}
 
 	rolloutBatch: replicas: int