blackteay/gin
3
0
Fork 0
mirror of https://github.com/gin-gonic/gin.git synced 2025-10-17 22:32:26 +08:00
Code Issues Projects Releases Wiki Activity

fix #2505, quick more naive version

Browse Source
This commit is contained in:
Tomasz Gałkowski 2020-10-01 22:58:34 +02:00
parent 540b1eff70
commit 7772aa912e
3 changed files with 21 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
logger.go
View File

@ -33,7 +33,7 @@ const (
reset = "\033[0m"
)
var consoleColorMode = autoColor
var ConsoleColorMode = autoColor
// LoggerConfig defines the config for Logger middleware.
type LoggerConfig struct {
@ -125,7 +125,7 @@ func (p *LogFormatterParams) ResetColor() string {
// IsOutputColor indicates whether can colors be outputted to the log.
func (p *LogFormatterParams) IsOutputColor() bool {
return consoleColorMode == forceColor || (consoleColorMode == autoColor && p.isTerm)
return ConsoleColorMode == forceColor || (ConsoleColorMode == autoColor && p.isTerm)
}
// defaultLogFormatter is the default log format function Logger middleware uses.
@ -154,12 +154,12 @@ var defaultLogFormatter = func(param LogFormatterParams) string {
// DisableConsoleColor disables color output in the console.
func DisableConsoleColor() {
consoleColorMode = disableColor
ConsoleColorMode = disableColor
}
// ForceConsoleColor force color output in the console.
func ForceConsoleColor() {
consoleColorMode = forceColor
ConsoleColorMode = forceColor
}
// ErrorLogger returns a handlerfunc for any error type.

18
logger_test.go
View File

@ -326,7 +326,7 @@ func TestIsOutputColor(t *testing.T) {
isTerm: true,
}
consoleColorMode = autoColor
ConsoleColorMode = autoColor
assert.Equal(t, true, p.IsOutputColor())
ForceConsoleColor()
@ -340,7 +340,7 @@ func TestIsOutputColor(t *testing.T) {
isTerm: false,
}
consoleColorMode = autoColor
ConsoleColorMode = autoColor
assert.Equal(t, false, p.IsOutputColor())
ForceConsoleColor()
@ -350,7 +350,7 @@ func TestIsOutputColor(t *testing.T) {
assert.Equal(t, false, p.IsOutputColor())
// reset console color mode.
consoleColorMode = autoColor
ConsoleColorMode = autoColor
}
func TestErrorLogger(t *testing.T) {
@ -415,20 +415,20 @@ func TestLoggerWithConfigSkippingPaths(t *testing.T) {
func TestDisableConsoleColor(t *testing.T) {
New()
assert.Equal(t, autoColor, consoleColorMode)
assert.Equal(t, autoColor, ConsoleColorMode)
DisableConsoleColor()
assert.Equal(t, disableColor, consoleColorMode)
assert.Equal(t, disableColor, ConsoleColorMode)
// reset console color mode.
consoleColorMode = autoColor
ConsoleColorMode = autoColor
}
func TestForceConsoleColor(t *testing.T) {
New()
assert.Equal(t, autoColor, consoleColorMode)
assert.Equal(t, autoColor, ConsoleColorMode)
ForceConsoleColor()
assert.Equal(t, forceColor, consoleColorMode)
assert.Equal(t, forceColor, ConsoleColorMode)
// reset console color mode.
consoleColorMode = autoColor
ConsoleColorMode = autoColor
}

9
recovery.go
View File

@ -50,8 +50,15 @@ func RecoveryWithWriter(out io.Writer, recovery ...RecoveryFunc) HandlerFunc {
// CustomRecoveryWithWriter returns a middleware for a given writer that recovers from any panics and calls the provided handle func to handle it.
func CustomRecoveryWithWriter(out io.Writer, handle RecoveryFunc) HandlerFunc {
var logger *log.Logger
// If gin.DisableConsoleColor() is invoked, respect it.
prefix := "\n\n"
if ConsoleColorMode != 1 {
prefix = "\n\n\x1b[31m"
}
if out != nil {
logger = log.New(out, "\n\n\x1b[31m", log.LstdFlags)
logger = log.New(out, prefix, log.LstdFlags)
}
return func(c *Context) {
defer func() {