From 5263cba44a9978e89d24134f6ce431a833260c71 Mon Sep 17 00:00:00 2001 From: loveuer Date: Mon, 12 Aug 2024 16:14:16 +0800 Subject: [PATCH] chore: expose default logger chore: add ctx alias method --- ctx.go | 7 +++++++ nft/log/default.go | 18 +++++++++--------- 2 files changed, 16 insertions(+), 9 deletions(-) diff --git a/ctx.go b/ctx.go index a341ed3..1e19ccb 100644 --- a/ctx.go +++ b/ctx.go @@ -295,10 +295,17 @@ func (c *Ctx) Status(code int) *Ctx { return c } +// Set set response header func (c *Ctx) Set(key string, value string) { c.Writer.Header().Set(key, value) } +// AddHeader add response header +func (c *Ctx) AddHeader(key string, value string) { + c.Writer.Header().Add(key, value) +} + +// SetHeader set response header func (c *Ctx) SetHeader(key string, value string) { c.Writer.Header().Set(key, value) } diff --git a/nft/log/default.go b/nft/log/default.go index 6ab24ee..7ecf5e7 100644 --- a/nft/log/default.go +++ b/nft/log/default.go @@ -21,7 +21,7 @@ var ( os.Exit(1) } - defaultLogger = &logger{ + DefaultLogger = &logger{ Mutex: sync.Mutex{}, timeFormat: "2006-01-02T15:04:05", writer: os.Stdout, @@ -36,32 +36,32 @@ var ( ) func SetTimeFormat(format string) { - defaultLogger.SetTimeFormat(format) + DefaultLogger.SetTimeFormat(format) } func SetLogLevel(level LogLevel) { - defaultLogger.SetLogLevel(level) + DefaultLogger.SetLogLevel(level) } func Debug(msg string, data ...any) { - defaultLogger.Debug(msg, data...) + DefaultLogger.Debug(msg, data...) } func Info(msg string, data ...any) { - defaultLogger.Info(msg, data...) + DefaultLogger.Info(msg, data...) } func Warn(msg string, data ...any) { - defaultLogger.Warn(msg, data...) + DefaultLogger.Warn(msg, data...) } func Error(msg string, data ...any) { - defaultLogger.Error(msg, data...) + DefaultLogger.Error(msg, data...) } func Panic(msg string, data ...any) { - defaultLogger.Panic(msg, data...) + DefaultLogger.Panic(msg, data...) } func Fatal(msg string, data ...any) { - defaultLogger.Fatal(msg, data...) + DefaultLogger.Fatal(msg, data...) }