extend cmd line support to accept multiple arguments

Signed-off-by: Henry Avetisyan <hga@yahooinc.com>
AthenZ · Sep 11, 2024 · 98eaebe · 98eaebe
1 parent 2e61197
commit 98eaebe
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 2 deletions.
diff --git a/libs/go/sia/agent/agent.go b/libs/go/sia/agent/agent.go
@@ -601,7 +601,14 @@ func SetupAgent(opts *options.Options, siaMainDir, siaLinkDir string) {
 	}
 }
 
-func RunAgent(siaCmd, ztsUrl string, opts *options.Options) {
+func RunAgent(siaCmds, ztsUrl string, opts *options.Options) {
+	cmds := strings.Split(siaCmds, ",")
+	for _, cmd := range cmds {
+		runAgentCommand(cmd, ztsUrl, opts)
+	}
+}
+
+func runAgentCommand(siaCmd, ztsUrl string, opts *options.Options) {
 
 	//make sure the meta endpoint is configured by the caller
 	if opts.MetaEndPoint == "" {

diff --git a/libs/go/sia/aws/agent/agent.go b/libs/go/sia/aws/agent/agent.go
@@ -597,7 +597,14 @@ func SetupAgent(opts *options.Options, siaMainDir, siaLinkDir string) {
 	}
 }
 
-func RunAgent(siaCmd, ztsUrl string, opts *options.Options) {
+func RunAgent(siaCmds, ztsUrl string, opts *options.Options) {
+	cmds := strings.Split(siaCmds, ",")
+	for _, cmd := range cmds {
+		runAgentCommand(cmd, ztsUrl, opts)
+	}
+}
+
+func runAgentCommand(siaCmd, ztsUrl string, opts *options.Options) {
 
 	//make sure the meta endpoint is configured by the caller
 	if opts.MetaEndPoint == "" {

diff --git a/pom.xml b/pom.xml
@@ -178,6 +178,7 @@
     <module>provider/buildkite/sia-buildkite</module>
     <module>provider/gcp/sia-gce</module>
     <module>provider/gcp/sia-gke</module>
+    <module>provider/gcp/sia-run</module>
     <module>provider/github/sia-actions</module>
     <module>utils/zms-cli</module>
     <module>utils/athenz-conf</module>