customize log level (#30)

Signed-off-by: gang.liu <[email protected]>
argoproj-labs · Aug 23, 2023 · bd02ed5 · bd02ed5
1 parent 83a4f22
commit bd02ed5
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 5 deletions.
diff --git a/main.go b/main.go
@@ -1,6 +1,7 @@
 package main
 
 import (
+	"flag"
 	"log/slog"
 
 	"github.com/argoproj-labs/rollouts-plugin-trafficrouter-contour/pkg/plugin"
@@ -20,8 +21,12 @@ var handshakeConfig = goPlugin.HandshakeConfig{
 	MagicCookieValue: "trafficrouter",
 }
 
+var lvl = flag.Int("l", int(slog.LevelInfo), "the logging level for 'log/slog', (default: 0)")
+
 func main() {
-	utils.InitLogger()
+	flag.Parse()
+
+	utils.InitLogger(slog.Level(*lvl))
 
 	rpcPluginImp := &plugin.RpcPlugin{}
 

diff --git a/pkg/utils/utils.go b/pkg/utils/utils.go
@@ -36,11 +36,11 @@ func Must1[T any](t T, err error) T {
 	return t
 }
 
-func InitLogger() {
-	lvl := &slog.LevelVar{}
-	lvl.Set(slog.LevelDebug)
+func InitLogger(lvl slog.Level) {
+	lvlVar := &slog.LevelVar{}
+	lvlVar.Set(lvl)
 	opts := slog.HandlerOptions{
-		Level: lvl,
+		Level: lvlVar,
 	}
 
 	attrs := []slog.Attr{