mirror of
https://github.com/Ed1s0nZ/CyberStrikeAI.git
synced 2026-08-19 01:17:16 +02:00
Delete logger directory
This commit is contained in:
@@ -1,68 +0,0 @@
|
|||||||
package logger
|
|
||||||
|
|
||||||
import (
|
|
||||||
"os"
|
|
||||||
|
|
||||||
"go.uber.org/zap"
|
|
||||||
"go.uber.org/zap/zapcore"
|
|
||||||
)
|
|
||||||
|
|
||||||
type Logger struct {
|
|
||||||
*zap.Logger
|
|
||||||
}
|
|
||||||
|
|
||||||
func New(level, output string) *Logger {
|
|
||||||
var zapLevel zapcore.Level
|
|
||||||
switch level {
|
|
||||||
case "debug":
|
|
||||||
zapLevel = zapcore.DebugLevel
|
|
||||||
case "info":
|
|
||||||
zapLevel = zapcore.InfoLevel
|
|
||||||
case "warn":
|
|
||||||
zapLevel = zapcore.WarnLevel
|
|
||||||
case "error":
|
|
||||||
zapLevel = zapcore.ErrorLevel
|
|
||||||
default:
|
|
||||||
zapLevel = zapcore.InfoLevel
|
|
||||||
}
|
|
||||||
|
|
||||||
config := zap.NewProductionConfig()
|
|
||||||
config.Level = zap.NewAtomicLevelAt(zapLevel)
|
|
||||||
config.EncoderConfig.TimeKey = "timestamp"
|
|
||||||
config.EncoderConfig.EncodeTime = zapcore.ISO8601TimeEncoder
|
|
||||||
|
|
||||||
var writeSyncer zapcore.WriteSyncer
|
|
||||||
if output == "stdout" {
|
|
||||||
writeSyncer = zapcore.AddSync(os.Stdout)
|
|
||||||
} else {
|
|
||||||
file, err := os.OpenFile(output, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0666)
|
|
||||||
if err != nil {
|
|
||||||
writeSyncer = zapcore.AddSync(os.Stdout)
|
|
||||||
} else {
|
|
||||||
writeSyncer = zapcore.AddSync(file)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
core := zapcore.NewCore(
|
|
||||||
zapcore.NewJSONEncoder(config.EncoderConfig),
|
|
||||||
writeSyncer,
|
|
||||||
zapLevel,
|
|
||||||
)
|
|
||||||
|
|
||||||
logger := zap.New(core, zap.AddCaller(), zap.AddStacktrace(zapcore.ErrorLevel))
|
|
||||||
|
|
||||||
return &Logger{Logger: logger}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (l *Logger) Fatal(msg string, fields ...interface{}) {
|
|
||||||
zapFields := make([]zap.Field, 0, len(fields))
|
|
||||||
for _, f := range fields {
|
|
||||||
switch v := f.(type) {
|
|
||||||
case error:
|
|
||||||
zapFields = append(zapFields, zap.Error(v))
|
|
||||||
default:
|
|
||||||
zapFields = append(zapFields, zap.Any("field", v))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
l.Logger.Fatal(msg, zapFields...)
|
|
||||||
}
|
|
||||||
Reference in New Issue
Block a user