Merge pull request #226 from kanisterio/scale-workload-func

Update ScaleWorkload to provide replicas as string & int value

pkg/function/scale_test.go

@@ -101,7 +101,7 @@ func newScaleBlueprint(kind string) *crv1alpha1.Blueprint {
 						Name: "testScale",
 						Func: "ScaleWorkload",
 						Args: map[string]interface{}{
-							ScaleWorkloadReplicas: 2,
+							ScaleWorkloadReplicas: "2",
 						},
 					},
 				},
@@ -234,7 +234,7 @@ func (s *ScaleSuite) TestGetArgs(c *C) {
 		{
 			tp: param.TemplateParams{},
 			args: map[string]interface{}{
-				ScaleWorkloadReplicas:     2,
+				ScaleWorkloadReplicas:     "2",
 				ScaleWorkloadNamespaceArg: "foo",
 				ScaleWorkloadNameArg:      "app",
 				ScaleWorkloadKindArg:      param.StatefulSetKind,
@@ -269,7 +269,7 @@ func (s *ScaleSuite) TestGetArgs(c *C) {
 				},
 			},
 			args: map[string]interface{}{
-				ScaleWorkloadReplicas: 2,
+				ScaleWorkloadReplicas: "2",
 			},
 			wantKind:      param.DeploymentKind,
 			wantName:      "app",

pkg/function/scale_workload.go

@@ -2,6 +2,7 @@ package function
 
 import (
 	"context"
+	"strconv"
 	"strings"
 
 	"github.com/pkg/errors"
@@ -59,11 +60,27 @@ func (*scaleWorkloadFunc) RequiredArgs() []string {
 }
 
 func getArgs(tp param.TemplateParams, args map[string]interface{}) (namespace, kind, name string, replicas int32, err error) {
-	err = Arg(args, ScaleWorkloadReplicas, &replicas)
+	var rep interface{}
+	err = Arg(args, ScaleWorkloadReplicas, &rep)
 	if err != nil {
 		return namespace, kind, name, replicas, err
 	}
 
+	var val int
+	switch rep.(type) {
+	case int:
+		val = rep.(int)
+	case string:
+		if val, err = strconv.Atoi(rep.(string)); err != nil {
+			err = errors.Wrapf(err, "Cannot convert %s to int ", rep.(string))
+			return
+		}
+	default:
+		err = errors.Errorf("Invalid arg type %T for Arg %s ", rep, ScaleWorkloadReplicas)
+		return
+	}
+
+	replicas = int32(val)
 	// Populate default values for optional arguments from template parameters
 	switch {
 	case tp.StatefulSet != nil: