Skip to content

Commit e6343da

Browse files
committed
feat: Allow configuring log level via env
1 parent d1d623c commit e6343da

File tree

1 file changed

+14
-4
lines changed

1 file changed

+14
-4
lines changed

main.go

Lines changed: 14 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@ package main
33
import (
44
"os"
55
"os/signal"
6+
"strings"
67
"syscall"
78
"time"
89

@@ -73,17 +74,29 @@ func catchInterrupt() {
7374
}
7475

7576
func initLogging() {
76-
7777
log.SetFormatter(&logrus.TextFormatter{
7878
FullTimestamp: true,
7979
})
80+
81+
var logLevel = logrus.InfoLevel
82+
83+
logLevelEnv := strings.ToLower(os.Getenv("SENTLOG_LOG_LEVEL"))
84+
switch logLevelEnv {
85+
case "debug":
86+
logLevel = logrus.DebugLevel
87+
case "info":
88+
logLevel = logrus.InfoLevel
89+
}
90+
log.SetLevel(logLevel)
8091
}
8192

8293
func showGreeting() {
8394

8495
}
8596

8697
func main() {
98+
initLogging()
99+
87100
args := CmdArgs{
88101
file: kingpin.Arg("file", "File to parse").String(),
89102
pattern: kingpin.Flag("pattern", "Pattern to look for").Short('p').String(),
@@ -93,11 +106,8 @@ func main() {
93106
config: kingpin.Flag("config", "Path to the configuration").Short('c').String(),
94107
verbose: kingpin.Flag("verbose", "Print every match").Short('v').Default("false").Bool(),
95108
}
96-
97109
kingpin.Parse()
98110

99-
initLogging()
100-
101111
showGreeting()
102112

103113
_isDryRun = *args.dryRun

0 commit comments

Comments
 (0)