Dynamic Configuration Refactoring

old/configuration/provider_aggregator.go

@@ -0,0 +1,111 @@
+package configuration
+
+import (
+	"encoding/json"
+
+	"github.com/containous/traefik/old/log"
+	"github.com/containous/traefik/old/provider"
+	"github.com/containous/traefik/old/types"
+	"github.com/containous/traefik/safe"
+)
+
+// ProviderAggregator aggregate providers
+type ProviderAggregator struct {
+	providers   []provider.Provider
+	constraints types.Constraints
+}
+
+// NewProviderAggregator return an aggregate of all the providers configured in GlobalConfiguration
+func NewProviderAggregator(gc *GlobalConfiguration) ProviderAggregator {
+	provider := ProviderAggregator{
+		constraints: gc.Constraints,
+	}
+	if gc.Docker != nil {
+		provider.quietAddProvider(gc.Docker)
+	}
+	if gc.Marathon != nil {
+		provider.quietAddProvider(gc.Marathon)
+	}
+	if gc.File != nil {
+		provider.quietAddProvider(gc.File)
+	}
+	if gc.Rest != nil {
+		provider.quietAddProvider(gc.Rest)
+	}
+	if gc.Consul != nil {
+		provider.quietAddProvider(gc.Consul)
+	}
+	if gc.ConsulCatalog != nil {
+		provider.quietAddProvider(gc.ConsulCatalog)
+	}
+	if gc.Etcd != nil {
+		provider.quietAddProvider(gc.Etcd)
+	}
+	if gc.Zookeeper != nil {
+		provider.quietAddProvider(gc.Zookeeper)
+	}
+	if gc.Boltdb != nil {
+		provider.quietAddProvider(gc.Boltdb)
+	}
+	if gc.Kubernetes != nil {
+		provider.quietAddProvider(gc.Kubernetes)
+	}
+	if gc.Mesos != nil {
+		provider.quietAddProvider(gc.Mesos)
+	}
+	if gc.Eureka != nil {
+		provider.quietAddProvider(gc.Eureka)
+	}
+	if gc.ECS != nil {
+		provider.quietAddProvider(gc.ECS)
+	}
+	if gc.Rancher != nil {
+		provider.quietAddProvider(gc.Rancher)
+	}
+	if gc.DynamoDB != nil {
+		provider.quietAddProvider(gc.DynamoDB)
+	}
+
+	return provider
+}
+
+func (p *ProviderAggregator) quietAddProvider(provider provider.Provider) {
+	err := p.AddProvider(provider)
+	if err != nil {
+		log.Errorf("Error initializing provider %T: %v", provider, err)
+	}
+}
+
+// AddProvider add a provider in the providers map
+func (p *ProviderAggregator) AddProvider(provider provider.Provider) error {
+	err := provider.Init(p.constraints)
+	if err != nil {
+		return err
+	}
+	p.providers = append(p.providers, provider)
+	return nil
+}
+
+// Init the provider
+func (p ProviderAggregator) Init(_ types.Constraints) error {
+	return nil
+}
+
+// Provide call the provide method of every providers
+func (p ProviderAggregator) Provide(configurationChan chan<- types.ConfigMessage, pool *safe.Pool) error {
+	for _, p := range p.providers {
+		jsonConf, err := json.Marshal(p)
+		if err != nil {
+			log.Debugf("Unable to marshal provider conf %T with error: %v", p, err)
+		}
+		log.Infof("Starting provider %T %s", p, jsonConf)
+		currentProvider := p
+		safe.Go(func() {
+			err := currentProvider.Provide(configurationChan, pool)
+			if err != nil {
+				log.Errorf("Error starting provider %T: %v", p, err)
+			}
+		})
+	}
+	return nil
+}