novalet base & golang cleanup

1 files changed, 25 insertions, 0 deletions

diff --git a/common/logger.go b/common/logger.go
new file mode 100644
index 0000000..31ee3db
--- /dev/null
+++ b/common/logger.go
@@ -0,0 +1,25 @@
+package common
+
+import (
+	"flag"
+	"os"
+
+	"github.com/rs/zerolog"
+	"github.com/rs/zerolog/log"
+)
+
+var (
+	debug  *bool = flag.Bool("debug", false, "enables the debug mode")
+	pretty *bool = flag.Bool("pretty", false, "enables the pretty log messages")
+)
+
+func SetupLogger() {
+	if *pretty {
+		log.Logger = log.Output(zerolog.ConsoleWriter{Out: os.Stderr})
+	}
+	zerolog.TimeFieldFormat = zerolog.TimeFormatUnix
+	zerolog.SetGlobalLevel(zerolog.InfoLevel)
+	if *debug {
+		zerolog.SetGlobalLevel(zerolog.DebugLevel)
+	}
+}