mirror of
https://github.com/d0zingcat/gocryptotrader.git
synced 2026-05-19 23:16:48 +00:00
112 lines
2.6 KiB
Go
112 lines
2.6 KiB
Go
package log
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
var (
|
|
errEmptyLoggerName = errors.New("cannot have empty logger name")
|
|
errSubLoggerAlreadyregistered = errors.New("sub logger already registered")
|
|
)
|
|
|
|
// NewSubLogger allows for a new sub logger to be registered.
|
|
func NewSubLogger(name string) (*SubLogger, error) {
|
|
if name == "" {
|
|
return nil, errEmptyLoggerName
|
|
}
|
|
name = strings.ToUpper(name)
|
|
if _, ok := SubLoggers[name]; ok {
|
|
return nil, errSubLoggerAlreadyregistered
|
|
}
|
|
return registerNewSubLogger(name), nil
|
|
}
|
|
|
|
// SetOutput overrides the default output with a new writer
|
|
func (s *SubLogger) SetOutput(o io.Writer) {
|
|
RWM.Lock()
|
|
defer RWM.Unlock()
|
|
|
|
s.output = o
|
|
}
|
|
|
|
func newLogger(c *Config) *Logger {
|
|
return &Logger{
|
|
Timestamp: c.AdvancedSettings.TimeStampFormat,
|
|
Spacer: c.AdvancedSettings.Spacer,
|
|
ErrorHeader: c.AdvancedSettings.Headers.Error,
|
|
InfoHeader: c.AdvancedSettings.Headers.Info,
|
|
WarnHeader: c.AdvancedSettings.Headers.Warn,
|
|
DebugHeader: c.AdvancedSettings.Headers.Debug,
|
|
ShowLogSystemName: *c.AdvancedSettings.ShowLogSystemName,
|
|
}
|
|
}
|
|
|
|
func (l *Logger) newLogEvent(data, header, slName string, w io.Writer) error {
|
|
if w == nil {
|
|
return errors.New("io.Writer not set")
|
|
}
|
|
|
|
e, ok := eventPool.Get().(*Event)
|
|
if !ok {
|
|
return errors.New("unable to type asset event")
|
|
}
|
|
e.output = w
|
|
e.data = append(e.data, []byte(header)...)
|
|
if l.ShowLogSystemName {
|
|
e.data = append(e.data, l.Spacer...)
|
|
e.data = append(e.data, slName...)
|
|
}
|
|
e.data = append(e.data, l.Spacer...)
|
|
if l.Timestamp != "" {
|
|
e.data = time.Now().AppendFormat(e.data, l.Timestamp)
|
|
}
|
|
e.data = append(e.data, l.Spacer...)
|
|
e.data = append(e.data, []byte(data)...)
|
|
if data == "" || data[len(data)-1] != '\n' {
|
|
e.data = append(e.data, '\n')
|
|
}
|
|
_, err := e.output.Write(e.data)
|
|
|
|
e.data = e.data[:0]
|
|
eventPool.Put(e)
|
|
|
|
return err
|
|
}
|
|
|
|
// CloseLogger is called on shutdown of application
|
|
func CloseLogger() error {
|
|
return GlobalLogFile.Close()
|
|
}
|
|
|
|
func validSubLogger(s string) (bool, *SubLogger) {
|
|
if v, found := SubLoggers[s]; found {
|
|
return true, v
|
|
}
|
|
return false, nil
|
|
}
|
|
|
|
// Level retries the current sublogger levels
|
|
func Level(s string) (*Levels, error) {
|
|
found, logger := validSubLogger(s)
|
|
if !found {
|
|
return nil, fmt.Errorf("logger %v not found", s)
|
|
}
|
|
|
|
return &logger.Levels, nil
|
|
}
|
|
|
|
// SetLevel sets sublogger levels
|
|
func SetLevel(s, level string) (*Levels, error) {
|
|
found, logger := validSubLogger(s)
|
|
if !found {
|
|
return nil, fmt.Errorf("logger %v not found", s)
|
|
}
|
|
logger.Levels = splitLevel(level)
|
|
|
|
return &logger.Levels, nil
|
|
}
|