robocars/robocar-steering-tflite-edgetpu
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: log configuration

Browse Source
This commit is contained in:
Cyrille Nofficial
2021-10-12 17:34:47 +02:00
parent 61d8bd22c4
commit 0001c3e36e
2 changed files with 34 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
cmd/rc-steering/rc-steering.go
View File

@ -18,12 +18,8 @@ func main() {
var cameraTopic, steeringTopic string
var modelPath string
var edgeVerbosity int
var debug bool
lgr, err := zap.NewProductionConfig().Build()
if err != nil {
log.Fatalf("unable to init logger: %v", err)
}
zap.ReplaceGlobals(lgr)
mqttQos := cli.InitIntFlag("MQTT_QOS", 0)
_, mqttRetain := os.LookupEnv("MQTT_RETAIN")
@ -34,6 +30,7 @@ func main() {
flag.StringVar(&steeringTopic, "mqtt-topic-road", os.Getenv("MQTT_TOPIC_STEERING"), "Mqtt topic to publish road detection result, use MQTT_TOPIC_STEERING if args not set")
flag.StringVar(&cameraTopic, "mqtt-topic-camera", os.Getenv("MQTT_TOPIC_CAMERA"), "Mqtt topic that contains camera frame values, use MQTT_TOPIC_CAMERA if args not set")
flag.IntVar(&edgeVerbosity, "edge-verbosity", 0, "Edge TPU Verbosity")
flag.BoolVar(&debug, "debug", false, "Display debug logs")
flag.Parse()
if len(os.Args) <= 1 {
@ -41,6 +38,23 @@ func main() {
os.Exit(1)
}
config := zap.NewDevelopmentConfig()
if debug {
config.Level = zap.NewAtomicLevelAt(zap.DebugLevel)
} else {
config.Level = zap.NewAtomicLevelAt(zap.InfoLevel)
}
lgr, err := config.Build()
if err != nil {
log.Fatalf("unable to init logger: %v", err)
}
defer func() {
if err := lgr.Sync(); err != nil {
log.Printf("unable to Sync logger: %v\n", err)
}
}()
zap.ReplaceGlobals(lgr)
if modelPath == "" {
zap.L().Error("model path is mandatory")
flag.PrintDefaults()