*: add --runner-sa flag to rg server subcmd

It's used to override existing runner group's server account. Signed-off-by: Wei Fu <weifu@microsoft.com>
Azure · Jan 29, 2024 · 69a9708 · 69a9708
1 parent e5c5ff0
commit 69a9708
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/cmd/kperf/commands/runnergroup/server.go b/cmd/kperf/commands/runnergroup/server.go
@@ -35,6 +35,10 @@ var serverCommand = cli.Command{
 			Name:  "runner-owner",
 			Usage: "The runners depend on this object (FORMAT: APIVersion:Kind:Name:UID)",
 		},
+		cli.StringFlag{
+			Name:  "runner-sa",
+			Usage: "Override runner group's service account",
+		},
 		cli.StringSliceFlag{
 			Name:     "address",
 			Usage:    "Address for the server",
@@ -89,6 +93,11 @@ func buildRunnerGroupHandlers(cliCtx *cli.Context, serverName string) ([]*runner
 		ownerRef = cliCtx.String("runner-owner")
 	}
 
+	sa := ""
+	if cliCtx.IsSet("runner-sa") {
+		sa = cliCtx.String("runner-sa")
+	}
+
 	groups := make([]*runnergroup.Handler, 0, len(specURIs))
 	for idx, specURI := range specURIs {
 		spec, err := runnergroup.NewRunnerGroupSpecFromURI(clientset, specURI)
@@ -100,6 +109,10 @@ func buildRunnerGroupHandlers(cliCtx *cli.Context, serverName string) ([]*runner
 			spec.OwnerReference = &ownerRef
 		}
 
+		if sa != "" {
+			spec.ServiceAccount = &sa
+		}
+
 		groupName := fmt.Sprintf("%s-%d", serverName, idx)
 		g, err := runnergroup.NewHandler(clientset, namespace, groupName, spec, imgRef)
 		if err != nil {

diff --git a/manifests/runnergroup/server/templates/pod.yaml b/manifests/runnergroup/server/templates/pod.yaml
@@ -18,6 +18,8 @@ spec:
     - {{ .Values.image }}
     - --runner-owner
     - v1:Pod:$(POD_NAME):$(POD_UID)
+    - --runner-sa
+    - {{ .Values.name }}
     - --address
     - $(POD_IP):8080
     - --address