92 lines
2.6 KiB
Go
92 lines
2.6 KiB
Go
|
package configuration
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"reflect"
|
||
|
|
|
||
|
|
"github.com/containous/traefik/log"
|
||
|
|
"github.com/containous/traefik/provider"
|
||
|
|
"github.com/containous/traefik/safe"
|
||
|
|
"github.com/containous/traefik/types"
|
||
|
|
)
|
||
|
|
|
||
|
|
type providerAggregator struct {
|
||
|
|
providers []provider.Provider
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewProviderAggregator return an aggregate of all the providers configured in GlobalConfiguration
|
||
|
|
func NewProviderAggregator(gc *GlobalConfiguration) provider.Provider {
|
||
|
|
provider := providerAggregator{}
|
||
|
|
if gc.Docker != nil {
|
||
|
|
provider.providers = append(provider.providers, gc.Docker)
|
||
|
|
}
|
||
|
|
if gc.Marathon != nil {
|
||
|
|
provider.providers = append(provider.providers, gc.Marathon)
|
||
|
|
}
|
||
|
|
if gc.File != nil {
|
||
|
|
provider.providers = append(provider.providers, gc.File)
|
||
|
|
}
|
||
|
|
if gc.Rest != nil {
|
||
|
|
provider.providers = append(provider.providers, gc.Rest)
|
||
|
|
}
|
||
|
|
if gc.Consul != nil {
|
||
|
|
provider.providers = append(provider.providers, gc.Consul)
|
||
|
|
}
|
||
|
|
if gc.ConsulCatalog != nil {
|
||
|
|
provider.providers = append(provider.providers, gc.ConsulCatalog)
|
||
|
|
}
|
||
|
|
if gc.Etcd != nil {
|
||
|
|
provider.providers = append(provider.providers, gc.Etcd)
|
||
|
|
}
|
||
|
|
if gc.Zookeeper != nil {
|
||
|
|
provider.providers = append(provider.providers, gc.Zookeeper)
|
||
|
|
}
|
||
|
|
if gc.Boltdb != nil {
|
||
|
|
provider.providers = append(provider.providers, gc.Boltdb)
|
||
|
|
}
|
||
|
|
if gc.Kubernetes != nil {
|
||
|
|
provider.providers = append(provider.providers, gc.Kubernetes)
|
||
|
|
}
|
||
|
|
if gc.Mesos != nil {
|
||
|
|
provider.providers = append(provider.providers, gc.Mesos)
|
||
|
|
}
|
||
|
|
if gc.Eureka != nil {
|
||
|
|
provider.providers = append(provider.providers, gc.Eureka)
|
||
|
|
}
|
||
|
|
if gc.ECS != nil {
|
||
|
|
provider.providers = append(provider.providers, gc.ECS)
|
||
|
|
}
|
||
|
|
if gc.Rancher != nil {
|
||
|
|
provider.providers = append(provider.providers, gc.Rancher)
|
||
|
|
}
|
||
|
|
if gc.DynamoDB != nil {
|
||
|
|
provider.providers = append(provider.providers, gc.DynamoDB)
|
||
|
|
}
|
||
|
|
if gc.ServiceFabric != nil {
|
||
|
|
provider.providers = append(provider.providers, gc.ServiceFabric)
|
||
|
|
}
|
||
|
|
if len(provider.providers) == 1 {
|
||
|
|
return provider.providers[0]
|
||
|
|
}
|
||
|
|
return provider
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p providerAggregator) Provide(configurationChan chan<- types.ConfigMessage, pool *safe.Pool, constraints types.Constraints) error {
|
||
|
|
for _, p := range p.providers {
|
||
|
|
providerType := reflect.TypeOf(p)
|
||
|
|
jsonConf, err := json.Marshal(p)
|
||
|
|
if err != nil {
|
||
|
|
log.Debugf("Unable to marshal provider conf %v with error: %v", providerType, err)
|
||
|
|
}
|
||
|
|
log.Infof("Starting provider %v %s", providerType, jsonConf)
|
||
|
|
currentProvider := p
|
||
|
|
safe.Go(func() {
|
||
|
|
err := currentProvider.Provide(configurationChan, pool, constraints)
|
||
|
|
if err != nil {
|
||
|
|
log.Errorf("Error starting provider %v: %s", providerType, err)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|