cmd/tskagent: add --update flag

tailscale · Oct 21, 2024 · ac54fe2 · ac54fe2
1 parent 0e1882d
commit ac54fe2
Showing 1 changed file with 15 additions and 3 deletions.
diff --git a/cmd/tskagent/tskagent.go b/cmd/tskagent/tskagent.go
@@ -14,6 +14,7 @@ import (
 	"os"
 	"os/signal"
 	"syscall"
+	"time"
 
 	"github.com/creachadair/command"
 	"github.com/creachadair/flax"
@@ -24,9 +25,10 @@ import (
 )
 
 var flags struct {
-	Server string `flag:"server,Secret server address (required)"`
-	Socket string `flag:"socket,Agent socket path (required)"`
-	Prefix string `flag:"prefix,Secret name prefix (required)"`
+	Server string        `flag:"server,Secret server address (required)"`
+	Socket string        `flag:"socket,Agent socket path (required)"`
+	Prefix string        `flag:"prefix,Secret name prefix (required)"`
+	Update time.Duration `flag:"update,Automatic update interval (0 means no updates)"`
 }
 
 func main() {
@@ -69,6 +71,16 @@ func run(env *command.Env) error {
 	if err := srv.Update(env.Context()); err != nil {
 		return fmt.Errorf("initialize agent: %w", err)
 	}
+	if flags.Update > 0 {
+		go func() {
+			for range time.NewTicker(flags.Update).C {
+				if err := srv.Update(env.Context()); err != nil {
+					log.Printf("WARNING: Update failed: %v", err)
+				}
+			}
+		}()
+		log.Printf("Enabled periodic updates (%v)", flags.Update)
+	}
 
 	var g taskgroup.Group
 	g.Run(func() {