2018-11-14 09:18:03 +00:00
|
|
|
package aggregator
|
|
|
|
|
|
|
|
import (
|
|
|
|
"encoding/json"
|
|
|
|
|
2020-09-16 13:46:04 +00:00
|
|
|
"github.com/traefik/traefik/v2/pkg/config/dynamic"
|
|
|
|
"github.com/traefik/traefik/v2/pkg/config/static"
|
|
|
|
"github.com/traefik/traefik/v2/pkg/log"
|
|
|
|
"github.com/traefik/traefik/v2/pkg/provider"
|
|
|
|
"github.com/traefik/traefik/v2/pkg/provider/file"
|
2021-01-28 15:16:05 +00:00
|
|
|
"github.com/traefik/traefik/v2/pkg/provider/traefik"
|
2020-09-16 13:46:04 +00:00
|
|
|
"github.com/traefik/traefik/v2/pkg/safe"
|
2018-11-14 09:18:03 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
// ProviderAggregator aggregates providers.
|
|
|
|
type ProviderAggregator struct {
|
2021-01-28 15:16:05 +00:00
|
|
|
internalProvider provider.Provider
|
|
|
|
fileProvider provider.Provider
|
|
|
|
providers []provider.Provider
|
2018-11-14 09:18:03 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// NewProviderAggregator returns an aggregate of all the providers configured in the static configuration.
|
2018-11-27 16:42:04 +00:00
|
|
|
func NewProviderAggregator(conf static.Providers) ProviderAggregator {
|
|
|
|
p := ProviderAggregator{}
|
2018-11-14 09:18:03 +00:00
|
|
|
|
|
|
|
if conf.File != nil {
|
|
|
|
p.quietAddProvider(conf.File)
|
|
|
|
}
|
|
|
|
|
2019-01-18 14:18:04 +00:00
|
|
|
if conf.Docker != nil {
|
|
|
|
p.quietAddProvider(conf.Docker)
|
|
|
|
}
|
|
|
|
|
2019-01-29 16:54:05 +00:00
|
|
|
if conf.Marathon != nil {
|
|
|
|
p.quietAddProvider(conf.Marathon)
|
|
|
|
}
|
|
|
|
|
2018-12-03 10:32:05 +00:00
|
|
|
if conf.Rest != nil {
|
|
|
|
p.quietAddProvider(conf.Rest)
|
|
|
|
}
|
|
|
|
|
2019-07-08 19:36:03 +00:00
|
|
|
if conf.KubernetesIngress != nil {
|
|
|
|
p.quietAddProvider(conf.KubernetesIngress)
|
2019-02-21 22:08:05 +00:00
|
|
|
}
|
|
|
|
|
2019-03-14 14:56:06 +00:00
|
|
|
if conf.KubernetesCRD != nil {
|
|
|
|
p.quietAddProvider(conf.KubernetesCRD)
|
|
|
|
}
|
2019-06-11 13:12:04 +00:00
|
|
|
|
2020-12-15 15:40:05 +00:00
|
|
|
if conf.KubernetesGateway != nil {
|
|
|
|
p.quietAddProvider(conf.KubernetesGateway)
|
|
|
|
}
|
|
|
|
|
2019-04-05 10:22:04 +00:00
|
|
|
if conf.Rancher != nil {
|
|
|
|
p.quietAddProvider(conf.Rancher)
|
|
|
|
}
|
2019-03-14 14:56:06 +00:00
|
|
|
|
2020-07-15 14:28:04 +00:00
|
|
|
if conf.Ecs != nil {
|
|
|
|
p.quietAddProvider(conf.Ecs)
|
|
|
|
}
|
|
|
|
|
2019-10-15 15:34:08 +00:00
|
|
|
if conf.ConsulCatalog != nil {
|
|
|
|
p.quietAddProvider(conf.ConsulCatalog)
|
|
|
|
}
|
|
|
|
|
2019-11-28 20:56:04 +00:00
|
|
|
if conf.Consul != nil {
|
|
|
|
p.quietAddProvider(conf.Consul)
|
|
|
|
}
|
|
|
|
|
|
|
|
if conf.Etcd != nil {
|
|
|
|
p.quietAddProvider(conf.Etcd)
|
|
|
|
}
|
|
|
|
|
|
|
|
if conf.ZooKeeper != nil {
|
|
|
|
p.quietAddProvider(conf.ZooKeeper)
|
|
|
|
}
|
|
|
|
|
|
|
|
if conf.Redis != nil {
|
|
|
|
p.quietAddProvider(conf.Redis)
|
|
|
|
}
|
|
|
|
|
2020-07-15 14:56:03 +00:00
|
|
|
if conf.HTTP != nil {
|
|
|
|
p.quietAddProvider(conf.HTTP)
|
|
|
|
}
|
|
|
|
|
2018-11-14 09:18:03 +00:00
|
|
|
return p
|
|
|
|
}
|
|
|
|
|
|
|
|
func (p *ProviderAggregator) quietAddProvider(provider provider.Provider) {
|
|
|
|
err := p.AddProvider(provider)
|
|
|
|
if err != nil {
|
|
|
|
log.WithoutContext().Errorf("Error while initializing provider %T: %v", provider, err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// AddProvider adds a provider in the providers map.
|
|
|
|
func (p *ProviderAggregator) AddProvider(provider provider.Provider) error {
|
2018-11-27 16:42:04 +00:00
|
|
|
err := provider.Init()
|
2018-11-14 09:18:03 +00:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2019-03-14 08:30:04 +00:00
|
|
|
|
2021-01-28 15:16:05 +00:00
|
|
|
switch provider.(type) {
|
|
|
|
case *file.Provider:
|
|
|
|
p.fileProvider = provider
|
|
|
|
case *traefik.Provider:
|
|
|
|
p.internalProvider = provider
|
|
|
|
default:
|
2019-03-14 08:30:04 +00:00
|
|
|
p.providers = append(p.providers, provider)
|
|
|
|
}
|
2021-01-28 15:16:05 +00:00
|
|
|
|
2018-11-14 09:18:03 +00:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2020-05-11 10:06:07 +00:00
|
|
|
// Init the provider.
|
2018-11-27 16:42:04 +00:00
|
|
|
func (p ProviderAggregator) Init() error {
|
2018-11-14 09:18:03 +00:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2020-05-11 10:06:07 +00:00
|
|
|
// Provide calls the provide method of every providers.
|
2019-07-10 07:26:04 +00:00
|
|
|
func (p ProviderAggregator) Provide(configurationChan chan<- dynamic.Message, pool *safe.Pool) error {
|
2019-03-14 08:30:04 +00:00
|
|
|
if p.fileProvider != nil {
|
|
|
|
launchProvider(configurationChan, pool, p.fileProvider)
|
|
|
|
}
|
|
|
|
|
2021-01-28 15:16:05 +00:00
|
|
|
if p.internalProvider != nil {
|
|
|
|
launchProvider(configurationChan, pool, p.internalProvider)
|
|
|
|
}
|
|
|
|
|
2018-11-14 09:18:03 +00:00
|
|
|
for _, prd := range p.providers {
|
2019-03-19 09:04:04 +00:00
|
|
|
prd := prd
|
2018-11-14 09:18:03 +00:00
|
|
|
safe.Go(func() {
|
2019-03-14 08:30:04 +00:00
|
|
|
launchProvider(configurationChan, pool, prd)
|
2018-11-14 09:18:03 +00:00
|
|
|
})
|
|
|
|
}
|
2021-01-28 15:16:05 +00:00
|
|
|
|
2018-11-14 09:18:03 +00:00
|
|
|
return nil
|
|
|
|
}
|
2019-03-14 08:30:04 +00:00
|
|
|
|
2019-07-10 07:26:04 +00:00
|
|
|
func launchProvider(configurationChan chan<- dynamic.Message, pool *safe.Pool, prd provider.Provider) {
|
2019-03-14 08:30:04 +00:00
|
|
|
jsonConf, err := json.Marshal(prd)
|
|
|
|
if err != nil {
|
|
|
|
log.WithoutContext().Debugf("Cannot marshal the provider configuration %T: %v", prd, err)
|
|
|
|
}
|
|
|
|
|
|
|
|
log.WithoutContext().Infof("Starting provider %T %s", prd, jsonConf)
|
|
|
|
|
|
|
|
currentProvider := prd
|
|
|
|
err = currentProvider.Provide(configurationChan, pool)
|
|
|
|
if err != nil {
|
|
|
|
log.WithoutContext().Errorf("Cannot start the provider %T: %v", prd, err)
|
|
|
|
}
|
|
|
|
}
|