traefik/pkg/testhelpers/config.go
Julien Salleyron 6fed76a687 WeightedRoundRobin load balancer
Co-authored-by: Ludovic Fernandez <ldez@users.noreply.github.com>
2019-08-26 01:30:05 -07:00

147 lines
4 KiB
Go

package testhelpers
import (
"github.com/containous/traefik/v2/pkg/config/dynamic"
)
// BuildConfiguration is a helper to create a configuration.
func BuildConfiguration(dynamicConfigBuilders ...func(*dynamic.HTTPConfiguration)) *dynamic.HTTPConfiguration {
conf := &dynamic.HTTPConfiguration{}
for _, build := range dynamicConfigBuilders {
build(conf)
}
return conf
}
// WithRouters is a helper to create a configuration.
func WithRouters(opts ...func(*dynamic.Router) string) func(*dynamic.HTTPConfiguration) {
return func(c *dynamic.HTTPConfiguration) {
c.Routers = make(map[string]*dynamic.Router)
for _, opt := range opts {
b := &dynamic.Router{}
name := opt(b)
c.Routers[name] = b
}
}
}
// WithRouter is a helper to create a configuration.
func WithRouter(routerName string, opts ...func(*dynamic.Router)) func(*dynamic.Router) string {
return func(r *dynamic.Router) string {
for _, opt := range opts {
opt(r)
}
return routerName
}
}
// WithRouterMiddlewares is a helper to create a configuration.
func WithRouterMiddlewares(middlewaresName ...string) func(*dynamic.Router) {
return func(r *dynamic.Router) {
r.Middlewares = middlewaresName
}
}
// WithServiceName is a helper to create a configuration.
func WithServiceName(serviceName string) func(*dynamic.Router) {
return func(r *dynamic.Router) {
r.Service = serviceName
}
}
// WithLoadBalancerServices is a helper to create a configuration.
func WithLoadBalancerServices(opts ...func(service *dynamic.ServersLoadBalancer) string) func(*dynamic.HTTPConfiguration) {
return func(c *dynamic.HTTPConfiguration) {
c.Services = make(map[string]*dynamic.Service)
for _, opt := range opts {
b := &dynamic.ServersLoadBalancer{}
name := opt(b)
c.Services[name] = &dynamic.Service{
LoadBalancer: b,
}
}
}
}
// WithService is a helper to create a configuration.
func WithService(name string, opts ...func(*dynamic.ServersLoadBalancer)) func(*dynamic.ServersLoadBalancer) string {
return func(r *dynamic.ServersLoadBalancer) string {
for _, opt := range opts {
opt(r)
}
return name
}
}
// WithMiddlewares is a helper to create a configuration.
func WithMiddlewares(opts ...func(*dynamic.Middleware) string) func(*dynamic.HTTPConfiguration) {
return func(c *dynamic.HTTPConfiguration) {
c.Middlewares = make(map[string]*dynamic.Middleware)
for _, opt := range opts {
b := &dynamic.Middleware{}
name := opt(b)
c.Middlewares[name] = b
}
}
}
// WithMiddleware is a helper to create a configuration.
func WithMiddleware(name string, opts ...func(*dynamic.Middleware)) func(*dynamic.Middleware) string {
return func(r *dynamic.Middleware) string {
for _, opt := range opts {
opt(r)
}
return name
}
}
// WithBasicAuth is a helper to create a configuration.
func WithBasicAuth(auth *dynamic.BasicAuth) func(*dynamic.Middleware) {
return func(r *dynamic.Middleware) {
r.BasicAuth = auth
}
}
// WithEntryPoints is a helper to create a configuration.
func WithEntryPoints(eps ...string) func(*dynamic.Router) {
return func(f *dynamic.Router) {
f.EntryPoints = eps
}
}
// WithRule is a helper to create a configuration.
func WithRule(rule string) func(*dynamic.Router) {
return func(f *dynamic.Router) {
f.Rule = rule
}
}
// WithServers is a helper to create a configuration.
func WithServers(opts ...func(*dynamic.Server)) func(*dynamic.ServersLoadBalancer) {
return func(b *dynamic.ServersLoadBalancer) {
for _, opt := range opts {
server := dynamic.Server{}
opt(&server)
b.Servers = append(b.Servers, server)
}
}
}
// WithServer is a helper to create a configuration.
func WithServer(url string, opts ...func(*dynamic.Server)) func(*dynamic.Server) {
return func(s *dynamic.Server) {
for _, opt := range opts {
opt(s)
}
s.URL = url
}
}
// WithSticky is a helper to create a configuration.
func WithSticky(cookieName string) func(*dynamic.ServersLoadBalancer) {
return func(b *dynamic.ServersLoadBalancer) {
b.Sticky = &dynamic.Sticky{
Cookie: &dynamic.Cookie{Name: cookieName},
}
}
}