mirror of
https://github.com/yusing/godoxy.git
synced 2026-03-18 15:34:38 +01:00
53f3397b7a
* Add comprehensive post-request rules support for response phase * Enable response body, status, and header manipulation via set commands * Refactor command handlers to support both request and response phases * Implement response modifier system for post-request template execution * Support response-based rule matching with status and header checks * Add comprehensive benchmarks for matcher performance * Refactor authentication and proxying commands for unified error handling * Support negated conditions with ! * Enhance error handling, error formatting and validation * Routes: add `rule_file` field with rule preset support * Environment variable substitution: now supports variables without `GODOXY_` prefix * new conditions: * `on resp_header <key> [<value>]` * `on status <status>` * new commands: * `require_auth` * `set resp_header <key> <template>` * `set resp_body <template>` * `set status <code>` * `log <level> <path> <template>` * `notify <level> <provider> <title_template> <body_template>`
92 lines
1.9 KiB
Go
92 lines
1.9 KiB
Go
package logging
|
|
|
|
import (
|
|
"io"
|
|
"log"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/rs/zerolog"
|
|
"github.com/yusing/godoxy/internal/common"
|
|
strutils "github.com/yusing/goutils/strings"
|
|
|
|
zerologlog "github.com/rs/zerolog/log"
|
|
)
|
|
|
|
var (
|
|
logger zerolog.Logger
|
|
timeFmt string
|
|
level zerolog.Level
|
|
prefix string
|
|
)
|
|
|
|
func init() {
|
|
switch {
|
|
case common.IsTrace:
|
|
timeFmt = "04:05"
|
|
level = zerolog.TraceLevel
|
|
case common.IsDebug:
|
|
timeFmt = "01-02 15:04"
|
|
level = zerolog.DebugLevel
|
|
default:
|
|
timeFmt = "01-02 15:04"
|
|
level = zerolog.InfoLevel
|
|
}
|
|
prefixLength := len(timeFmt) + 5 // level takes 3 + 2 spaces
|
|
prefix = strings.Repeat(" ", prefixLength)
|
|
InitLogger(os.Stdout)
|
|
}
|
|
|
|
func fmtMessage(msg string) string {
|
|
lines := strutils.SplitRune(msg, '\n')
|
|
if len(lines) == 1 {
|
|
return msg
|
|
}
|
|
for i := 1; i < len(lines); i++ {
|
|
lines[i] = prefix + lines[i]
|
|
}
|
|
return strutils.JoinRune(lines, '\n')
|
|
}
|
|
|
|
func NewLogger(out ...io.Writer) zerolog.Logger {
|
|
writer := zerolog.ConsoleWriter{
|
|
Out: zerolog.MultiLevelWriter(out...),
|
|
TimeFormat: timeFmt,
|
|
FormatMessage: func(msgI interface{}) string { // pad spaces for each line
|
|
return fmtMessage(msgI.(string))
|
|
},
|
|
}
|
|
return zerolog.New(
|
|
writer,
|
|
).Level(level).With().Timestamp().Logger()
|
|
}
|
|
|
|
func NewLoggerWithFixedLevel(level zerolog.Level, out ...io.Writer) zerolog.Logger {
|
|
levelStr := level.String()
|
|
writer := zerolog.ConsoleWriter{
|
|
Out: zerolog.MultiLevelWriter(out...),
|
|
TimeFormat: timeFmt,
|
|
FormatMessage: func(msgI interface{}) string { // pad spaces for each line
|
|
if msgI == nil {
|
|
return ""
|
|
}
|
|
return fmtMessage(msgI.(string))
|
|
},
|
|
FormatLevel: func(_ any) string {
|
|
return levelStr
|
|
},
|
|
}
|
|
return zerolog.New(
|
|
writer,
|
|
).Level(level).With().Timestamp().Logger()
|
|
}
|
|
|
|
func InitLogger(out ...io.Writer) {
|
|
logger = NewLogger(out...)
|
|
log.SetOutput(logger)
|
|
log.SetPrefix("")
|
|
log.SetFlags(0)
|
|
zerolog.TimeFieldFormat = timeFmt
|
|
zerologlog.Logger = logger
|
|
}
|