Logger skip paths use Context.fullPath

This is more flexible and is breaking only in
very rare cases.

Restore original behavior

Update logger path skipping tests
This commit is contained in:
Hanyu Cui 2022-05-05 22:31:01 -07:00
parent 90e7073d56
commit 40991b4a82
2 changed files with 27 additions and 6 deletions

View File

@ -234,13 +234,14 @@ func LoggerWithConfig(conf LoggerConfig) HandlerFunc {
// Start timer
start := time.Now()
path := c.Request.URL.Path
fullPath := c.FullPath()
raw := c.Request.URL.RawQuery
// Process request
c.Next()
// Log only when path is not being skipped
if _, ok := skip[path]; !ok {
// Log only when fullPath is not being skipped
if _, ok := skip[fullPath]; !ok {
param := LogFormatterParams{
Request: c.Request,
isTerm: isTerm,

View File

@ -384,16 +384,26 @@ func TestErrorLogger(t *testing.T) {
func TestLoggerWithWriterSkippingPaths(t *testing.T) {
buffer := new(bytes.Buffer)
router := New()
router.Use(LoggerWithWriter(buffer, "/skipped"))
router.Use(LoggerWithWriter(buffer, "/skipped", "/skippedUsers/:id"))
router.GET("/logged", func(c *Context) {})
router.GET("/loggedUsers/:id", func(c *Context) {})
router.GET("/skipped", func(c *Context) {})
router.GET("/skippedUsers/:id", func(c *Context) {})
PerformRequest(router, "GET", "/logged")
assert.Contains(t, buffer.String(), "200")
buffer.Reset()
PerformRequest(router, "GET", "/loggedUsers/2")
assert.Contains(t, buffer.String(), "200")
buffer.Reset()
PerformRequest(router, "GET", "/skipped")
assert.Contains(t, buffer.String(), "")
assert.Equal(t, "", buffer.String())
buffer.Reset()
PerformRequest(router, "GET", "/skippedUsers/3")
assert.Equal(t, "", buffer.String())
}
func TestLoggerWithConfigSkippingPaths(t *testing.T) {
@ -401,17 +411,27 @@ func TestLoggerWithConfigSkippingPaths(t *testing.T) {
router := New()
router.Use(LoggerWithConfig(LoggerConfig{
Output: buffer,
SkipPaths: []string{"/skipped"},
SkipPaths: []string{"/skipped", "/skippedUsers/:id"},
}))
router.GET("/logged", func(c *Context) {})
router.GET("/loggedUsers/:id", func(c *Context) {})
router.GET("/skipped", func(c *Context) {})
router.GET("/skippedUsers/:id", func(c *Context) {})
PerformRequest(router, "GET", "/logged")
assert.Contains(t, buffer.String(), "200")
buffer.Reset()
PerformRequest(router, "GET", "/loggedUsers/2")
assert.Contains(t, buffer.String(), "200")
buffer.Reset()
PerformRequest(router, "GET", "/skipped")
assert.Contains(t, buffer.String(), "")
assert.Equal(t, "", buffer.String())
buffer.Reset()
PerformRequest(router, "GET", "/skippedUsers/3")
assert.Equal(t, "", buffer.String())
}
func TestDisableConsoleColor(t *testing.T) {