Modify functionRunner

Author: yuwenma

go/fn/examples/data/setlabels-configmap.yaml

@@ -0,0 +1,15 @@
+apiVersion: config.kubernetes.io/v1
+kind: ResourceList
+items:
+  - apiVersion: v1
+    kind: Service
+    metadata:
+      name: example
+functionConfig:
+  apiVersion: v1
+  kind: ConfigMap
+  metadata:
+    name: setlabel
+  data:
+    owner: kpt
+    org: google

go/fn/examples/data/setlabels-resourcelist.yaml renamed to go/fn/examples/data/setlabels-customFnConfig.yaml

@@ -10,3 +10,5 @@ functionConfig:
   kind: SetLabels
   metadata:
     name: setlabel-fn-config
+  owner: kpt
+  org: google

go/fn/examples/example_asmain_runner_test.go

@@ -22,40 +22,26 @@ import (
 var _ fn.Runner = &SetLabels{}
 
 type SetLabels struct {
-	Labels map[string]string `json:"labels,omitempty"`
+	Owner string
+	Org   string
 }
 
 // Run is the main function logic.
 // `ctx` provides easy methods to add info, error or warning result to `ResourceList.Results`.
 // `items` is parsed from the STDIN "ResourceList.Items".
 // `functionConfig` is from the STDIN "ResourceList.FunctionConfig". The value has been assigned to the r.Labels
 //  the functionConfig is validated to have kind "SetLabels" and apiVersion "fn.kpt.dev/v1alpha1"
-func (r *SetLabels) Run(ctx *fn.Context, functionConfig *fn.KubeObject, items []*fn.KubeObject) {
+func (r *SetLabels) Run(ctx *fn.Context, functionConfig *fn.KubeObject, items fn.KubeObjects) {
 	for _, o := range items {
-		for k, newLabel := range r.Labels {
-			o.SetLabel(k, newLabel)
-		}
+		o.SetLabel("owner", r.Owner)
+		o.SetLabel("org", r.Org)
 	}
 	ctx.ResultInfo("updated labels", nil)
 }
 
 // This example uses a SetLabels object, which implements `Runner.Run` methods.
-//
-// The input from ./data/setlabels-resourcelist.yaml:
-// apiVersion: config.kubernetes.io/v1
-// kind: ResourceList
-// items:
-//   - apiVersion: v1
-//     kind: Service
-//     metadata:
-//       name: example
-// functionConfig:
-//   apiVersion: fn.kpt.dev/v1alpha1
-//   kind: SetLabels
-//   metadata:
-//     name: setlabel-fn-config
-func Example_asMain() {
-	file, _ := os.Open("./data/setlabels-resourcelist.yaml")
+func Example_asMainCustomFnConfig() {
+	file, _ := os.Open("./data/setlabels-customFnConfig.yaml")
 	defer file.Close()
 	os.Stdin = file
 
@@ -70,11 +56,48 @@ func Example_asMain() {
 	//   kind: Service
 	//   metadata:
 	//     name: example
+	//     labels:
+	//       owner: kpt
+	//       org: google
 	// functionConfig:
 	//   apiVersion: fn.kpt.dev/v1alpha1
 	//   kind: SetLabels
 	//   metadata:
 	//     name: setlabel-fn-config
+	//   owner: kpt
+	//   org: google
+	// results:
+	// - message: updated labels
+	//   severity: info
+}
+
+func Example_asMainConfigMap() {
+	file, _ := os.Open("./data/setlabels-configmap.yaml")
+	defer file.Close()
+	os.Stdin = file
+
+	if err := fn.AsMain(&SetLabels{}); err != nil {
+		os.Exit(1)
+	}
+	// Output:
+	// apiVersion: config.kubernetes.io/v1
+	// kind: ResourceList
+	// items:
+	// - apiVersion: v1
+	//   kind: Service
+	//   metadata:
+	//     name: example
+	//     labels:
+	//       owner: kpt
+	//       org: google
+	// functionConfig:
+	//   apiVersion: v1
+	//   kind: ConfigMap
+	//   metadata:
+	//     name: setlabel
+	//   data:
+	//     owner: kpt
+	//     org: google
 	// results:
 	// - message: updated labels
 	//   severity: info

go/fn/functionrunner.go

@@ -15,5 +15,5 @@
 package fn
 
 type Runner interface {
-	Run(context *Context, functionConfig *KubeObject, items []*KubeObject)
+	Run(context *Context, functionConfig *KubeObject, items KubeObjects)
 }

go/fn/runnerProcessor.go

@@ -17,6 +17,7 @@ package fn
 import (
 	"fmt"
 	"reflect"
+	"strings"
 )
 
 type runnerProcessor struct {
@@ -39,9 +40,19 @@ func (r *runnerProcessor) config(ctx *Context, o *KubeObject) {
 		data := o.NestedStringMapOrDie("data")
 		fnRunnerElem := reflect.ValueOf(r.fnRunner).Elem()
 		for i := 0; i < fnRunnerElem.NumField(); i++ {
-			if fnRunnerElem.Field(i).Kind() == reflect.Map {
-				fnRunnerElem.Field(i).Set(reflect.ValueOf(data))
-				break
+			switch fnRunnerElem.Field(i).Kind() {
+			case reflect.String:
+				for k, v := range data {
+					lowerKey := strings.ToLower(k)
+					if lowerKey == strings.ToLower(fnRunnerElem.Type().Field(i).Name) {
+						fnRunnerElem.Field(i).SetString(v)
+						break
+					}
+				}
+			case reflect.Map:
+				if "data" == strings.ToLower(fnRunnerElem.Type().Field(i).Name) {
+					fnRunnerElem.Field(i).Set(reflect.ValueOf(data))
+				}
 			}
 		}
 	case o.IsGVK("fn.kpt.dev", "v1alpha1", fnName):