mirror of
https://github.com/project-zot/zot.git
synced 2024-12-16 21:56:37 -05:00
d2aa016cdb
Behavior controlled by configuration (default=off) It is a trade-off between performance and consistency. References: [1] https://github.com/golang/go/issues/20599 Signed-off-by: Ramkumar Chinchani <rchincha@cisco.com>
91 lines
1.7 KiB
Go
91 lines
1.7 KiB
Go
package log
|
|
|
|
import (
|
|
"os"
|
|
"runtime"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/rs/zerolog"
|
|
)
|
|
|
|
const defaultPerms = 0o0600
|
|
|
|
// Logger extends zerolog's Logger.
|
|
type Logger struct {
|
|
zerolog.Logger
|
|
}
|
|
|
|
func (l Logger) Println(v ...interface{}) {
|
|
l.Logger.Error().Msg("panic recovered")
|
|
}
|
|
|
|
func NewLogger(level string, output string) Logger {
|
|
zerolog.TimeFieldFormat = time.RFC3339Nano
|
|
|
|
lvl, err := zerolog.ParseLevel(level)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
zerolog.SetGlobalLevel(lvl)
|
|
|
|
var log zerolog.Logger
|
|
|
|
if output == "" {
|
|
log = zerolog.New(os.Stdout)
|
|
} else {
|
|
file, err := os.OpenFile(output, os.O_APPEND|os.O_WRONLY|os.O_CREATE, defaultPerms)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
log = zerolog.New(file)
|
|
}
|
|
|
|
return Logger{Logger: log.Hook(goroutineHook{}).With().Caller().Timestamp().Logger()}
|
|
}
|
|
|
|
func NewAuditLogger(level string, audit string) *Logger {
|
|
zerolog.TimeFieldFormat = time.RFC3339Nano
|
|
|
|
lvl, err := zerolog.ParseLevel(level)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
zerolog.SetGlobalLevel(lvl)
|
|
|
|
var auditLog zerolog.Logger
|
|
|
|
auditFile, err := os.OpenFile(audit, os.O_APPEND|os.O_WRONLY|os.O_CREATE, defaultPerms)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
auditLog = zerolog.New(auditFile)
|
|
|
|
return &Logger{Logger: auditLog.With().Timestamp().Logger()}
|
|
}
|
|
|
|
// GoroutineID adds goroutine-id to logs to help debug concurrency issues.
|
|
func GoroutineID() int {
|
|
var buf [64]byte
|
|
n := runtime.Stack(buf[:], false)
|
|
idField := strings.Fields(strings.TrimPrefix(string(buf[:n]), "goroutine "))[0]
|
|
|
|
id, err := strconv.Atoi(idField)
|
|
if err != nil {
|
|
return -1
|
|
}
|
|
|
|
return id
|
|
}
|
|
|
|
type goroutineHook struct{}
|
|
|
|
func (h goroutineHook) Run(e *zerolog.Event, level zerolog.Level, msg string) {
|
|
if level != zerolog.NoLevel {
|
|
e.Int("goroutine", GoroutineID())
|
|
}
|
|
}
|