IntenseWebs/grafana
3
0
Fork 0
mirror of https://github.com/grafana/grafana.git synced 2025-02-25 18:55:37 -06:00
Code Issues Packages Projects Releases Wiki Activity

HTTPServer: add possibility to use additional middlewares (#26514)

Browse Source
This commit is contained in:
Agnès Toulet 2020-07-23 09:54:07 +02:00 committed by GitHub
parent b12d08d9de
commit 1ba95183f6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
pkg/api/http_server.go
View File

@ -53,6 +53,7 @@ type HTTPServer struct {
context context.Context context context.Context
streamManager *live.StreamManager streamManager *live.StreamManager
httpSrv *http.Server httpSrv *http.Server
middlewares []macaron.Handler
RouteRegister routing.RouteRegister `inject:""` RouteRegister routing.RouteRegister `inject:""`
Bus bus.Bus `inject:""` Bus bus.Bus `inject:""`
@ -82,6 +83,10 @@ func (hs *HTTPServer) Init() error {
return nil return nil
} }
func (hs *HTTPServer) AddMiddleware(middleware macaron.Handler) {
hs.middlewares = append(hs.middlewares, middleware)
}
func (hs *HTTPServer) Run(ctx context.Context) error { func (hs *HTTPServer) Run(ctx context.Context) error {
hs.context = ctx hs.context = ctx
@ -323,6 +328,10 @@ func (hs *HTTPServer) addMiddlewaresAndStaticRoutes() {
} }
m.Use(middleware.HandleNoCacheHeader()) m.Use(middleware.HandleNoCacheHeader())
for _, mw := range hs.middlewares {
m.Use(mw)
}
} }
func (hs *HTTPServer) metricsEndpoint(ctx *macaron.Context) { func (hs *HTTPServer) metricsEndpoint(ctx *macaron.Context) {