mirror of
https://git.sr.ht/~ashkeel/strimertul
synced 2024-09-18 01:50:50 +00:00
feat: add log-level flag
This commit is contained in:
parent
bb8f3f754e
commit
668620a090
2 changed files with 12 additions and 5 deletions
|
@ -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
9
main.go
|
@ -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 {
|
||||||
|
|
Loading…
Reference in a new issue