mirror of https://github.com/dnomd343/XProxy.git
Dnomd343
1 year ago
5 changed files with 258 additions and 4 deletions
@ -0,0 +1,99 @@ |
|||||
|
package logger |
||||
|
|
||||
|
import ( |
||||
|
"bytes" |
||||
|
"fmt" |
||||
|
"github.com/gookit/color" |
||||
|
"go.uber.org/zap/zapcore" |
||||
|
"path/filepath" |
||||
|
"runtime" |
||||
|
"strconv" |
||||
|
"strings" |
||||
|
"time" |
||||
|
) |
||||
|
|
||||
|
// getGid get goroutine ID only for debugging.
|
||||
|
// -> https://blog.sgmansfield.com/2015/12/goroutine-ids/
|
||||
|
func getGid() uint64 { |
||||
|
b := make([]byte, 64) |
||||
|
b = b[:runtime.Stack(b, false)] |
||||
|
b = bytes.TrimPrefix(b, []byte("goroutine ")) |
||||
|
b = b[:bytes.IndexByte(b, ' ')] |
||||
|
n, _ := strconv.ParseUint(string(b), 10, 64) |
||||
|
return n |
||||
|
} |
||||
|
|
||||
|
// getCaller calculate relative source path of caller.
|
||||
|
func getCaller(ec zapcore.EntryCaller, verbose bool) string { |
||||
|
file, err := filepath.Rel(logHandle.path, ec.File) |
||||
|
if err != nil { |
||||
|
return "undefined" |
||||
|
} |
||||
|
if verbose { |
||||
|
return file + ":" + strconv.Itoa(ec.Line) |
||||
|
} |
||||
|
file, _ = strings.CutSuffix(file, ".go") // remove `.go` suffix
|
||||
|
return file |
||||
|
} |
||||
|
|
||||
|
// timeEncoder formats the time as a string.
|
||||
|
func timeEncoder(t time.Time, enc zapcore.PrimitiveArrayEncoder) { |
||||
|
enc.AppendString(t.Format("2006-01-02 15:04:05.000")) |
||||
|
} |
||||
|
|
||||
|
// timeColoredEncoder formats the time as a colored string
|
||||
|
// with custom prefix.
|
||||
|
func timeColoredEncoder(t time.Time, enc zapcore.PrimitiveArrayEncoder) { |
||||
|
enc.AppendString(fmt.Sprintf( |
||||
|
"%s %s", |
||||
|
color.Cyan.Render(logHandle.prefix), // colored prefix
|
||||
|
color.Gray.Render(t.Format("2006-01-02 15:04:05.000")), |
||||
|
)) |
||||
|
} |
||||
|
|
||||
|
// callerEncoder formats caller in square brackets.
|
||||
|
func callerEncoder(ec zapcore.EntryCaller, enc zapcore.PrimitiveArrayEncoder) { |
||||
|
if !logHandle.verbose { |
||||
|
enc.AppendString("[" + getCaller(ec, false) + "]") |
||||
|
return |
||||
|
} |
||||
|
enc.AppendString(fmt.Sprintf("[%d] [%s]", getGid(), getCaller(ec, true))) |
||||
|
} |
||||
|
|
||||
|
// callerColoredEncoder formats caller in square brackets with
|
||||
|
// magenta color.
|
||||
|
func callerColoredEncoder(ec zapcore.EntryCaller, enc zapcore.PrimitiveArrayEncoder) { |
||||
|
if !logHandle.verbose { |
||||
|
enc.AppendString(color.Magenta.Render("[" + getCaller(ec, false) + "]")) |
||||
|
return |
||||
|
} |
||||
|
enc.AppendString(fmt.Sprintf( |
||||
|
"%s %s", |
||||
|
color.Blue.Render(fmt.Sprintf("[%d]", getGid())), |
||||
|
color.Magenta.Render("["+getCaller(ec, true)+"]"), |
||||
|
)) |
||||
|
} |
||||
|
|
||||
|
// levelEncoder formats log level using square brackets.
|
||||
|
func levelEncoder(level zapcore.Level, enc zapcore.PrimitiveArrayEncoder) { |
||||
|
enc.AppendString("[" + level.CapitalString() + "]") |
||||
|
} |
||||
|
|
||||
|
// levelColoredEncoder formats log level using square brackets
|
||||
|
// and uses different colors.
|
||||
|
func levelColoredEncoder(level zapcore.Level, enc zapcore.PrimitiveArrayEncoder) { |
||||
|
levelStr := "[" + level.CapitalString() + "]" |
||||
|
switch level { |
||||
|
case zapcore.DebugLevel: |
||||
|
levelStr = color.FgDefault.Render(levelStr) |
||||
|
case zapcore.InfoLevel: |
||||
|
levelStr = color.Green.Render(levelStr) |
||||
|
case zapcore.WarnLevel: |
||||
|
levelStr = color.Yellow.Render(levelStr) |
||||
|
case zapcore.ErrorLevel: |
||||
|
levelStr = color.Red.Render(levelStr) |
||||
|
case zapcore.PanicLevel: |
||||
|
levelStr = color.LightRed.Render(levelStr) |
||||
|
} |
||||
|
enc.AppendString(levelStr) |
||||
|
} |
||||
@ -0,0 +1,59 @@ |
|||||
|
package logger |
||||
|
|
||||
|
import ( |
||||
|
"go.uber.org/zap/zapcore" |
||||
|
"io" |
||||
|
) |
||||
|
|
||||
|
const ( |
||||
|
DebugLevel = zapcore.DebugLevel |
||||
|
InfoLevel = zapcore.InfoLevel |
||||
|
WarnLevel = zapcore.WarnLevel |
||||
|
ErrorLevel = zapcore.ErrorLevel |
||||
|
PanicLevel = zapcore.PanicLevel |
||||
|
) |
||||
|
|
||||
|
func Debugf(template string, args ...interface{}) { |
||||
|
logHandle.sugar.Debugf(template, args...) |
||||
|
} |
||||
|
|
||||
|
func Infof(template string, args ...interface{}) { |
||||
|
logHandle.sugar.Infof(template, args...) |
||||
|
} |
||||
|
|
||||
|
func Warnf(template string, args ...interface{}) { |
||||
|
logHandle.sugar.Warnf(template, args...) |
||||
|
} |
||||
|
|
||||
|
func Errorf(template string, args ...interface{}) { |
||||
|
logHandle.sugar.Errorf(template, args...) |
||||
|
} |
||||
|
|
||||
|
func Panicf(template string, args ...interface{}) { |
||||
|
logHandle.sugar.Panicf(template, args...) |
||||
|
} |
||||
|
|
||||
|
// GetLevel return the current logger level.
|
||||
|
func GetLevel() zapcore.Level { |
||||
|
return logHandle.level.Level() |
||||
|
} |
||||
|
|
||||
|
// SetLevel configure logger output level. Note that debug level
|
||||
|
// will output more information and reduce performance.
|
||||
|
func SetLevel(level zapcore.Level) { |
||||
|
logHandle.level.SetLevel(level) |
||||
|
if level == DebugLevel { |
||||
|
logHandle.verbose = true |
||||
|
} else { |
||||
|
logHandle.verbose = false |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// AddOutputs adds more plain output channel to the logger module.
|
||||
|
func AddOutputs(outputs ...io.Writer) { |
||||
|
var writers []zapcore.WriteSyncer |
||||
|
for _, output := range outputs { |
||||
|
writers = append(writers, zapcore.AddSync(output)) |
||||
|
} |
||||
|
addWrites(writers...) |
||||
|
} |
||||
@ -0,0 +1,78 @@ |
|||||
|
package logger |
||||
|
|
||||
|
import ( |
||||
|
"go.uber.org/zap" |
||||
|
"go.uber.org/zap/zapcore" |
||||
|
"os" |
||||
|
"path" |
||||
|
"runtime" |
||||
|
) |
||||
|
|
||||
|
type logger struct { |
||||
|
logger *zap.Logger |
||||
|
level *zap.AtomicLevel |
||||
|
sugar *zap.SugaredLogger |
||||
|
writers []zapcore.WriteSyncer |
||||
|
stderr zapcore.Core // fixed stderr output
|
||||
|
prefix string // custom output prefix
|
||||
|
path string // project absolute path
|
||||
|
verbose bool // show goroutine id and caller line
|
||||
|
} |
||||
|
|
||||
|
var logHandle *logger // singleton logger handle
|
||||
|
|
||||
|
// logConfig generates log config for XProxy.
|
||||
|
func logConfig(colored bool) zapcore.EncoderConfig { |
||||
|
config := zapcore.EncoderConfig{ |
||||
|
ConsoleSeparator: " ", |
||||
|
MessageKey: "msg", |
||||
|
LevelKey: "level", |
||||
|
TimeKey: "time", |
||||
|
CallerKey: "caller", |
||||
|
EncodeTime: timeEncoder, |
||||
|
EncodeLevel: levelEncoder, |
||||
|
EncodeCaller: callerEncoder, |
||||
|
} |
||||
|
if colored { |
||||
|
config.EncodeTime = timeColoredEncoder |
||||
|
config.EncodeLevel = levelColoredEncoder |
||||
|
config.EncodeCaller = callerColoredEncoder |
||||
|
} |
||||
|
return config |
||||
|
} |
||||
|
|
||||
|
func init() { |
||||
|
zapLevel := zap.NewAtomicLevelAt(InfoLevel) // using info level in default
|
||||
|
zapCore := zapcore.NewCore( |
||||
|
zapcore.NewConsoleEncoder(logConfig(true)), // colorful output
|
||||
|
zapcore.Lock(os.Stderr), |
||||
|
zapLevel, |
||||
|
) |
||||
|
zapLogger := zap.New(zapCore, zap.AddCaller(), zap.AddCallerSkip(1)) |
||||
|
_, src, _, _ := runtime.Caller(0) // absolute path of current code
|
||||
|
logHandle = &logger{ |
||||
|
logger: zapLogger, |
||||
|
level: &zapLevel, |
||||
|
stderr: zapCore, |
||||
|
sugar: zapLogger.Sugar(), |
||||
|
writers: []zapcore.WriteSyncer{}, |
||||
|
path: path.Join(path.Dir(src), "../"), |
||||
|
prefix: "[XProxy]", |
||||
|
verbose: false, |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// addWrites adds more plain log writers.
|
||||
|
func addWrites(writers ...zapcore.WriteSyncer) { |
||||
|
logHandle.writers = append(logHandle.writers, writers...) |
||||
|
plainCore := zapcore.NewCore( |
||||
|
zapcore.NewConsoleEncoder(logConfig(false)), // without colored
|
||||
|
zap.CombineWriteSyncers(logHandle.writers...), |
||||
|
logHandle.level, |
||||
|
) |
||||
|
logHandle.logger = zap.New( |
||||
|
zapcore.NewTee(logHandle.stderr, plainCore), |
||||
|
zap.AddCaller(), zap.AddCallerSkip(1), |
||||
|
) |
||||
|
logHandle.sugar = logHandle.logger.Sugar() |
||||
|
} |
||||
Loading…
Reference in new issue