mirror of
https://github.com/gofiber/fiber.git
synced 2025-02-22 06:33:09 +00:00
132 lines
3.4 KiB
Go
132 lines
3.4 KiB
Go
package logger
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"sync"
|
|
|
|
"github.com/gofiber/fiber/v3"
|
|
"github.com/gofiber/utils/v2"
|
|
"github.com/mattn/go-colorable"
|
|
"github.com/mattn/go-isatty"
|
|
"github.com/valyala/bytebufferpool"
|
|
"github.com/valyala/fasthttp"
|
|
)
|
|
|
|
var mu sync.Mutex
|
|
|
|
// default logger for fiber
|
|
func defaultLoggerInstance(c fiber.Ctx, data *Data, cfg Config) error {
|
|
// Alias colors
|
|
colors := c.App().Config().ColorScheme
|
|
|
|
// Get new buffer
|
|
buf := bytebufferpool.Get()
|
|
|
|
// Default output when no custom Format or io.Writer is given
|
|
if cfg.Format == defaultFormat {
|
|
// Format error if exist
|
|
formatErr := ""
|
|
if cfg.enableColors {
|
|
if data.ChainErr != nil {
|
|
formatErr = colors.Red + " | " + data.ChainErr.Error() + colors.Reset
|
|
}
|
|
_, _ = buf.WriteString( //nolint:errcheck // This will never fail
|
|
fmt.Sprintf("%s |%s %3d %s| %13v | %15s |%s %-7s %s| %-"+data.ErrPaddingStr+"s %s\n",
|
|
data.Timestamp.Load().(string),
|
|
statusColor(c.Response().StatusCode(), colors), c.Response().StatusCode(), colors.Reset,
|
|
data.Stop.Sub(data.Start),
|
|
c.IP(),
|
|
methodColor(c.Method(), colors), c.Method(), colors.Reset,
|
|
c.Path(),
|
|
formatErr,
|
|
),
|
|
)
|
|
} else {
|
|
if data.ChainErr != nil {
|
|
formatErr = " | " + data.ChainErr.Error()
|
|
}
|
|
_, _ = buf.WriteString( //nolint:errcheck // This will never fail
|
|
fmt.Sprintf("%s | %3d | %13v | %15s | %-7s | %-"+data.ErrPaddingStr+"s %s\n",
|
|
data.Timestamp.Load().(string),
|
|
c.Response().StatusCode(),
|
|
data.Stop.Sub(data.Start),
|
|
c.IP(),
|
|
c.Method(),
|
|
c.Path(),
|
|
formatErr,
|
|
),
|
|
)
|
|
}
|
|
|
|
// Write buffer to output
|
|
_, _ = cfg.Output.Write(buf.Bytes()) //nolint:errcheck // This will never fail
|
|
|
|
if cfg.Done != nil {
|
|
cfg.Done(c, buf.Bytes())
|
|
}
|
|
|
|
// Put buffer back to pool
|
|
bytebufferpool.Put(buf)
|
|
|
|
// End chain
|
|
return nil
|
|
}
|
|
|
|
var err error
|
|
// Loop over template parts execute dynamic parts and add fixed parts to the buffer
|
|
for i, logFunc := range data.LogFuncChain {
|
|
if logFunc == nil {
|
|
_, _ = buf.Write(data.TemplateChain[i]) //nolint:errcheck // This will never fail
|
|
} else if data.TemplateChain[i] == nil {
|
|
_, err = logFunc(buf, c, data, "")
|
|
} else {
|
|
_, err = logFunc(buf, c, data, utils.UnsafeString(data.TemplateChain[i]))
|
|
}
|
|
if err != nil {
|
|
break
|
|
}
|
|
}
|
|
|
|
// Also write errors to the buffer
|
|
if err != nil {
|
|
_, _ = buf.WriteString(err.Error()) //nolint:errcheck // This will never fail
|
|
}
|
|
mu.Lock()
|
|
// Write buffer to output
|
|
if _, err := cfg.Output.Write(buf.Bytes()); err != nil {
|
|
// Write error to output
|
|
if _, err := cfg.Output.Write([]byte(err.Error())); err != nil {
|
|
// There is something wrong with the given io.Writer
|
|
_, _ = fmt.Fprintf(os.Stderr, "Failed to write to log, %v\n", err)
|
|
}
|
|
}
|
|
mu.Unlock()
|
|
|
|
if cfg.Done != nil {
|
|
cfg.Done(c, buf.Bytes())
|
|
}
|
|
|
|
// Put buffer back to pool
|
|
bytebufferpool.Put(buf)
|
|
|
|
return nil
|
|
}
|
|
|
|
// run something before returning the handler
|
|
func beforeHandlerFunc(cfg Config) {
|
|
// If colors are enabled, check terminal compatibility
|
|
if cfg.enableColors {
|
|
cfg.Output = colorable.NewColorableStdout()
|
|
if os.Getenv("TERM") == "dumb" || os.Getenv("NO_COLOR") == "1" || (!isatty.IsTerminal(os.Stdout.Fd()) && !isatty.IsCygwinTerminal(os.Stdout.Fd())) {
|
|
cfg.Output = colorable.NewNonColorable(os.Stdout)
|
|
}
|
|
}
|
|
}
|
|
|
|
func appendInt(output Buffer, v int) (int, error) {
|
|
old := output.Len()
|
|
output.Set(fasthttp.AppendUint(output.Bytes(), v))
|
|
return output.Len() - old, nil
|
|
}
|