5bf8d5445e
Make router logger more friendly, show the related function name/file/line. [BREAKING] This PR substantially changes the logging format of the router logger. If you use this logging for monitoring e.g. fail2ban you will need to update this to match the new format.
118 lines
3.2 KiB
Go
118 lines
3.2 KiB
Go
// Copyright 2021 The Gitea Authors. All rights reserved.
|
|
// Use of this source code is governed by a MIT-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package web
|
|
|
|
import (
|
|
goctx "context"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"code.gitea.io/gitea/modules/context"
|
|
"code.gitea.io/gitea/modules/web/routing"
|
|
)
|
|
|
|
// Wrap converts all kinds of routes to standard library one
|
|
func Wrap(handlers ...interface{}) http.HandlerFunc {
|
|
if len(handlers) == 0 {
|
|
panic("No handlers found")
|
|
}
|
|
|
|
ourHandlers := make([]wrappedHandlerFunc, 0, len(handlers))
|
|
|
|
for _, handler := range handlers {
|
|
ourHandlers = append(ourHandlers, convertHandler(handler))
|
|
}
|
|
return wrapInternal(ourHandlers)
|
|
}
|
|
|
|
func wrapInternal(handlers []wrappedHandlerFunc) http.HandlerFunc {
|
|
return func(resp http.ResponseWriter, req *http.Request) {
|
|
var defers []func()
|
|
defer func() {
|
|
for i := len(defers) - 1; i >= 0; i-- {
|
|
defers[i]()
|
|
}
|
|
}()
|
|
for i := 0; i < len(handlers); i++ {
|
|
handler := handlers[i]
|
|
others := handlers[i+1:]
|
|
done, deferrable := handler(resp, req, others...)
|
|
if deferrable != nil {
|
|
defers = append(defers, deferrable)
|
|
}
|
|
if done {
|
|
return
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// Middle wrap a context function as a chi middleware
|
|
func Middle(f func(ctx *context.Context)) func(next http.Handler) http.Handler {
|
|
funcInfo := routing.GetFuncInfo(f)
|
|
return func(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(resp http.ResponseWriter, req *http.Request) {
|
|
routing.UpdateFuncInfo(req.Context(), funcInfo)
|
|
ctx := context.GetContext(req)
|
|
f(ctx)
|
|
if ctx.Written() {
|
|
return
|
|
}
|
|
next.ServeHTTP(ctx.Resp, ctx.Req)
|
|
})
|
|
}
|
|
}
|
|
|
|
// MiddleCancel wrap a context function as a chi middleware
|
|
func MiddleCancel(f func(ctx *context.Context) goctx.CancelFunc) func(netx http.Handler) http.Handler {
|
|
funcInfo := routing.GetFuncInfo(f)
|
|
return func(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(resp http.ResponseWriter, req *http.Request) {
|
|
routing.UpdateFuncInfo(req.Context(), funcInfo)
|
|
ctx := context.GetContext(req)
|
|
cancel := f(ctx)
|
|
if cancel != nil {
|
|
defer cancel()
|
|
}
|
|
if ctx.Written() {
|
|
return
|
|
}
|
|
next.ServeHTTP(ctx.Resp, ctx.Req)
|
|
})
|
|
}
|
|
}
|
|
|
|
// MiddleAPI wrap a context function as a chi middleware
|
|
func MiddleAPI(f func(ctx *context.APIContext)) func(next http.Handler) http.Handler {
|
|
funcInfo := routing.GetFuncInfo(f)
|
|
return func(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(resp http.ResponseWriter, req *http.Request) {
|
|
routing.UpdateFuncInfo(req.Context(), funcInfo)
|
|
ctx := context.GetAPIContext(req)
|
|
f(ctx)
|
|
if ctx.Written() {
|
|
return
|
|
}
|
|
next.ServeHTTP(ctx.Resp, ctx.Req)
|
|
})
|
|
}
|
|
}
|
|
|
|
// WrapWithPrefix wraps a provided handler function at a prefix
|
|
func WrapWithPrefix(pathPrefix string, handler http.HandlerFunc, friendlyName ...string) func(next http.Handler) http.Handler {
|
|
funcInfo := routing.GetFuncInfo(handler, friendlyName...)
|
|
|
|
return func(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(resp http.ResponseWriter, req *http.Request) {
|
|
if !strings.HasPrefix(req.URL.Path, pathPrefix) {
|
|
next.ServeHTTP(resp, req)
|
|
return
|
|
}
|
|
routing.UpdateFuncInfo(req.Context(), funcInfo)
|
|
handler(resp, req)
|
|
})
|
|
}
|
|
}
|