From 2cc7ee507044a80b8a55b727e4efd3e207d06ce9 Mon Sep 17 00:00:00 2001 From: Cyrille Nofficial Date: Mon, 3 Jan 2022 19:15:51 +0100 Subject: [PATCH] refacto(log): configure log level --- cmd/rc-camera/rc-camera.go | 13 ++++--------- 1 file changed, 4 insertions(+), 9 deletions(-) diff --git a/cmd/rc-camera/rc-camera.go b/cmd/rc-camera/rc-camera.go index 9cdc42a..30003cf 100644 --- a/cmd/rc-camera/rc-camera.go +++ b/cmd/rc-camera/rc-camera.go @@ -17,7 +17,6 @@ func main() { var mqttBroker, username, password, clientId, topicBase, topicRoi string var pubFrequency, horizon int var device, videoWidth, videoHeight int - var debug bool mqttQos := cli.InitIntFlag("MQTT_QOS", 0) _, mqttRetain := os.LookupEnv("MQTT_RETAIN") @@ -37,20 +36,16 @@ func main() { flag.IntVar(&horizon, "horizon", horizon, "Limit region of interest to horizon in pixels from top, use HORIZON if args not set") - flag.BoolVar(&debug, "debug", false, "Display raw value to debug") - + logLevel := zap.LevelFlag("log", zap.InfoLevel, "log level") flag.Parse() + if len(os.Args) <= 1 { flag.PrintDefaults() os.Exit(1) } config := zap.NewDevelopmentConfig() - if debug { - config.Level = zap.NewAtomicLevelAt(zap.DebugLevel) - } else { - config.Level = zap.NewAtomicLevelAt(zap.InfoLevel) - } + config.Level = zap.NewAtomicLevelAt(*logLevel) lgr, err := config.Build() if err != nil { log.Fatalf("unable to init logger: %v", err) @@ -73,7 +68,7 @@ func main() { videoProperties[gocv.VideoCaptureFrameHeight] = float64(videoHeight) if topicRoi == "" { - topicRoi = fmt.Sprintf( "%s-roi", topicBase) + topicRoi = fmt.Sprintf("%s-roi", topicBase) } c := camera.New(client, topicBase, topicRoi, pubFrequency, videoProperties, horizon) defer c.Stop()