tornadocontrib/bsc
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

log: support custom time format configuration (#1401)

Browse Source
This commit is contained in:
dylanhuang 2023-04-03 10:22:49 +08:00 committed by GitHub
parent af7f0dd30b
commit 3b547a9855
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 40 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
log/format.go
View File

@ -14,9 +14,12 @@ import (
"unicode/utf8"
)
var (
timeFormat = "2006-01-02T15:04:05-0700"
termTimeFormat = "01-02|15:04:05.000"
)
const (
timeFormat = "2006-01-02T15:04:05-0700"
termTimeFormat = "01-02|15:04:05.000"
floatFormat = 'f'
termMsgJust = 40
termCtxMaxPadding = 40
@ -482,3 +485,11 @@ func escapeString(s string) string {
}
return strconv.Quote(s)
}
func SetTermTimeFormat(format string) {
termTimeFormat = format
}
func SetTimeFormat(format string) {
timeFormat = format
}

13
node/config.go
View File

@ -502,8 +502,13 @@ func (c *Config) warnOnce(w *bool, format string, args ...interface{}) {
}
type LogConfig struct {
FileRoot string
FilePath string
MaxBytesSize uint
Level string
FileRoot *string
FilePath *string
MaxBytesSize *uint
Level *string
// TermTimeFormat is the time format used for console logging.
TermTimeFormat *string
// TimeFormat is the time format used for file logging.
TimeFormat *string
}

24
node/node.go
View File

@ -86,13 +86,25 @@ func New(conf *Config) (*Node, error) {
conf.DataDir = absdatadir
}
if conf.LogConfig != nil {
logFilePath := ""
if conf.LogConfig.FileRoot == "" {
logFilePath = path.Join(conf.DataDir, conf.LogConfig.FilePath)
} else {
logFilePath = path.Join(conf.LogConfig.FileRoot, conf.LogConfig.FilePath)
if conf.LogConfig.TermTimeFormat != nil && *conf.LogConfig.TermTimeFormat != "" {
log.SetTermTimeFormat(*conf.LogConfig.TermTimeFormat)
}
if conf.LogConfig.TimeFormat != nil && *conf.LogConfig.TimeFormat != "" {
log.SetTimeFormat(*conf.LogConfig.TimeFormat)
}
if conf.LogConfig.FileRoot != nil && conf.LogConfig.FilePath != nil &&
conf.LogConfig.MaxBytesSize != nil && conf.LogConfig.Level != nil {
// log to file
logFilePath := ""
if *conf.LogConfig.FileRoot == "" {
logFilePath = path.Join(conf.DataDir, *conf.LogConfig.FilePath)
} else {
logFilePath = path.Join(*conf.LogConfig.FileRoot, *conf.LogConfig.FilePath)
}
log.Root().SetHandler(log.NewFileLvlHandler(logFilePath, *conf.LogConfig.MaxBytesSize, *conf.LogConfig.Level))
}
log.Root().SetHandler(log.NewFileLvlHandler(logFilePath, conf.LogConfig.MaxBytesSize, conf.LogConfig.Level))
}
if conf.Logger == nil {
conf.Logger = log.New()