Pass replicas from RestoreSession to Function (#848)

stashed · Jul 29, 2019 · 8df5be9 · 8df5be9
1 parent ae5eabd
commit 8df5be9
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/apis/constants.go b/apis/constants.go
@@ -23,6 +23,7 @@ const (
 	TargetMountPath   = "TARGET_MOUNT_PATH"
 	TargetDirectories = "TARGET_DIRECTORIES"
 	TargetAppVersion  = "TARGET_APP_VERSION"
+	TargetAppReplicas = "TARGET_APP_REPLICAS"
 
 	RestoreDirectories = "RESTORE_DIRECTORIES"
 	RestoreSnapshots   = "RESTORE_SNAPSHOTS"

diff --git a/pkg/controller/inputs.go b/pkg/controller/inputs.go
@@ -52,6 +52,13 @@ func (c *StashController) inputsForRestoreSession(restoreSession api.RestoreSess
 	// always enable cache if nothing specified
 	inputs[apis.EnableCache] = strconv.FormatBool(!restoreSession.Spec.TempDir.DisableCaching)
 
+	// pass replicas field to function. if not set pass default 1.
+	replicas := int32(1)
+	if restoreSession.Spec.Target != nil && restoreSession.Spec.Target.Replicas != nil {
+		replicas = *restoreSession.Spec.Target.Replicas
+	}
+	inputs[apis.TargetAppReplicas] = string(replicas)
+
 	// add PushgatewayURL as input
 	metricInputs := c.inputForMetrics()
 	inputs = core_util.UpsertMap(inputs, metricInputs)