Add flag for number periods gathering/posting data (#176)

* Add flag for number periods gathering/posting data

Number of periods is integer.
Takes -1 as default (runs perpetually) - Any -ve will do this

Signed-off-by: Jamie Leppard <[email protected]>

* Adds one-shot flag to run agent a single time

Signed-off-by: Jamie Leppard <[email protected]>

* One-shot flag w/ break and no short option

Signed-off-by: Jamie Leppard <[email protected]>

Co-authored-by: jamie <JammyL>
Co-authored-by: Jamie Leppard <[email protected]>

Author: JammyL

cmd/agent.go

@@ -57,4 +57,12 @@ func init() {
 		"",
 		"Location of the credentials file. For OAuth2 based authentication.",
 	)
+	agentCmd.PersistentFlags().BoolVarP(
+		&agent.OneShot,
+		"one-shot",
+		"",
+		false,
+		"Runs agent a single time if true, or continously if false",
+	)
+
 }

pkg/agent/run.go

@@ -28,15 +28,21 @@ var AuthToken string
 // Period is the number of seconds between scans
 var Period uint
 
+// Agent will run only once if OneShot is enabled
+var OneShot bool
+
 // CredentialsPath is where the agent will try to loads the credentials. (Experimental)
 var CredentialsPath string
 
+
 // Run starts the agent process
 func Run(cmd *cobra.Command, args []string) {
 	ctx := context.Background()
-
 	for {
 		gatherAndPostData(ctx)
+		if OneShot {
+			break
+		}
 		time.Sleep(time.Duration(Period) * time.Second)
 	}
 }