You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
35 lines
885 B
35 lines
885 B
3 years ago
|
package log
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"github.com/alibaba/RedisShake/internal/config"
|
||
|
|
"github.com/rs/zerolog"
|
||
|
|
"os"
|
||
|
|
)
|
||
|
|
|
||
|
|
var logger zerolog.Logger
|
||
|
|
|
||
|
|
func Init() {
|
||
|
|
|
||
|
|
// log level
|
||
|
|
switch config.Config.Advanced.LogLevel {
|
||
|
|
case "debug":
|
||
|
|
zerolog.SetGlobalLevel(zerolog.DebugLevel)
|
||
|
|
case "info":
|
||
|
|
zerolog.SetGlobalLevel(zerolog.InfoLevel)
|
||
|
|
case "warn":
|
||
|
|
zerolog.SetGlobalLevel(zerolog.WarnLevel)
|
||
|
|
default:
|
||
|
|
panic(fmt.Sprintf("unknown log level: %s", config.Config.Advanced.LogLevel))
|
||
|
|
}
|
||
|
|
|
||
|
|
// log file
|
||
|
|
consoleWriter := zerolog.ConsoleWriter{Out: os.Stdout, TimeFormat: "2006-01-02 15:04:05"}
|
||
|
|
fileWriter, err := os.OpenFile(config.Config.Advanced.LogFile, os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0666)
|
||
|
|
if err != nil {
|
||
|
|
panic(fmt.Sprintf("open log file failed: %s", err))
|
||
|
|
}
|
||
|
|
multi := zerolog.MultiLevelWriter(consoleWriter, fileWriter)
|
||
|
|
logger = zerolog.New(multi).With().Timestamp().Logger()
|
||
|
|
}
|