54 lines
1.3 KiB
Go
54 lines
1.3 KiB
Go
package application
|
|
|
|
import (
|
|
"log/slog"
|
|
|
|
"github.com/spf13/cobra"
|
|
|
|
"git.flexabyte.io/flexabyte/go-kit/slogd"
|
|
)
|
|
|
|
const (
|
|
LogOutputStdOut = "stdout"
|
|
LogOutputStdErr = "stderr"
|
|
LogOutputFile = "file"
|
|
)
|
|
|
|
var (
|
|
logLevelFlagName string = "log-level"
|
|
logLevelFlag string
|
|
logOutputFlagName string = "log-output"
|
|
logOutputFlag string
|
|
logTypeFlagName = "log-type"
|
|
logTypeFlag string
|
|
)
|
|
|
|
func addLogLevelFlag(cmd *cobra.Command) {
|
|
cmd.PersistentFlags().StringVarP(&logLevelFlag, logLevelFlagName, "", "info", "Set log level (trace, debug, info, warn, error, fatal)")
|
|
}
|
|
|
|
func addLogOutputFlag(cmd *cobra.Command) {
|
|
cmd.PersistentFlags().StringVarP(&logOutputFlag, logOutputFlagName, "", "stderr", "Set log output (stdout, stderr, file)")
|
|
}
|
|
|
|
func addLogTypeFlag(cmd *cobra.Command) {
|
|
cmd.PersistentFlags().StringVarP(&logTypeFlag, logTypeFlagName, "", "text", "Set log type (text, json, color)")
|
|
}
|
|
|
|
func configureLoggingFlags(cmd *cobra.Command) {
|
|
addLogLevelFlag(cmd)
|
|
addLogOutputFlag(cmd)
|
|
addLogTypeFlag(cmd)
|
|
|
|
cmd.MarkFlagsMutuallyExclusive("no-color", "log-type")
|
|
}
|
|
|
|
func GetLogLevelFromArgs(args []string) slog.Level {
|
|
for i, arg := range args {
|
|
if arg == "--log-level" && i+1 < len(args) {
|
|
return slogd.Level(args[i+1])
|
|
}
|
|
}
|
|
return slogd.LevelDefault
|
|
}
|