From cd5316bbb9c908395acbc0db71680e27cb537bea Mon Sep 17 00:00:00 2001 From: Sachin Kamboj Date: Mon, 6 Apr 2020 22:50:17 -0400 Subject: [PATCH] Allow debug logging Signed-off-by: Sachin Kamboj --- cmd/goldpinger/main.go | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/cmd/goldpinger/main.go b/cmd/goldpinger/main.go index b0919f5..df8fc7d 100644 --- a/cmd/goldpinger/main.go +++ b/cmd/goldpinger/main.go @@ -16,6 +16,7 @@ package main import ( "os" + "strconv" "github.com/go-openapi/loads" "go.uber.org/zap" @@ -35,7 +36,16 @@ var ( ) func getLogger() *zap.Logger { - logger, err := zap.NewProduction() + var logger *zap.Logger + var err error + + // We haven't parsed flags at this stage and that might be error prone + // so just use an envvar + if debug, err := strconv.ParseBool(os.Getenv("DEBUG")); err == nil && debug { + logger, err = zap.NewDevelopment() + } else { + logger, err = zap.NewProduction() + } if err != nil { panic(err) }