119 lines
3.3 KiB
Go
119 lines
3.3 KiB
Go
package router
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/containous/alice"
|
|
"github.com/containous/mux"
|
|
"github.com/containous/traefik/api"
|
|
"github.com/containous/traefik/config/static"
|
|
"github.com/containous/traefik/log"
|
|
"github.com/containous/traefik/metrics"
|
|
"github.com/containous/traefik/safe"
|
|
"github.com/containous/traefik/types"
|
|
)
|
|
|
|
// chainBuilder The contract of the middleware builder
|
|
type chainBuilder interface {
|
|
BuildChain(ctx context.Context, middlewares []string) (*alice.Chain, error)
|
|
}
|
|
|
|
// NewRouteAppenderAggregator Creates a new RouteAppenderAggregator
|
|
func NewRouteAppenderAggregator(ctx context.Context, chainBuilder chainBuilder, conf static.Configuration, entryPointName string, currentConfiguration *safe.Safe) *RouteAppenderAggregator {
|
|
logger := log.FromContext(ctx)
|
|
|
|
aggregator := &RouteAppenderAggregator{}
|
|
|
|
// FIXME add REST
|
|
|
|
if conf.API != nil && conf.API.EntryPoint == entryPointName {
|
|
chain, err := chainBuilder.BuildChain(ctx, conf.API.Middlewares)
|
|
if err != nil {
|
|
logger.Error(err)
|
|
} else {
|
|
aggregator.AddAppender(&WithMiddleware{
|
|
appender: api.Handler{
|
|
EntryPoint: conf.API.EntryPoint,
|
|
Dashboard: conf.API.Dashboard,
|
|
Statistics: conf.API.Statistics,
|
|
DashboardAssets: conf.API.DashboardAssets,
|
|
CurrentConfigurations: currentConfiguration,
|
|
},
|
|
routerMiddlewares: chain,
|
|
})
|
|
}
|
|
}
|
|
|
|
if conf.Ping != nil && conf.Ping.EntryPoint == entryPointName {
|
|
chain, err := chainBuilder.BuildChain(ctx, conf.Ping.Middlewares)
|
|
if err != nil {
|
|
logger.Error(err)
|
|
} else {
|
|
aggregator.AddAppender(&WithMiddleware{
|
|
appender: conf.Ping,
|
|
routerMiddlewares: chain,
|
|
})
|
|
}
|
|
}
|
|
|
|
if conf.Metrics != nil && conf.Metrics.Prometheus != nil && conf.Metrics.Prometheus.EntryPoint == entryPointName {
|
|
chain, err := chainBuilder.BuildChain(ctx, conf.Metrics.Prometheus.Middlewares)
|
|
if err != nil {
|
|
logger.Error(err)
|
|
} else {
|
|
aggregator.AddAppender(&WithMiddleware{
|
|
appender: metrics.PrometheusHandler{},
|
|
routerMiddlewares: chain,
|
|
})
|
|
}
|
|
}
|
|
|
|
return aggregator
|
|
}
|
|
|
|
// RouteAppenderAggregator RouteAppender that aggregate other RouteAppender
|
|
type RouteAppenderAggregator struct {
|
|
appenders []types.RouteAppender
|
|
}
|
|
|
|
// Append Adds routes to the router
|
|
func (r *RouteAppenderAggregator) Append(systemRouter *mux.Router) {
|
|
for _, router := range r.appenders {
|
|
router.Append(systemRouter)
|
|
}
|
|
}
|
|
|
|
// AddAppender adds a router in the aggregator
|
|
func (r *RouteAppenderAggregator) AddAppender(router types.RouteAppender) {
|
|
r.appenders = append(r.appenders, router)
|
|
}
|
|
|
|
// WithMiddleware router with internal middleware
|
|
type WithMiddleware struct {
|
|
appender types.RouteAppender
|
|
routerMiddlewares *alice.Chain
|
|
}
|
|
|
|
// Append Adds routes to the router
|
|
func (wm *WithMiddleware) Append(systemRouter *mux.Router) {
|
|
realRouter := systemRouter.PathPrefix("/").Subrouter()
|
|
|
|
wm.appender.Append(realRouter)
|
|
|
|
if err := realRouter.Walk(wrapRoute(wm.routerMiddlewares)); err != nil {
|
|
log.WithoutContext().Error(err)
|
|
}
|
|
}
|
|
|
|
// wrapRoute with middlewares
|
|
func wrapRoute(middlewares *alice.Chain) func(*mux.Route, *mux.Router, []*mux.Route) error {
|
|
return func(route *mux.Route, router *mux.Router, ancestors []*mux.Route) error {
|
|
handler, err := middlewares.Then(route.GetHandler())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
route.Handler(handler)
|
|
return nil
|
|
}
|
|
}
|