hamcha/strimertul
1
0
Fork 0
mirror of https://git.sr.ht/~ashkeel/strimertul synced 2024-09-20 02:00:49 +00:00
Code Activity Actions

feat: add log-level flag

Browse source
This commit is contained in:
Ash Keel 2022-11-24 13:15:59 +01:00
parent bb8f3f754e
commit 668620a090
No known key found for this signature in database
GPG key ID: BAD8D93E7314ED3E
2 changed files with 12 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
logging.go
View file

@ -18,14 +18,14 @@ var (
incomingLogs chan LogEntry incomingLogs chan LogEntry
) )
func initLogger() { func initLogger(level zapcore.Level) {
lastLogs = make([]LogEntry, 0) lastLogs = make([]LogEntry, 0)
incomingLogs = make(chan LogEntry, 100) incomingLogs = make(chan LogEntry, 100)
logStorage := NewLogStorage(zap.InfoLevel) logStorage := NewLogStorage(level)
consoleLogger := zapcore.NewCore( consoleLogger := zapcore.NewCore(
zapcore.NewConsoleEncoder(zap.NewDevelopmentEncoderConfig()), zapcore.NewConsoleEncoder(zap.NewDevelopmentEncoderConfig()),
zapcore.Lock(os.Stderr), zapcore.Lock(os.Stderr),
zap.InfoLevel, level,
) )
fileLogger := zapcore.NewCore( fileLogger := zapcore.NewCore(
zapcore.NewJSONEncoder(zap.NewProductionEncoderConfig()), zapcore.NewJSONEncoder(zap.NewProductionEncoderConfig()),
@ -35,7 +35,7 @@ func initLogger() {
MaxBackups: 3, MaxBackups: 3,
MaxAge: 28, MaxAge: 28,
}), }),
zap.DebugLevel, level,
) )
core := zapcore.NewTee( core := zapcore.NewTee(
consoleLogger, consoleLogger,

9
main.go
View file

@ -8,6 +8,8 @@ import (
"os" "os"
"time" "time"
"go.uber.org/zap/zapcore"
"github.com/strimertul/strimertul/modules" "github.com/strimertul/strimertul/modules"
"github.com/strimertul/strimertul/modules/loyalty" "github.com/strimertul/strimertul/modules/loyalty"
"github.com/strimertul/strimertul/modules/twitch" "github.com/strimertul/strimertul/modules/twitch"
@ -45,6 +47,7 @@ func main() {
Version: appVersion, Version: appVersion,
Action: cliMain, Action: cliMain,
Flags: []cli.Flag{ Flags: []cli.Flag{
&cli.StringFlag{Name: "log-level", Usage: "logging level (debug,info,warn,error)", Value: "info"},
&cli.StringFlag{Name: "driver", Usage: "specify database driver", Value: "auto"}, &cli.StringFlag{Name: "driver", Usage: "specify database driver", Value: "auto"},
&cli.StringFlag{Name: "database-dir", Aliases: []string{"db-dir"}, Usage: "specify database directory", Value: "data"}, &cli.StringFlag{Name: "database-dir", Aliases: []string{"db-dir"}, Usage: "specify database directory", Value: "data"},
&cli.StringFlag{Name: "backup-dir", Aliases: []string{"b-dir"}, Usage: "specify backup directory", Value: "backups"}, &cli.StringFlag{Name: "backup-dir", Aliases: []string{"b-dir"}, Usage: "specify backup directory", Value: "backups"},
@ -85,7 +88,11 @@ func main() {
rand.Seed(time.Now().UnixNano()) rand.Seed(time.Now().UnixNano())
// Initialize logger with global flags // Initialize logger with global flags
initLogger() level, err := zapcore.ParseLevel(ctx.String("log-level"))
if err != nil {
level = zapcore.InfoLevel
}
initLogger(level)
return nil return nil
}, },
After: func(ctx *cli.Context) error { After: func(ctx *cli.Context) error {