Add --delay

Author: wzshiming

cmd/kube-apiserver-audit-exporter/main.go

@@ -4,6 +4,7 @@ import (
 	"log/slog"
 	"os"
 	"strings"
+	"time"
 
 	"github.com/spf13/pflag"
 	"github.com/wzshiming/kube-apiserver-audit-exporter/exporter"
@@ -14,32 +15,39 @@ var (
 	address      = ":8080"
 	cluster      = ""
 	replay       = false
+	delay        time.Duration
 )
 
 func init() {
 	pflag.StringArrayVar(&auditLogPath, "audit-log-path", auditLogPath, "Path to audit log files, path[:clusterName]")
 	pflag.StringVar(&address, "address", address, "Address to listen on")
 	pflag.StringVar(&cluster, "cluster-label", cluster, "Default cluster label of metrics")
 	pflag.BoolVar(&replay, "replay", replay, "replay the audit log")
+	pflag.DurationVar(&delay, "delay", 0, "delay to start")
 	pflag.Parse()
 }
 
 func main() {
-	for _, p := range auditLogPath {
-		ns := strings.SplitN(p, ":", 2)
-		path := ns[0]
-		clusterLabel := cluster
-		if len(ns) > 1 {
-			clusterLabel = ns[1]
+	go func() {
+		if delay > 0 {
+			time.Sleep(delay)
 		}
+		for _, p := range auditLogPath {
+			ns := strings.SplitN(p, ":", 2)
+			path := ns[0]
+			clusterLabel := cluster
+			if len(ns) > 1 {
+				clusterLabel = ns[1]
+			}
 
-		e := exporter.NewExporter(
-			exporter.WithReplay(replay),
-			exporter.WithFile(path),
-			exporter.WithClusterLabel(clusterLabel),
-		)
-		e.Start()
-	}
+			e := exporter.NewExporter(
+				exporter.WithReplay(replay),
+				exporter.WithFile(path),
+				exporter.WithClusterLabel(clusterLabel),
+			)
+			e.Start()
+		}
+	}()
 
 	if err := exporter.ListenAndServe(address); err != nil {
 		slog.Error("Failed", "err", err)