feat: agent reports workloads and handles workload commands

2026-08-07 08:58:46 +01:00
parent ac093bcf5c
commit 6c6b8a8a4a
3 changed files with 185 additions and 0 deletions
@@ -70,6 +70,8 @@ func Run(ctx context.Context, cfg *config.Config, version string) error {

 	go runInventory(ctx, cfg)

+	go runWorkloads(ctx, cfg)
+
 	go monitors.Run(ctx, cfg)

 	ticker := time.NewTicker(cfg.PollInterval)
@@ -347,6 +349,15 @@ func connectAndHandleStream(ctx context.Context, cfg *config.Config) error {
 		if cmd.OpenProxy != nil {
 			go handleOpenProxy(ctx, cfg, cmd.OpenProxy)
 		}
+		if cmd.RefreshWorkloads != nil {
+			go handleRefreshWorkloads(cfg)
+		}
+		if cmd.ControlWorkload != nil {
+			go handleControlWorkload(send, cfg, cmd.CommandId, cmd.ControlWorkload)
+		}
+		if cmd.WorkloadLogs != nil {
+			go handleWorkloadLogs(send, cfg, cmd.CommandId, cmd.WorkloadLogs)
+		}
 		if cmd.RunStep != nil {
 			go func(rc *pb.RunStepCmd, cid string) {
 				emit := func(seq uint64, data []byte) {