Merge branch 'master' into docker-tls
This commit is contained in:
commit
76bd04e349
14 changed files with 2319 additions and 438 deletions
|
@ -15,7 +15,7 @@ func (s *FileSuite) TestSimpleConfiguration(c *check.C) {
|
||||||
c.Assert(err, checker.IsNil)
|
c.Assert(err, checker.IsNil)
|
||||||
defer cmd.Process.Kill()
|
defer cmd.Process.Kill()
|
||||||
|
|
||||||
time.Sleep(500 * time.Millisecond)
|
time.Sleep(1000 * time.Millisecond)
|
||||||
resp, err := http.Get("http://127.0.0.1/")
|
resp, err := http.Get("http://127.0.0.1/")
|
||||||
|
|
||||||
// Expected a 404 as we did not configure anything
|
// Expected a 404 as we did not configure anything
|
||||||
|
@ -30,7 +30,7 @@ func (s *FileSuite) TestSimpleConfigurationNoPanic(c *check.C) {
|
||||||
c.Assert(err, checker.IsNil)
|
c.Assert(err, checker.IsNil)
|
||||||
defer cmd.Process.Kill()
|
defer cmd.Process.Kill()
|
||||||
|
|
||||||
time.Sleep(500 * time.Millisecond)
|
time.Sleep(1000 * time.Millisecond)
|
||||||
resp, err := http.Get("http://127.0.0.1/")
|
resp, err := http.Get("http://127.0.0.1/")
|
||||||
|
|
||||||
// Expected a 404 as we did not configure anything
|
// Expected a 404 as we did not configure anything
|
||||||
|
|
|
@ -1,19 +1,20 @@
|
||||||
package provider
|
package provider
|
||||||
|
|
||||||
import "github.com/emilevauge/traefik/types"
|
import (
|
||||||
|
"github.com/docker/libkv/store"
|
||||||
|
"github.com/docker/libkv/store/boltdb"
|
||||||
|
"github.com/emilevauge/traefik/types"
|
||||||
|
)
|
||||||
|
|
||||||
// BoltDb holds configurations of the BoltDb provider.
|
// BoltDb holds configurations of the BoltDb provider.
|
||||||
type BoltDb struct {
|
type BoltDb struct {
|
||||||
Watch bool
|
Kv
|
||||||
Endpoint string
|
|
||||||
Prefix string
|
|
||||||
Filename string
|
|
||||||
KvProvider *Kv
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Provide allows the provider to provide configurations to traefik
|
// Provide allows the provider to provide configurations to traefik
|
||||||
// using the given configuration channel.
|
// using the given configuration channel.
|
||||||
func (provider *BoltDb) Provide(configurationChan chan<- types.ConfigMessage) error {
|
func (provider *BoltDb) Provide(configurationChan chan<- types.ConfigMessage) error {
|
||||||
provider.KvProvider = NewBoltDbProvider(provider)
|
provider.StoreType = store.BOLTDB
|
||||||
return provider.KvProvider.provide(configurationChan)
|
boltdb.Register()
|
||||||
|
return provider.provide(configurationChan)
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,19 +1,20 @@
|
||||||
package provider
|
package provider
|
||||||
|
|
||||||
import "github.com/emilevauge/traefik/types"
|
import (
|
||||||
|
"github.com/docker/libkv/store"
|
||||||
|
"github.com/docker/libkv/store/consul"
|
||||||
|
"github.com/emilevauge/traefik/types"
|
||||||
|
)
|
||||||
|
|
||||||
// Consul holds configurations of the Consul provider.
|
// Consul holds configurations of the Consul provider.
|
||||||
type Consul struct {
|
type Consul struct {
|
||||||
Watch bool
|
Kv
|
||||||
Endpoint string
|
|
||||||
Prefix string
|
|
||||||
Filename string
|
|
||||||
KvProvider *Kv
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Provide allows the provider to provide configurations to traefik
|
// Provide allows the provider to provide configurations to traefik
|
||||||
// using the given configuration channel.
|
// using the given configuration channel.
|
||||||
func (provider *Consul) Provide(configurationChan chan<- types.ConfigMessage) error {
|
func (provider *Consul) Provide(configurationChan chan<- types.ConfigMessage) error {
|
||||||
provider.KvProvider = NewConsulProvider(provider)
|
provider.StoreType = store.CONSUL
|
||||||
return provider.KvProvider.provide(configurationChan)
|
consul.Register()
|
||||||
|
return provider.provide(configurationChan)
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,7 +1,6 @@
|
||||||
package provider
|
package provider
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"strconv"
|
"strconv"
|
||||||
|
@ -9,20 +8,17 @@ import (
|
||||||
"text/template"
|
"text/template"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/BurntSushi/toml"
|
|
||||||
"github.com/BurntSushi/ty/fun"
|
"github.com/BurntSushi/ty/fun"
|
||||||
log "github.com/Sirupsen/logrus"
|
log "github.com/Sirupsen/logrus"
|
||||||
"github.com/cenkalti/backoff"
|
"github.com/cenkalti/backoff"
|
||||||
"github.com/emilevauge/traefik/autogen"
|
|
||||||
"github.com/emilevauge/traefik/types"
|
"github.com/emilevauge/traefik/types"
|
||||||
"github.com/fsouza/go-dockerclient"
|
"github.com/fsouza/go-dockerclient"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Docker holds configurations of the Docker provider.
|
// Docker holds configurations of the Docker provider.
|
||||||
type Docker struct {
|
type Docker struct {
|
||||||
Watch bool
|
baseProvider
|
||||||
Endpoint string
|
Endpoint string
|
||||||
Filename string
|
|
||||||
Domain string
|
Domain string
|
||||||
TLS *DockerTLS
|
TLS *DockerTLS
|
||||||
}
|
}
|
||||||
|
@ -75,9 +71,12 @@ func (provider *Docker) Provide(configurationChan chan<- types.ConfigMessage) er
|
||||||
}
|
}
|
||||||
if event.Status == "start" || event.Status == "die" {
|
if event.Status == "start" || event.Status == "die" {
|
||||||
log.Debugf("Docker event receveived %+v", event)
|
log.Debugf("Docker event receveived %+v", event)
|
||||||
configuration := provider.loadDockerConfig(dockerClient)
|
configuration := provider.loadDockerConfig(listContainers(dockerClient))
|
||||||
if configuration != nil {
|
if configuration != nil {
|
||||||
configurationChan <- types.ConfigMessage{"docker", configuration}
|
configurationChan <- types.ConfigMessage{
|
||||||
|
ProviderName: "docker",
|
||||||
|
Configuration: configuration,
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -92,94 +91,31 @@ func (provider *Docker) Provide(configurationChan chan<- types.ConfigMessage) er
|
||||||
}()
|
}()
|
||||||
}
|
}
|
||||||
|
|
||||||
configuration := provider.loadDockerConfig(dockerClient)
|
configuration := provider.loadDockerConfig(listContainers(dockerClient))
|
||||||
configurationChan <- types.ConfigMessage{"docker", configuration}
|
configurationChan <- types.ConfigMessage{
|
||||||
|
ProviderName: "docker",
|
||||||
|
Configuration: configuration,
|
||||||
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (provider *Docker) loadDockerConfig(dockerClient *docker.Client) *types.Configuration {
|
func (provider *Docker) loadDockerConfig(containersInspected []docker.Container) *types.Configuration {
|
||||||
var DockerFuncMap = template.FuncMap{
|
var DockerFuncMap = template.FuncMap{
|
||||||
"getBackend": func(container docker.Container) string {
|
"getBackend": provider.getBackend,
|
||||||
if label, err := provider.getLabel(container, "traefik.backend"); err == nil {
|
"getPort": provider.getPort,
|
||||||
return label
|
"getWeight": provider.getWeight,
|
||||||
}
|
"getDomain": provider.getDomain,
|
||||||
return provider.getEscapedName(container.Name)
|
"getProtocol": provider.getProtocol,
|
||||||
},
|
"getPassHostHeader": provider.getPassHostHeader,
|
||||||
"getPort": func(container docker.Container) string {
|
"getFrontendValue": provider.getFrontendValue,
|
||||||
if label, err := provider.getLabel(container, "traefik.port"); err == nil {
|
"getFrontendRule": provider.getFrontendRule,
|
||||||
return label
|
"replace": replace,
|
||||||
}
|
|
||||||
for key := range container.NetworkSettings.Ports {
|
|
||||||
return key.Port()
|
|
||||||
}
|
|
||||||
return ""
|
|
||||||
},
|
|
||||||
"getWeight": func(container docker.Container) string {
|
|
||||||
if label, err := provider.getLabel(container, "traefik.weight"); err == nil {
|
|
||||||
return label
|
|
||||||
}
|
|
||||||
return "0"
|
|
||||||
},
|
|
||||||
"getDomain": func(container docker.Container) string {
|
|
||||||
if label, err := provider.getLabel(container, "traefik.domain"); err == nil {
|
|
||||||
return label
|
|
||||||
}
|
|
||||||
return provider.Domain
|
|
||||||
},
|
|
||||||
"getProtocol": func(container docker.Container) string {
|
|
||||||
if label, err := provider.getLabel(container, "traefik.protocol"); err == nil {
|
|
||||||
return label
|
|
||||||
}
|
|
||||||
return "http"
|
|
||||||
},
|
|
||||||
"getPassHostHeader": func(container docker.Container) string {
|
|
||||||
if passHostHeader, err := provider.getLabel(container, "traefik.frontend.passHostHeader"); err == nil {
|
|
||||||
return passHostHeader
|
|
||||||
}
|
|
||||||
return "false"
|
|
||||||
},
|
|
||||||
"getFrontendValue": provider.GetFrontendValue,
|
|
||||||
"getFrontendRule": provider.GetFrontendRule,
|
|
||||||
"replace": func(s1 string, s2 string, s3 string) string {
|
|
||||||
return strings.Replace(s3, s1, s2, -1)
|
|
||||||
},
|
|
||||||
}
|
|
||||||
configuration := new(types.Configuration)
|
|
||||||
containerList, _ := dockerClient.ListContainers(docker.ListContainersOptions{})
|
|
||||||
containersInspected := []docker.Container{}
|
|
||||||
frontends := map[string][]docker.Container{}
|
|
||||||
|
|
||||||
// get inspect containers
|
|
||||||
for _, container := range containerList {
|
|
||||||
containerInspected, _ := dockerClient.InspectContainer(container.ID)
|
|
||||||
containersInspected = append(containersInspected, *containerInspected)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// filter containers
|
// filter containers
|
||||||
filteredContainers := fun.Filter(func(container docker.Container) bool {
|
filteredContainers := fun.Filter(containerFilter, containersInspected).([]docker.Container)
|
||||||
if len(container.NetworkSettings.Ports) == 0 {
|
|
||||||
log.Debugf("Filtering container without port %s", container.Name)
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
_, err := strconv.Atoi(container.Config.Labels["traefik.port"])
|
|
||||||
if len(container.NetworkSettings.Ports) > 1 && err != nil {
|
|
||||||
log.Debugf("Filtering container with more than 1 port and no traefik.port label %s", container.Name)
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
if container.Config.Labels["traefik.enable"] == "false" {
|
|
||||||
log.Debugf("Filtering disabled container %s", container.Name)
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
labels, err := provider.getLabels(container, []string{"traefik.frontend.rule", "traefik.frontend.value"})
|
|
||||||
if len(labels) != 0 && err != nil {
|
|
||||||
log.Debugf("Filtering bad labeled container %s", container.Name)
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
return true
|
|
||||||
}, containersInspected).([]docker.Container)
|
|
||||||
|
|
||||||
|
frontends := map[string][]docker.Container{}
|
||||||
for _, container := range filteredContainers {
|
for _, container := range filteredContainers {
|
||||||
frontends[provider.getFrontendName(container)] = append(frontends[provider.getFrontendName(container)], container)
|
frontends[provider.getFrontendName(container)] = append(frontends[provider.getFrontendName(container)], container)
|
||||||
}
|
}
|
||||||
|
@ -193,53 +129,112 @@ func (provider *Docker) loadDockerConfig(dockerClient *docker.Client) *types.Con
|
||||||
frontends,
|
frontends,
|
||||||
provider.Domain,
|
provider.Domain,
|
||||||
}
|
}
|
||||||
tmpl := template.New(provider.Filename).Funcs(DockerFuncMap)
|
|
||||||
if len(provider.Filename) > 0 {
|
|
||||||
_, err := tmpl.ParseFiles(provider.Filename)
|
|
||||||
if err != nil {
|
|
||||||
log.Error("Error reading file", err)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
buf, err := autogen.Asset("templates/docker.tmpl")
|
|
||||||
if err != nil {
|
|
||||||
log.Error("Error reading file", err)
|
|
||||||
}
|
|
||||||
_, err = tmpl.Parse(string(buf))
|
|
||||||
if err != nil {
|
|
||||||
log.Error("Error reading file", err)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
var buffer bytes.Buffer
|
configuration, err := provider.getConfiguration("templates/docker.tmpl", DockerFuncMap, templateObjects)
|
||||||
err := tmpl.Execute(&buffer, templateObjects)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Error("Error with docker template", err)
|
log.Error(err)
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
if _, err := toml.Decode(buffer.String(), configuration); err != nil {
|
|
||||||
log.Error("Error creating docker configuration ", err)
|
|
||||||
return nil
|
|
||||||
}
|
}
|
||||||
return configuration
|
return configuration
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func containerFilter(container docker.Container) bool {
|
||||||
|
if len(container.NetworkSettings.Ports) == 0 {
|
||||||
|
log.Debugf("Filtering container without port %s", container.Name)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
_, err := strconv.Atoi(container.Config.Labels["traefik.port"])
|
||||||
|
if len(container.NetworkSettings.Ports) > 1 && err != nil {
|
||||||
|
log.Debugf("Filtering container with more than 1 port and no traefik.port label %s", container.Name)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
if container.Config.Labels["traefik.enable"] == "false" {
|
||||||
|
log.Debugf("Filtering disabled container %s", container.Name)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
labels, err := getLabels(container, []string{"traefik.frontend.rule", "traefik.frontend.value"})
|
||||||
|
if len(labels) != 0 && err != nil {
|
||||||
|
log.Debugf("Filtering bad labeled container %s", container.Name)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
func (provider *Docker) getFrontendName(container docker.Container) string {
|
func (provider *Docker) getFrontendName(container docker.Container) string {
|
||||||
// Replace '.' with '-' in quoted keys because of this issue https://github.com/BurntSushi/toml/issues/78
|
// Replace '.' with '-' in quoted keys because of this issue https://github.com/BurntSushi/toml/issues/78
|
||||||
frontendName := fmt.Sprintf("%s-%s", provider.GetFrontendRule(container), provider.GetFrontendValue(container))
|
frontendName := fmt.Sprintf("%s-%s", provider.getFrontendRule(container), provider.getFrontendValue(container))
|
||||||
frontendName = strings.Replace(frontendName, "[", "", -1)
|
frontendName = strings.Replace(frontendName, "[", "", -1)
|
||||||
frontendName = strings.Replace(frontendName, "]", "", -1)
|
frontendName = strings.Replace(frontendName, "]", "", -1)
|
||||||
|
|
||||||
return strings.Replace(frontendName, ".", "-", -1)
|
return strings.Replace(frontendName, ".", "-", -1)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (provider *Docker) getEscapedName(name string) string {
|
// GetFrontendValue returns the frontend value for the specified container, using
|
||||||
return strings.Replace(name, "/", "", -1)
|
// it's label. It returns a default one if the label is not present.
|
||||||
|
func (provider *Docker) getFrontendValue(container docker.Container) string {
|
||||||
|
if label, err := getLabel(container, "traefik.frontend.value"); err == nil {
|
||||||
|
return label
|
||||||
|
}
|
||||||
|
return getEscapedName(container.Name) + "." + provider.Domain
|
||||||
}
|
}
|
||||||
|
|
||||||
func (provider *Docker) getLabel(container docker.Container, label string) (string, error) {
|
// GetFrontendRule returns the frontend rule for the specified container, using
|
||||||
|
// it's label. It returns a default one (Host) if the label is not present.
|
||||||
|
func (provider *Docker) getFrontendRule(container docker.Container) string {
|
||||||
|
if label, err := getLabel(container, "traefik.frontend.rule"); err == nil {
|
||||||
|
return label
|
||||||
|
}
|
||||||
|
return "Host"
|
||||||
|
}
|
||||||
|
|
||||||
|
func (provider *Docker) getBackend(container docker.Container) string {
|
||||||
|
if label, err := getLabel(container, "traefik.backend"); err == nil {
|
||||||
|
return label
|
||||||
|
}
|
||||||
|
return getEscapedName(container.Name)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (provider *Docker) getPort(container docker.Container) string {
|
||||||
|
if label, err := getLabel(container, "traefik.port"); err == nil {
|
||||||
|
return label
|
||||||
|
}
|
||||||
|
for key := range container.NetworkSettings.Ports {
|
||||||
|
return key.Port()
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
func (provider *Docker) getWeight(container docker.Container) string {
|
||||||
|
if label, err := getLabel(container, "traefik.weight"); err == nil {
|
||||||
|
return label
|
||||||
|
}
|
||||||
|
return "0"
|
||||||
|
}
|
||||||
|
|
||||||
|
func (provider *Docker) getDomain(container docker.Container) string {
|
||||||
|
if label, err := getLabel(container, "traefik.domain"); err == nil {
|
||||||
|
return label
|
||||||
|
}
|
||||||
|
return provider.Domain
|
||||||
|
}
|
||||||
|
|
||||||
|
func (provider *Docker) getProtocol(container docker.Container) string {
|
||||||
|
if label, err := getLabel(container, "traefik.protocol"); err == nil {
|
||||||
|
return label
|
||||||
|
}
|
||||||
|
return "http"
|
||||||
|
}
|
||||||
|
|
||||||
|
func (provider *Docker) getPassHostHeader(container docker.Container) string {
|
||||||
|
if passHostHeader, err := getLabel(container, "traefik.frontend.passHostHeader"); err == nil {
|
||||||
|
return passHostHeader
|
||||||
|
}
|
||||||
|
return "false"
|
||||||
|
}
|
||||||
|
|
||||||
|
func getLabel(container docker.Container, label string) (string, error) {
|
||||||
for key, value := range container.Config.Labels {
|
for key, value := range container.Config.Labels {
|
||||||
if key == label {
|
if key == label {
|
||||||
return value, nil
|
return value, nil
|
||||||
|
@ -248,11 +243,11 @@ func (provider *Docker) getLabel(container docker.Container, label string) (stri
|
||||||
return "", errors.New("Label not found:" + label)
|
return "", errors.New("Label not found:" + label)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (provider *Docker) getLabels(container docker.Container, labels []string) (map[string]string, error) {
|
func getLabels(container docker.Container, labels []string) (map[string]string, error) {
|
||||||
var globalErr error
|
var globalErr error
|
||||||
foundLabels := map[string]string{}
|
foundLabels := map[string]string{}
|
||||||
for _, label := range labels {
|
for _, label := range labels {
|
||||||
foundLabel, err := provider.getLabel(container, label)
|
foundLabel, err := getLabel(container, label)
|
||||||
// Error out only if one of them is defined.
|
// Error out only if one of them is defined.
|
||||||
if err != nil {
|
if err != nil {
|
||||||
globalErr = errors.New("Label not found: " + label)
|
globalErr = errors.New("Label not found: " + label)
|
||||||
|
@ -264,20 +259,14 @@ func (provider *Docker) getLabels(container docker.Container, labels []string) (
|
||||||
return foundLabels, globalErr
|
return foundLabels, globalErr
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetFrontendValue returns the frontend value for the specified container, using
|
func listContainers(dockerClient *docker.Client) []docker.Container {
|
||||||
// it's label. It returns a default one if the label is not present.
|
containerList, _ := dockerClient.ListContainers(docker.ListContainersOptions{})
|
||||||
func (provider *Docker) GetFrontendValue(container docker.Container) string {
|
containersInspected := []docker.Container{}
|
||||||
if label, err := provider.getLabel(container, "traefik.frontend.value"); err == nil {
|
|
||||||
return label
|
|
||||||
}
|
|
||||||
return provider.getEscapedName(container.Name) + "." + provider.Domain
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetFrontendRule returns the frontend rule for the specified container, using
|
// get inspect containers
|
||||||
// it's label. It returns a default one (Host) if the label is not present.
|
for _, container := range containerList {
|
||||||
func (provider *Docker) GetFrontendRule(container docker.Container) string {
|
containerInspected, _ := dockerClient.InspectContainer(container.ID)
|
||||||
if label, err := provider.getLabel(container, "traefik.frontend.rule"); err == nil {
|
containersInspected = append(containersInspected, *containerInspected)
|
||||||
return label
|
|
||||||
}
|
}
|
||||||
return "Host"
|
return containersInspected
|
||||||
}
|
}
|
||||||
|
|
788
provider/docker_test.go
Normal file
788
provider/docker_test.go
Normal file
|
@ -0,0 +1,788 @@
|
||||||
|
package provider
|
||||||
|
|
||||||
|
import (
|
||||||
|
"reflect"
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/emilevauge/traefik/types"
|
||||||
|
"github.com/fsouza/go-dockerclient"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestDockerGetFrontendName(t *testing.T) {
|
||||||
|
provider := &Docker{
|
||||||
|
Domain: "docker.localhost",
|
||||||
|
}
|
||||||
|
|
||||||
|
containers := []struct {
|
||||||
|
container docker.Container
|
||||||
|
expected string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Name: "foo",
|
||||||
|
Config: &docker.Config{},
|
||||||
|
},
|
||||||
|
expected: "Host-foo-docker-localhost",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Name: "bar",
|
||||||
|
Config: &docker.Config{
|
||||||
|
Labels: map[string]string{
|
||||||
|
"traefik.frontend.rule": "Header",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: "Header-bar-docker-localhost",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Name: "test",
|
||||||
|
Config: &docker.Config{
|
||||||
|
Labels: map[string]string{
|
||||||
|
"traefik.frontend.value": "foo.bar",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: "Host-foo-bar",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Name: "test",
|
||||||
|
Config: &docker.Config{
|
||||||
|
Labels: map[string]string{
|
||||||
|
"traefik.frontend.value": "foo.bar",
|
||||||
|
"traefik.frontend.rule": "Header",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: "Header-foo-bar",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Name: "test",
|
||||||
|
Config: &docker.Config{
|
||||||
|
Labels: map[string]string{
|
||||||
|
"traefik.frontend.value": "[foo.bar]",
|
||||||
|
"traefik.frontend.rule": "Header",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: "Header-foo-bar",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, e := range containers {
|
||||||
|
actual := provider.getFrontendName(e.container)
|
||||||
|
if actual != e.expected {
|
||||||
|
t.Fatalf("expected %q, got %q", e.expected, actual)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestDockerGetFrontendValue(t *testing.T) {
|
||||||
|
provider := &Docker{
|
||||||
|
Domain: "docker.localhost",
|
||||||
|
}
|
||||||
|
|
||||||
|
containers := []struct {
|
||||||
|
container docker.Container
|
||||||
|
expected string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Name: "foo",
|
||||||
|
Config: &docker.Config{},
|
||||||
|
},
|
||||||
|
expected: "foo.docker.localhost",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Name: "bar",
|
||||||
|
Config: &docker.Config{},
|
||||||
|
},
|
||||||
|
expected: "bar.docker.localhost",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Name: "test",
|
||||||
|
Config: &docker.Config{
|
||||||
|
Labels: map[string]string{
|
||||||
|
"traefik.frontend.value": "foo.bar",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: "foo.bar",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, e := range containers {
|
||||||
|
actual := provider.getFrontendValue(e.container)
|
||||||
|
if actual != e.expected {
|
||||||
|
t.Fatalf("expected %q, got %q", e.expected, actual)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestDockerGetFrontendRule(t *testing.T) {
|
||||||
|
provider := &Docker{}
|
||||||
|
|
||||||
|
containers := []struct {
|
||||||
|
container docker.Container
|
||||||
|
expected string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Name: "foo",
|
||||||
|
Config: &docker.Config{},
|
||||||
|
},
|
||||||
|
expected: "Host",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Name: "test",
|
||||||
|
Config: &docker.Config{
|
||||||
|
Labels: map[string]string{
|
||||||
|
"traefik.frontend.rule": "foo",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: "foo",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, e := range containers {
|
||||||
|
actual := provider.getFrontendRule(e.container)
|
||||||
|
if actual != e.expected {
|
||||||
|
t.Fatalf("expected %q, got %q", e.expected, actual)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestDockerGetBackend(t *testing.T) {
|
||||||
|
provider := &Docker{}
|
||||||
|
|
||||||
|
containers := []struct {
|
||||||
|
container docker.Container
|
||||||
|
expected string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Name: "foo",
|
||||||
|
Config: &docker.Config{},
|
||||||
|
},
|
||||||
|
expected: "foo",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Name: "bar",
|
||||||
|
Config: &docker.Config{},
|
||||||
|
},
|
||||||
|
expected: "bar",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Name: "test",
|
||||||
|
Config: &docker.Config{
|
||||||
|
Labels: map[string]string{
|
||||||
|
"traefik.backend": "foobar",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: "foobar",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, e := range containers {
|
||||||
|
actual := provider.getBackend(e.container)
|
||||||
|
if actual != e.expected {
|
||||||
|
t.Fatalf("expected %q, got %q", e.expected, actual)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestDockerGetPort(t *testing.T) {
|
||||||
|
provider := &Docker{}
|
||||||
|
|
||||||
|
containers := []struct {
|
||||||
|
container docker.Container
|
||||||
|
expected string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Name: "foo",
|
||||||
|
Config: &docker.Config{},
|
||||||
|
NetworkSettings: &docker.NetworkSettings{},
|
||||||
|
},
|
||||||
|
expected: "",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Name: "bar",
|
||||||
|
Config: &docker.Config{},
|
||||||
|
NetworkSettings: &docker.NetworkSettings{
|
||||||
|
Ports: map[docker.Port][]docker.PortBinding{
|
||||||
|
"80/tcp": {},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: "80",
|
||||||
|
},
|
||||||
|
// FIXME handle this better..
|
||||||
|
// {
|
||||||
|
// container: docker.Container{
|
||||||
|
// Name: "bar",
|
||||||
|
// Config: &docker.Config{},
|
||||||
|
// NetworkSettings: &docker.NetworkSettings{
|
||||||
|
// Ports: map[docker.Port][]docker.PortBinding{
|
||||||
|
// "80/tcp": []docker.PortBinding{},
|
||||||
|
// "443/tcp": []docker.PortBinding{},
|
||||||
|
// },
|
||||||
|
// },
|
||||||
|
// },
|
||||||
|
// expected: "80",
|
||||||
|
// },
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Name: "test",
|
||||||
|
Config: &docker.Config{
|
||||||
|
Labels: map[string]string{
|
||||||
|
"traefik.port": "8080",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
NetworkSettings: &docker.NetworkSettings{
|
||||||
|
Ports: map[docker.Port][]docker.PortBinding{
|
||||||
|
"80/tcp": {},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: "8080",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, e := range containers {
|
||||||
|
actual := provider.getPort(e.container)
|
||||||
|
if actual != e.expected {
|
||||||
|
t.Fatalf("expected %q, got %q", e.expected, actual)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestDockerGetWeight(t *testing.T) {
|
||||||
|
provider := &Docker{}
|
||||||
|
|
||||||
|
containers := []struct {
|
||||||
|
container docker.Container
|
||||||
|
expected string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Name: "foo",
|
||||||
|
Config: &docker.Config{},
|
||||||
|
},
|
||||||
|
expected: "0",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Name: "test",
|
||||||
|
Config: &docker.Config{
|
||||||
|
Labels: map[string]string{
|
||||||
|
"traefik.weight": "10",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: "10",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, e := range containers {
|
||||||
|
actual := provider.getWeight(e.container)
|
||||||
|
if actual != e.expected {
|
||||||
|
t.Fatalf("expected %q, got %q", e.expected, actual)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestDockerGetDomain(t *testing.T) {
|
||||||
|
provider := &Docker{
|
||||||
|
Domain: "docker.localhost",
|
||||||
|
}
|
||||||
|
|
||||||
|
containers := []struct {
|
||||||
|
container docker.Container
|
||||||
|
expected string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Name: "foo",
|
||||||
|
Config: &docker.Config{},
|
||||||
|
},
|
||||||
|
expected: "docker.localhost",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Name: "test",
|
||||||
|
Config: &docker.Config{
|
||||||
|
Labels: map[string]string{
|
||||||
|
"traefik.domain": "foo.bar",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: "foo.bar",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, e := range containers {
|
||||||
|
actual := provider.getDomain(e.container)
|
||||||
|
if actual != e.expected {
|
||||||
|
t.Fatalf("expected %q, got %q", e.expected, actual)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestDockerGetProtocol(t *testing.T) {
|
||||||
|
provider := &Docker{}
|
||||||
|
|
||||||
|
containers := []struct {
|
||||||
|
container docker.Container
|
||||||
|
expected string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Name: "foo",
|
||||||
|
Config: &docker.Config{},
|
||||||
|
},
|
||||||
|
expected: "http",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Name: "test",
|
||||||
|
Config: &docker.Config{
|
||||||
|
Labels: map[string]string{
|
||||||
|
"traefik.protocol": "https",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: "https",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, e := range containers {
|
||||||
|
actual := provider.getProtocol(e.container)
|
||||||
|
if actual != e.expected {
|
||||||
|
t.Fatalf("expected %q, got %q", e.expected, actual)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestDockerGetPassHostHeader(t *testing.T) {
|
||||||
|
provider := &Docker{}
|
||||||
|
|
||||||
|
containers := []struct {
|
||||||
|
container docker.Container
|
||||||
|
expected string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Name: "foo",
|
||||||
|
Config: &docker.Config{},
|
||||||
|
},
|
||||||
|
expected: "false",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Name: "test",
|
||||||
|
Config: &docker.Config{
|
||||||
|
Labels: map[string]string{
|
||||||
|
"traefik.frontend.passHostHeader": "true",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: "true",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, e := range containers {
|
||||||
|
actual := provider.getPassHostHeader(e.container)
|
||||||
|
if actual != e.expected {
|
||||||
|
t.Fatalf("expected %q, got %q", e.expected, actual)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestDockerGetLabel(t *testing.T) {
|
||||||
|
containers := []struct {
|
||||||
|
container docker.Container
|
||||||
|
expected string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Config: &docker.Config{},
|
||||||
|
},
|
||||||
|
expected: "Label not found:",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Config: &docker.Config{
|
||||||
|
Labels: map[string]string{
|
||||||
|
"foo": "bar",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: "",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, e := range containers {
|
||||||
|
label, err := getLabel(e.container, "foo")
|
||||||
|
if e.expected != "" {
|
||||||
|
if err == nil || !strings.Contains(err.Error(), e.expected) {
|
||||||
|
t.Fatalf("expected an error with %q, got %v", e.expected, err)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
if label != "bar" {
|
||||||
|
t.Fatalf("expected label 'bar', got %s", label)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestDockerGetLabels(t *testing.T) {
|
||||||
|
containers := []struct {
|
||||||
|
container docker.Container
|
||||||
|
expectedLabels map[string]string
|
||||||
|
expectedError string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Config: &docker.Config{},
|
||||||
|
},
|
||||||
|
expectedLabels: map[string]string{},
|
||||||
|
expectedError: "Label not found:",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Config: &docker.Config{
|
||||||
|
Labels: map[string]string{
|
||||||
|
"foo": "fooz",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expectedLabels: map[string]string{
|
||||||
|
"foo": "fooz",
|
||||||
|
},
|
||||||
|
expectedError: "Label not found: bar",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Config: &docker.Config{
|
||||||
|
Labels: map[string]string{
|
||||||
|
"foo": "fooz",
|
||||||
|
"bar": "barz",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expectedLabels: map[string]string{
|
||||||
|
"foo": "fooz",
|
||||||
|
"bar": "barz",
|
||||||
|
},
|
||||||
|
expectedError: "",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, e := range containers {
|
||||||
|
labels, err := getLabels(e.container, []string{"foo", "bar"})
|
||||||
|
if !reflect.DeepEqual(labels, e.expectedLabels) {
|
||||||
|
t.Fatalf("expect %v, got %v", e.expectedLabels, labels)
|
||||||
|
}
|
||||||
|
if e.expectedError != "" {
|
||||||
|
if err == nil || !strings.Contains(err.Error(), e.expectedError) {
|
||||||
|
t.Fatalf("expected an error with %q, got %v", e.expectedError, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestDockerTraefikFilter(t *testing.T) {
|
||||||
|
containers := []struct {
|
||||||
|
container docker.Container
|
||||||
|
expected bool
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Config: &docker.Config{},
|
||||||
|
NetworkSettings: &docker.NetworkSettings{},
|
||||||
|
},
|
||||||
|
expected: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Config: &docker.Config{
|
||||||
|
Labels: map[string]string{
|
||||||
|
"traefik.enable": "false",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
NetworkSettings: &docker.NetworkSettings{
|
||||||
|
Ports: map[docker.Port][]docker.PortBinding{
|
||||||
|
"80/tcp": {},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Config: &docker.Config{
|
||||||
|
Labels: map[string]string{
|
||||||
|
"traefik.frontend.rule": "Host",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
NetworkSettings: &docker.NetworkSettings{
|
||||||
|
Ports: map[docker.Port][]docker.PortBinding{
|
||||||
|
"80/tcp": {},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Config: &docker.Config{
|
||||||
|
Labels: map[string]string{
|
||||||
|
"traefik.frontend.value": "foo.bar",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
NetworkSettings: &docker.NetworkSettings{
|
||||||
|
Ports: map[docker.Port][]docker.PortBinding{
|
||||||
|
"80/tcp": {},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Config: &docker.Config{},
|
||||||
|
NetworkSettings: &docker.NetworkSettings{
|
||||||
|
Ports: map[docker.Port][]docker.PortBinding{
|
||||||
|
"80/tcp": {},
|
||||||
|
"443/tcp": {},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Config: &docker.Config{},
|
||||||
|
NetworkSettings: &docker.NetworkSettings{
|
||||||
|
Ports: map[docker.Port][]docker.PortBinding{
|
||||||
|
"80/tcp": {},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Config: &docker.Config{
|
||||||
|
Labels: map[string]string{
|
||||||
|
"traefik.port": "80",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
NetworkSettings: &docker.NetworkSettings{
|
||||||
|
Ports: map[docker.Port][]docker.PortBinding{
|
||||||
|
"80/tcp": {},
|
||||||
|
"443/tcp": {},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Config: &docker.Config{
|
||||||
|
Labels: map[string]string{
|
||||||
|
"traefik.enable": "true",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
NetworkSettings: &docker.NetworkSettings{
|
||||||
|
Ports: map[docker.Port][]docker.PortBinding{
|
||||||
|
"80/tcp": {},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Config: &docker.Config{
|
||||||
|
Labels: map[string]string{
|
||||||
|
"traefik.enable": "anything",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
NetworkSettings: &docker.NetworkSettings{
|
||||||
|
Ports: map[docker.Port][]docker.PortBinding{
|
||||||
|
"80/tcp": {},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
container: docker.Container{
|
||||||
|
Config: &docker.Config{
|
||||||
|
Labels: map[string]string{
|
||||||
|
"traefik.frontend.rule": "Host",
|
||||||
|
"traefik.frontend.value": "foo.bar",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
NetworkSettings: &docker.NetworkSettings{
|
||||||
|
Ports: map[docker.Port][]docker.PortBinding{
|
||||||
|
"80/tcp": {},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: true,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, e := range containers {
|
||||||
|
actual := containerFilter(e.container)
|
||||||
|
if actual != e.expected {
|
||||||
|
t.Fatalf("expected %v, got %v", e.expected, actual)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestDockerLoadDockerConfig(t *testing.T) {
|
||||||
|
cases := []struct {
|
||||||
|
containers []docker.Container
|
||||||
|
expectedFrontends map[string]*types.Frontend
|
||||||
|
expectedBackends map[string]*types.Backend
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
containers: []docker.Container{},
|
||||||
|
expectedFrontends: map[string]*types.Frontend{},
|
||||||
|
expectedBackends: map[string]*types.Backend{},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
containers: []docker.Container{
|
||||||
|
{
|
||||||
|
Name: "test",
|
||||||
|
Config: &docker.Config{},
|
||||||
|
NetworkSettings: &docker.NetworkSettings{
|
||||||
|
Ports: map[docker.Port][]docker.PortBinding{
|
||||||
|
"80/tcp": {},
|
||||||
|
},
|
||||||
|
IPAddress: "127.0.0.1",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expectedFrontends: map[string]*types.Frontend{
|
||||||
|
`"frontend-Host-test-docker-localhost"`: {
|
||||||
|
Backend: "backend-test",
|
||||||
|
Routes: map[string]types.Route{
|
||||||
|
`"route-frontend-Host-test-docker-localhost"`: {
|
||||||
|
Rule: "Host",
|
||||||
|
Value: "test.docker.localhost",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expectedBackends: map[string]*types.Backend{
|
||||||
|
"backend-test": {
|
||||||
|
Servers: map[string]types.Server{
|
||||||
|
"server-test": {
|
||||||
|
URL: "http://127.0.0.1:80",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
CircuitBreaker: nil,
|
||||||
|
LoadBalancer: nil,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
containers: []docker.Container{
|
||||||
|
{
|
||||||
|
Name: "test1",
|
||||||
|
Config: &docker.Config{
|
||||||
|
Labels: map[string]string{
|
||||||
|
"traefik.backend": "foobar",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
NetworkSettings: &docker.NetworkSettings{
|
||||||
|
Ports: map[docker.Port][]docker.PortBinding{
|
||||||
|
"80/tcp": {},
|
||||||
|
},
|
||||||
|
IPAddress: "127.0.0.1",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "test2",
|
||||||
|
Config: &docker.Config{
|
||||||
|
Labels: map[string]string{
|
||||||
|
"traefik.backend": "foobar",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
NetworkSettings: &docker.NetworkSettings{
|
||||||
|
Ports: map[docker.Port][]docker.PortBinding{
|
||||||
|
"80/tcp": {},
|
||||||
|
},
|
||||||
|
IPAddress: "127.0.0.1",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expectedFrontends: map[string]*types.Frontend{
|
||||||
|
`"frontend-Host-test1-docker-localhost"`: {
|
||||||
|
Backend: "backend-foobar",
|
||||||
|
Routes: map[string]types.Route{
|
||||||
|
`"route-frontend-Host-test1-docker-localhost"`: {
|
||||||
|
Rule: "Host",
|
||||||
|
Value: "test1.docker.localhost",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
`"frontend-Host-test2-docker-localhost"`: {
|
||||||
|
Backend: "backend-foobar",
|
||||||
|
Routes: map[string]types.Route{
|
||||||
|
`"route-frontend-Host-test2-docker-localhost"`: {
|
||||||
|
Rule: "Host",
|
||||||
|
Value: "test2.docker.localhost",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expectedBackends: map[string]*types.Backend{
|
||||||
|
"backend-foobar": {
|
||||||
|
Servers: map[string]types.Server{
|
||||||
|
"server-test1": {
|
||||||
|
URL: "http://127.0.0.1:80",
|
||||||
|
},
|
||||||
|
"server-test2": {
|
||||||
|
URL: "http://127.0.0.1:80",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
CircuitBreaker: nil,
|
||||||
|
LoadBalancer: nil,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
provider := &Docker{
|
||||||
|
Domain: "docker.localhost",
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, c := range cases {
|
||||||
|
actualConfig := provider.loadDockerConfig(c.containers)
|
||||||
|
// Compare backends
|
||||||
|
if !reflect.DeepEqual(actualConfig.Backends, c.expectedBackends) {
|
||||||
|
t.Fatalf("expected %#v, got %#v", c.expectedBackends, actualConfig.Backends)
|
||||||
|
}
|
||||||
|
if !reflect.DeepEqual(actualConfig.Frontends, c.expectedFrontends) {
|
||||||
|
t.Fatalf("expected %#v, got %#v", c.expectedFrontends, actualConfig.Frontends)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
|
@ -1,19 +1,20 @@
|
||||||
package provider
|
package provider
|
||||||
|
|
||||||
import "github.com/emilevauge/traefik/types"
|
import (
|
||||||
|
"github.com/docker/libkv/store"
|
||||||
|
"github.com/docker/libkv/store/etcd"
|
||||||
|
"github.com/emilevauge/traefik/types"
|
||||||
|
)
|
||||||
|
|
||||||
// Etcd holds configurations of the Etcd provider.
|
// Etcd holds configurations of the Etcd provider.
|
||||||
type Etcd struct {
|
type Etcd struct {
|
||||||
Watch bool
|
Kv
|
||||||
Endpoint string
|
|
||||||
Prefix string
|
|
||||||
Filename string
|
|
||||||
KvProvider *Kv
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Provide allows the provider to provide configurations to traefik
|
// Provide allows the provider to provide configurations to traefik
|
||||||
// using the given configuration channel.
|
// using the given configuration channel.
|
||||||
func (provider *Etcd) Provide(configurationChan chan<- types.ConfigMessage) error {
|
func (provider *Etcd) Provide(configurationChan chan<- types.ConfigMessage) error {
|
||||||
provider.KvProvider = NewEtcdProvider(provider)
|
provider.StoreType = store.ETCD
|
||||||
return provider.KvProvider.provide(configurationChan)
|
etcd.Register()
|
||||||
|
return provider.provide(configurationChan)
|
||||||
}
|
}
|
||||||
|
|
|
@ -13,8 +13,7 @@ import (
|
||||||
|
|
||||||
// File holds configurations of the File provider.
|
// File holds configurations of the File provider.
|
||||||
type File struct {
|
type File struct {
|
||||||
Watch bool
|
baseProvider
|
||||||
Filename string
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Provide allows the provider to provide configurations to traefik
|
// Provide allows the provider to provide configurations to traefik
|
||||||
|
@ -44,7 +43,10 @@ func (provider *File) Provide(configurationChan chan<- types.ConfigMessage) erro
|
||||||
log.Debug("File event:", event)
|
log.Debug("File event:", event)
|
||||||
configuration := provider.loadFileConfig(file.Name())
|
configuration := provider.loadFileConfig(file.Name())
|
||||||
if configuration != nil {
|
if configuration != nil {
|
||||||
configurationChan <- types.ConfigMessage{"file", configuration}
|
configurationChan <- types.ConfigMessage{
|
||||||
|
ProviderName: "file",
|
||||||
|
Configuration: configuration,
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
case error := <-watcher.Errors:
|
case error := <-watcher.Errors:
|
||||||
|
@ -60,7 +62,10 @@ func (provider *File) Provide(configurationChan chan<- types.ConfigMessage) erro
|
||||||
}
|
}
|
||||||
|
|
||||||
configuration := provider.loadFileConfig(file.Name())
|
configuration := provider.loadFileConfig(file.Name())
|
||||||
configurationChan <- types.ConfigMessage{"file", configuration}
|
configurationChan <- types.ConfigMessage{
|
||||||
|
ProviderName: "file",
|
||||||
|
Configuration: configuration,
|
||||||
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
139
provider/kv.go
139
provider/kv.go
|
@ -2,92 +2,27 @@
|
||||||
package provider
|
package provider
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
|
||||||
"errors"
|
|
||||||
"strings"
|
"strings"
|
||||||
"text/template"
|
"text/template"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/BurntSushi/toml"
|
|
||||||
"github.com/BurntSushi/ty/fun"
|
"github.com/BurntSushi/ty/fun"
|
||||||
log "github.com/Sirupsen/logrus"
|
log "github.com/Sirupsen/logrus"
|
||||||
"github.com/docker/libkv"
|
"github.com/docker/libkv"
|
||||||
"github.com/docker/libkv/store"
|
"github.com/docker/libkv/store"
|
||||||
"github.com/docker/libkv/store/boltdb"
|
|
||||||
"github.com/docker/libkv/store/consul"
|
|
||||||
"github.com/docker/libkv/store/etcd"
|
|
||||||
"github.com/docker/libkv/store/zookeeper"
|
|
||||||
"github.com/emilevauge/traefik/autogen"
|
|
||||||
"github.com/emilevauge/traefik/types"
|
"github.com/emilevauge/traefik/types"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Kv holds common configurations of key-value providers.
|
// Kv holds common configurations of key-value providers.
|
||||||
type Kv struct {
|
type Kv struct {
|
||||||
Watch bool
|
baseProvider
|
||||||
Endpoint string
|
Endpoint string
|
||||||
Prefix string
|
Prefix string
|
||||||
Filename string
|
|
||||||
StoreType store.Backend
|
StoreType store.Backend
|
||||||
kvclient store.Store
|
kvclient store.Store
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewConsulProvider returns a Consul provider.
|
|
||||||
func NewConsulProvider(provider *Consul) *Kv {
|
|
||||||
kvProvider := new(Kv)
|
|
||||||
kvProvider.Watch = provider.Watch
|
|
||||||
kvProvider.Endpoint = provider.Endpoint
|
|
||||||
kvProvider.Prefix = provider.Prefix
|
|
||||||
kvProvider.Filename = provider.Filename
|
|
||||||
kvProvider.StoreType = store.CONSUL
|
|
||||||
return kvProvider
|
|
||||||
}
|
|
||||||
|
|
||||||
// NewEtcdProvider returns a Etcd provider.
|
|
||||||
func NewEtcdProvider(provider *Etcd) *Kv {
|
|
||||||
kvProvider := new(Kv)
|
|
||||||
kvProvider.Watch = provider.Watch
|
|
||||||
kvProvider.Endpoint = provider.Endpoint
|
|
||||||
kvProvider.Prefix = provider.Prefix
|
|
||||||
kvProvider.Filename = provider.Filename
|
|
||||||
kvProvider.StoreType = store.ETCD
|
|
||||||
return kvProvider
|
|
||||||
}
|
|
||||||
|
|
||||||
// NewZkProvider returns a Zookepper provider.
|
|
||||||
func NewZkProvider(provider *Zookepper) *Kv {
|
|
||||||
kvProvider := new(Kv)
|
|
||||||
kvProvider.Watch = provider.Watch
|
|
||||||
kvProvider.Endpoint = provider.Endpoint
|
|
||||||
kvProvider.Prefix = provider.Prefix
|
|
||||||
kvProvider.Filename = provider.Filename
|
|
||||||
kvProvider.StoreType = store.ZK
|
|
||||||
return kvProvider
|
|
||||||
}
|
|
||||||
|
|
||||||
// NewBoltDbProvider returns a BoldDb provider.
|
|
||||||
func NewBoltDbProvider(provider *BoltDb) *Kv {
|
|
||||||
kvProvider := new(Kv)
|
|
||||||
kvProvider.Watch = provider.Watch
|
|
||||||
kvProvider.Endpoint = provider.Endpoint
|
|
||||||
kvProvider.Prefix = provider.Prefix
|
|
||||||
kvProvider.Filename = provider.Filename
|
|
||||||
kvProvider.StoreType = store.BOLTDB
|
|
||||||
return kvProvider
|
|
||||||
}
|
|
||||||
|
|
||||||
func (provider *Kv) provide(configurationChan chan<- types.ConfigMessage) error {
|
func (provider *Kv) provide(configurationChan chan<- types.ConfigMessage) error {
|
||||||
switch provider.StoreType {
|
|
||||||
case store.CONSUL:
|
|
||||||
consul.Register()
|
|
||||||
case store.ETCD:
|
|
||||||
etcd.Register()
|
|
||||||
case store.ZK:
|
|
||||||
zookeeper.Register()
|
|
||||||
case store.BOLTDB:
|
|
||||||
boltdb.Register()
|
|
||||||
default:
|
|
||||||
return errors.New("Invalid kv store: " + string(provider.StoreType))
|
|
||||||
}
|
|
||||||
kv, err := libkv.NewStore(
|
kv, err := libkv.NewStore(
|
||||||
provider.StoreType,
|
provider.StoreType,
|
||||||
[]string{provider.Endpoint},
|
[]string{provider.Endpoint},
|
||||||
|
@ -114,26 +49,43 @@ func (provider *Kv) provide(configurationChan chan<- types.ConfigMessage) error
|
||||||
<-chanKeys
|
<-chanKeys
|
||||||
configuration := provider.loadConfig()
|
configuration := provider.loadConfig()
|
||||||
if configuration != nil {
|
if configuration != nil {
|
||||||
configurationChan <- types.ConfigMessage{string(provider.StoreType), configuration}
|
configurationChan <- types.ConfigMessage{
|
||||||
|
ProviderName: string(provider.StoreType),
|
||||||
|
Configuration: configuration,
|
||||||
|
}
|
||||||
}
|
}
|
||||||
defer close(stopCh)
|
defer close(stopCh)
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
}
|
}
|
||||||
configuration := provider.loadConfig()
|
configuration := provider.loadConfig()
|
||||||
configurationChan <- types.ConfigMessage{string(provider.StoreType), configuration}
|
configurationChan <- types.ConfigMessage{
|
||||||
|
ProviderName: string(provider.StoreType),
|
||||||
|
Configuration: configuration,
|
||||||
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (provider *Kv) loadConfig() *types.Configuration {
|
func (provider *Kv) loadConfig() *types.Configuration {
|
||||||
configuration := new(types.Configuration)
|
|
||||||
templateObjects := struct {
|
templateObjects := struct {
|
||||||
Prefix string
|
Prefix string
|
||||||
}{
|
}{
|
||||||
provider.Prefix,
|
provider.Prefix,
|
||||||
}
|
}
|
||||||
var KvFuncMap = template.FuncMap{
|
var KvFuncMap = template.FuncMap{
|
||||||
"List": func(keys ...string) []string {
|
"List": provider.list,
|
||||||
|
"Get": provider.get,
|
||||||
|
"Last": provider.last,
|
||||||
|
}
|
||||||
|
|
||||||
|
configuration, err := provider.getConfiguration("templates/kv.tmpl", KvFuncMap, templateObjects)
|
||||||
|
if err != nil {
|
||||||
|
log.Error(err)
|
||||||
|
}
|
||||||
|
return configuration
|
||||||
|
}
|
||||||
|
|
||||||
|
func (provider *Kv) list(keys ...string) []string {
|
||||||
joinedKeys := strings.Join(keys, "")
|
joinedKeys := strings.Join(keys, "")
|
||||||
keysPairs, err := provider.kvclient.List(joinedKeys)
|
keysPairs, err := provider.kvclient.List(joinedKeys)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
@ -146,8 +98,9 @@ func (provider *Kv) loadConfig() *types.Configuration {
|
||||||
directoryKeys[directory] = joinedKeys + directory
|
directoryKeys[directory] = joinedKeys + directory
|
||||||
}
|
}
|
||||||
return fun.Values(directoryKeys).([]string)
|
return fun.Values(directoryKeys).([]string)
|
||||||
},
|
}
|
||||||
"Get": func(keys ...string) string {
|
|
||||||
|
func (provider *Kv) get(keys ...string) string {
|
||||||
joinedKeys := strings.Join(keys, "")
|
joinedKeys := strings.Join(keys, "")
|
||||||
keyPair, err := provider.kvclient.Get(joinedKeys)
|
keyPair, err := provider.kvclient.Get(joinedKeys)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
@ -157,45 +110,9 @@ func (provider *Kv) loadConfig() *types.Configuration {
|
||||||
return ""
|
return ""
|
||||||
}
|
}
|
||||||
return string(keyPair.Value)
|
return string(keyPair.Value)
|
||||||
},
|
}
|
||||||
"Last": func(key string) string {
|
|
||||||
|
func (provider *Kv) last(key string) string {
|
||||||
splittedKey := strings.Split(key, "/")
|
splittedKey := strings.Split(key, "/")
|
||||||
return splittedKey[len(splittedKey)-1]
|
return splittedKey[len(splittedKey)-1]
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
tmpl := template.New(provider.Filename).Funcs(KvFuncMap)
|
|
||||||
if len(provider.Filename) > 0 {
|
|
||||||
_, err := tmpl.ParseFiles(provider.Filename)
|
|
||||||
if err != nil {
|
|
||||||
log.Error("Error reading file", err)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
buf, err := autogen.Asset("templates/kv.tmpl")
|
|
||||||
if err != nil {
|
|
||||||
log.Error("Error reading file", err)
|
|
||||||
}
|
|
||||||
_, err = tmpl.Parse(string(buf))
|
|
||||||
if err != nil {
|
|
||||||
log.Error("Error reading file", err)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
var buffer bytes.Buffer
|
|
||||||
|
|
||||||
err := tmpl.Execute(&buffer, templateObjects)
|
|
||||||
if err != nil {
|
|
||||||
log.Error("Error with kv template:", err)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
if _, err := toml.Decode(buffer.String(), configuration); err != nil {
|
|
||||||
log.Error("Error creating kv configuration:", err)
|
|
||||||
log.Error(buffer.String())
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
return configuration
|
|
||||||
}
|
}
|
||||||
|
|
312
provider/kv_test.go
Normal file
312
provider/kv_test.go
Normal file
|
@ -0,0 +1,312 @@
|
||||||
|
package provider
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/docker/libkv/store"
|
||||||
|
"reflect"
|
||||||
|
"sort"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestKvList(t *testing.T) {
|
||||||
|
cases := []struct {
|
||||||
|
provider *Kv
|
||||||
|
keys []string
|
||||||
|
expected []string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
provider: &Kv{
|
||||||
|
kvclient: &Mock{},
|
||||||
|
},
|
||||||
|
keys: []string{},
|
||||||
|
expected: []string{},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
provider: &Kv{
|
||||||
|
kvclient: &Mock{},
|
||||||
|
},
|
||||||
|
keys: []string{"traefik"},
|
||||||
|
expected: []string{},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
provider: &Kv{
|
||||||
|
kvclient: &Mock{
|
||||||
|
KVPairs: []*store.KVPair{
|
||||||
|
{
|
||||||
|
Key: "foo",
|
||||||
|
Value: []byte("bar"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
keys: []string{"bar"},
|
||||||
|
expected: []string{},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
provider: &Kv{
|
||||||
|
kvclient: &Mock{
|
||||||
|
KVPairs: []*store.KVPair{
|
||||||
|
{
|
||||||
|
Key: "foo",
|
||||||
|
Value: []byte("bar"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
keys: []string{"foo"},
|
||||||
|
expected: []string{"foo"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
provider: &Kv{
|
||||||
|
kvclient: &Mock{
|
||||||
|
KVPairs: []*store.KVPair{
|
||||||
|
{
|
||||||
|
Key: "foo/baz/1",
|
||||||
|
Value: []byte("bar"),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Key: "foo/baz/2",
|
||||||
|
Value: []byte("bar"),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Key: "foo/baz/biz/1",
|
||||||
|
Value: []byte("bar"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
keys: []string{"foo", "/baz/"},
|
||||||
|
expected: []string{"foo/baz/biz", "foo/baz/1", "foo/baz/2"},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, c := range cases {
|
||||||
|
actual := c.provider.list(c.keys...)
|
||||||
|
sort.Strings(actual)
|
||||||
|
sort.Strings(c.expected)
|
||||||
|
if !reflect.DeepEqual(actual, c.expected) {
|
||||||
|
t.Fatalf("expected %v, got %v for %v and %v", c.expected, actual, c.keys, c.provider)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Error case
|
||||||
|
provider := &Kv{
|
||||||
|
kvclient: &Mock{
|
||||||
|
Error: true,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
actual := provider.list("anything")
|
||||||
|
if actual != nil {
|
||||||
|
t.Fatalf("Should have return nil, got %v", actual)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestKvGet(t *testing.T) {
|
||||||
|
cases := []struct {
|
||||||
|
provider *Kv
|
||||||
|
keys []string
|
||||||
|
expected string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
provider: &Kv{
|
||||||
|
kvclient: &Mock{},
|
||||||
|
},
|
||||||
|
keys: []string{},
|
||||||
|
expected: "",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
provider: &Kv{
|
||||||
|
kvclient: &Mock{},
|
||||||
|
},
|
||||||
|
keys: []string{"traefik"},
|
||||||
|
expected: "",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
provider: &Kv{
|
||||||
|
kvclient: &Mock{
|
||||||
|
KVPairs: []*store.KVPair{
|
||||||
|
{
|
||||||
|
Key: "foo",
|
||||||
|
Value: []byte("bar"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
keys: []string{"bar"},
|
||||||
|
expected: "",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
provider: &Kv{
|
||||||
|
kvclient: &Mock{
|
||||||
|
KVPairs: []*store.KVPair{
|
||||||
|
{
|
||||||
|
Key: "foo",
|
||||||
|
Value: []byte("bar"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
keys: []string{"foo"},
|
||||||
|
expected: "bar",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
provider: &Kv{
|
||||||
|
kvclient: &Mock{
|
||||||
|
KVPairs: []*store.KVPair{
|
||||||
|
{
|
||||||
|
Key: "foo/baz/1",
|
||||||
|
Value: []byte("bar1"),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Key: "foo/baz/2",
|
||||||
|
Value: []byte("bar2"),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Key: "foo/baz/biz/1",
|
||||||
|
Value: []byte("bar3"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
keys: []string{"foo", "/baz/", "2"},
|
||||||
|
expected: "bar2",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, c := range cases {
|
||||||
|
actual := c.provider.get(c.keys...)
|
||||||
|
if actual != c.expected {
|
||||||
|
t.Fatalf("expected %v, got %v for %v and %v", c.expected, actual, c.keys, c.provider)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Error case
|
||||||
|
provider := &Kv{
|
||||||
|
kvclient: &Mock{
|
||||||
|
Error: true,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
actual := provider.get("anything")
|
||||||
|
if actual != "" {
|
||||||
|
t.Fatalf("Should have return nil, got %v", actual)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestKvLast(t *testing.T) {
|
||||||
|
cases := []struct {
|
||||||
|
key string
|
||||||
|
expected string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
key: "",
|
||||||
|
expected: "",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
key: "foo",
|
||||||
|
expected: "foo",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
key: "foo/bar",
|
||||||
|
expected: "bar",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
key: "foo/bar/baz",
|
||||||
|
expected: "baz",
|
||||||
|
},
|
||||||
|
// FIXME is this wanted ?
|
||||||
|
{
|
||||||
|
key: "foo/bar/",
|
||||||
|
expected: "",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
provider := &Kv{}
|
||||||
|
for _, c := range cases {
|
||||||
|
actual := provider.last(c.key)
|
||||||
|
if actual != c.expected {
|
||||||
|
t.Fatalf("expected %s, got %s", c.expected, actual)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Extremely limited mock store so we can test initialization
|
||||||
|
type Mock struct {
|
||||||
|
Error bool
|
||||||
|
KVPairs []*store.KVPair
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Mock) Put(key string, value []byte, opts *store.WriteOptions) error {
|
||||||
|
return errors.New("Put not supported")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Mock) Get(key string) (*store.KVPair, error) {
|
||||||
|
if s.Error {
|
||||||
|
return nil, errors.New("Error")
|
||||||
|
}
|
||||||
|
for _, kvPair := range s.KVPairs {
|
||||||
|
if kvPair.Key == key {
|
||||||
|
return kvPair, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Mock) Delete(key string) error {
|
||||||
|
return errors.New("Delete not supported")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Exists mock
|
||||||
|
func (s *Mock) Exists(key string) (bool, error) {
|
||||||
|
return false, errors.New("Exists not supported")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Watch mock
|
||||||
|
func (s *Mock) Watch(key string, stopCh <-chan struct{}) (<-chan *store.KVPair, error) {
|
||||||
|
return nil, errors.New("Watch not supported")
|
||||||
|
}
|
||||||
|
|
||||||
|
// WatchTree mock
|
||||||
|
func (s *Mock) WatchTree(prefix string, stopCh <-chan struct{}) (<-chan []*store.KVPair, error) {
|
||||||
|
return nil, errors.New("WatchTree not supported")
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewLock mock
|
||||||
|
func (s *Mock) NewLock(key string, options *store.LockOptions) (store.Locker, error) {
|
||||||
|
return nil, errors.New("NewLock not supported")
|
||||||
|
}
|
||||||
|
|
||||||
|
// List mock
|
||||||
|
func (s *Mock) List(prefix string) ([]*store.KVPair, error) {
|
||||||
|
if s.Error {
|
||||||
|
return nil, errors.New("Error")
|
||||||
|
}
|
||||||
|
kv := []*store.KVPair{}
|
||||||
|
for _, kvPair := range s.KVPairs {
|
||||||
|
if strings.HasPrefix(kvPair.Key, prefix) {
|
||||||
|
kv = append(kv, kvPair)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return kv, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// DeleteTree mock
|
||||||
|
func (s *Mock) DeleteTree(prefix string) error {
|
||||||
|
return errors.New("DeleteTree not supported")
|
||||||
|
}
|
||||||
|
|
||||||
|
// AtomicPut mock
|
||||||
|
func (s *Mock) AtomicPut(key string, value []byte, previous *store.KVPair, opts *store.WriteOptions) (bool, *store.KVPair, error) {
|
||||||
|
return false, nil, errors.New("AtomicPut not supported")
|
||||||
|
}
|
||||||
|
|
||||||
|
// AtomicDelete mock
|
||||||
|
func (s *Mock) AtomicDelete(key string, previous *store.KVPair) (bool, error) {
|
||||||
|
return false, errors.New("AtomicDelete not supported")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Close mock
|
||||||
|
func (s *Mock) Close() {
|
||||||
|
return
|
||||||
|
}
|
|
@ -1,28 +1,29 @@
|
||||||
package provider
|
package provider
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
|
||||||
"errors"
|
"errors"
|
||||||
|
"net/url"
|
||||||
"strconv"
|
"strconv"
|
||||||
"strings"
|
|
||||||
"text/template"
|
"text/template"
|
||||||
|
|
||||||
"github.com/BurntSushi/toml"
|
|
||||||
"github.com/BurntSushi/ty/fun"
|
"github.com/BurntSushi/ty/fun"
|
||||||
log "github.com/Sirupsen/logrus"
|
log "github.com/Sirupsen/logrus"
|
||||||
"github.com/emilevauge/traefik/autogen"
|
|
||||||
"github.com/emilevauge/traefik/types"
|
"github.com/emilevauge/traefik/types"
|
||||||
"github.com/gambol99/go-marathon"
|
"github.com/gambol99/go-marathon"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Marathon holds configuration of the Marathon provider.
|
// Marathon holds configuration of the Marathon provider.
|
||||||
type Marathon struct {
|
type Marathon struct {
|
||||||
Watch bool
|
baseProvider
|
||||||
Endpoint string
|
Endpoint string
|
||||||
Domain string
|
Domain string
|
||||||
Filename string
|
|
||||||
NetworkInterface string
|
NetworkInterface string
|
||||||
marathonClient marathon.Marathon
|
marathonClient lightMarathonClient
|
||||||
|
}
|
||||||
|
|
||||||
|
type lightMarathonClient interface {
|
||||||
|
Applications(url.Values) (*marathon.Applications, error)
|
||||||
|
AllTasks() (*marathon.Tasks, error)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Provide allows the provider to provide configurations to traefik
|
// Provide allows the provider to provide configurations to traefik
|
||||||
|
@ -48,7 +49,10 @@ func (provider *Marathon) Provide(configurationChan chan<- types.ConfigMessage)
|
||||||
log.Debug("Marathon event receveived", event)
|
log.Debug("Marathon event receveived", event)
|
||||||
configuration := provider.loadMarathonConfig()
|
configuration := provider.loadMarathonConfig()
|
||||||
if configuration != nil {
|
if configuration != nil {
|
||||||
configurationChan <- types.ConfigMessage{"marathon", configuration}
|
configurationChan <- types.ConfigMessage{
|
||||||
|
ProviderName: "marathon",
|
||||||
|
Configuration: configuration,
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
|
@ -56,59 +60,24 @@ func (provider *Marathon) Provide(configurationChan chan<- types.ConfigMessage)
|
||||||
}
|
}
|
||||||
|
|
||||||
configuration := provider.loadMarathonConfig()
|
configuration := provider.loadMarathonConfig()
|
||||||
configurationChan <- types.ConfigMessage{"marathon", configuration}
|
configurationChan <- types.ConfigMessage{
|
||||||
|
ProviderName: "marathon",
|
||||||
|
Configuration: configuration,
|
||||||
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (provider *Marathon) loadMarathonConfig() *types.Configuration {
|
func (provider *Marathon) loadMarathonConfig() *types.Configuration {
|
||||||
var MarathonFuncMap = template.FuncMap{
|
var MarathonFuncMap = template.FuncMap{
|
||||||
"getPort": func(task marathon.Task) string {
|
"getPort": provider.getPort,
|
||||||
for _, port := range task.Ports {
|
"getWeight": provider.getWeight,
|
||||||
return strconv.Itoa(port)
|
"getDomain": provider.getDomain,
|
||||||
|
"getProtocol": provider.getProtocol,
|
||||||
|
"getPassHostHeader": provider.getPassHostHeader,
|
||||||
|
"getFrontendValue": provider.getFrontendValue,
|
||||||
|
"getFrontendRule": provider.getFrontendRule,
|
||||||
|
"replace": replace,
|
||||||
}
|
}
|
||||||
return ""
|
|
||||||
},
|
|
||||||
"getWeight": func(task marathon.Task, applications []marathon.Application) string {
|
|
||||||
application, errApp := getApplication(task, applications)
|
|
||||||
if errApp != nil {
|
|
||||||
log.Errorf("Unable to get marathon application from task %s", task.AppID)
|
|
||||||
return "0"
|
|
||||||
}
|
|
||||||
if label, err := provider.getLabel(application, "traefik.weight"); err == nil {
|
|
||||||
return label
|
|
||||||
}
|
|
||||||
return "0"
|
|
||||||
},
|
|
||||||
"getDomain": func(application marathon.Application) string {
|
|
||||||
if label, err := provider.getLabel(application, "traefik.domain"); err == nil {
|
|
||||||
return label
|
|
||||||
}
|
|
||||||
return provider.Domain
|
|
||||||
},
|
|
||||||
"replace": func(s1 string, s2 string, s3 string) string {
|
|
||||||
return strings.Replace(s3, s1, s2, -1)
|
|
||||||
},
|
|
||||||
"getProtocol": func(task marathon.Task, applications []marathon.Application) string {
|
|
||||||
application, errApp := getApplication(task, applications)
|
|
||||||
if errApp != nil {
|
|
||||||
log.Errorf("Unable to get marathon application from task %s", task.AppID)
|
|
||||||
return "http"
|
|
||||||
}
|
|
||||||
if label, err := provider.getLabel(application, "traefik.protocol"); err == nil {
|
|
||||||
return label
|
|
||||||
}
|
|
||||||
return "http"
|
|
||||||
},
|
|
||||||
"getPassHostHeader": func(application marathon.Application) string {
|
|
||||||
if passHostHeader, err := provider.getLabel(application, "traefik.frontend.passHostHeader"); err == nil {
|
|
||||||
return passHostHeader
|
|
||||||
}
|
|
||||||
return "false"
|
|
||||||
},
|
|
||||||
"getFrontendValue": provider.GetFrontendValue,
|
|
||||||
"getFrontendRule": provider.GetFrontendRule,
|
|
||||||
}
|
|
||||||
configuration := new(types.Configuration)
|
|
||||||
|
|
||||||
applications, err := provider.marathonClient.Applications(nil)
|
applications, err := provider.marathonClient.Applications(nil)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
@ -124,6 +93,32 @@ func (provider *Marathon) loadMarathonConfig() *types.Configuration {
|
||||||
|
|
||||||
//filter tasks
|
//filter tasks
|
||||||
filteredTasks := fun.Filter(func(task marathon.Task) bool {
|
filteredTasks := fun.Filter(func(task marathon.Task) bool {
|
||||||
|
return taskFilter(task, applications)
|
||||||
|
}, tasks.Tasks).([]marathon.Task)
|
||||||
|
|
||||||
|
//filter apps
|
||||||
|
filteredApps := fun.Filter(func(app marathon.Application) bool {
|
||||||
|
return applicationFilter(app, filteredTasks)
|
||||||
|
}, applications.Apps).([]marathon.Application)
|
||||||
|
|
||||||
|
templateObjects := struct {
|
||||||
|
Applications []marathon.Application
|
||||||
|
Tasks []marathon.Task
|
||||||
|
Domain string
|
||||||
|
}{
|
||||||
|
filteredApps,
|
||||||
|
filteredTasks,
|
||||||
|
provider.Domain,
|
||||||
|
}
|
||||||
|
|
||||||
|
configuration, err := provider.getConfiguration("templates/marathon.tmpl", MarathonFuncMap, templateObjects)
|
||||||
|
if err != nil {
|
||||||
|
log.Error(err)
|
||||||
|
}
|
||||||
|
return configuration
|
||||||
|
}
|
||||||
|
|
||||||
|
func taskFilter(task marathon.Task, applications *marathon.Applications) bool {
|
||||||
if len(task.Ports) == 0 {
|
if len(task.Ports) == 0 {
|
||||||
log.Debug("Filtering marathon task without port %s", task.AppID)
|
log.Debug("Filtering marathon task without port %s", task.AppID)
|
||||||
return false
|
return false
|
||||||
|
@ -158,65 +153,12 @@ func (provider *Marathon) loadMarathonConfig() *types.Configuration {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return true
|
return true
|
||||||
}, tasks.Tasks).([]marathon.Task)
|
|
||||||
|
|
||||||
//filter apps
|
|
||||||
filteredApps := fun.Filter(func(app marathon.Application) bool {
|
|
||||||
//get ports from app tasks
|
|
||||||
if !fun.Exists(func(task marathon.Task) bool {
|
|
||||||
if task.AppID == app.ID {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}, filteredTasks) {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
return true
|
|
||||||
}, applications.Apps).([]marathon.Application)
|
|
||||||
|
|
||||||
templateObjects := struct {
|
|
||||||
Applications []marathon.Application
|
|
||||||
Tasks []marathon.Task
|
|
||||||
Domain string
|
|
||||||
}{
|
|
||||||
filteredApps,
|
|
||||||
filteredTasks,
|
|
||||||
provider.Domain,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
tmpl := template.New(provider.Filename).Funcs(MarathonFuncMap)
|
func applicationFilter(app marathon.Application, filteredTasks []marathon.Task) bool {
|
||||||
if len(provider.Filename) > 0 {
|
return fun.Exists(func(task marathon.Task) bool {
|
||||||
_, err := tmpl.ParseFiles(provider.Filename)
|
return task.AppID == app.ID
|
||||||
if err != nil {
|
}, filteredTasks)
|
||||||
log.Error("Error reading file", err)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
buf, err := autogen.Asset("templates/marathon.tmpl")
|
|
||||||
if err != nil {
|
|
||||||
log.Error("Error reading file", err)
|
|
||||||
}
|
|
||||||
_, err = tmpl.Parse(string(buf))
|
|
||||||
if err != nil {
|
|
||||||
log.Error("Error reading file", err)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
var buffer bytes.Buffer
|
|
||||||
|
|
||||||
err = tmpl.Execute(&buffer, templateObjects)
|
|
||||||
if err != nil {
|
|
||||||
log.Error("Error with marathon template:", err)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
if _, err := toml.Decode(buffer.String(), configuration); err != nil {
|
|
||||||
log.Error("Error creating marathon configuration:", err)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
return configuration
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func getApplication(task marathon.Task, apps []marathon.Application) (marathon.Application, error) {
|
func getApplication(task marathon.Task, apps []marathon.Application) (marathon.Application, error) {
|
||||||
|
@ -237,22 +179,63 @@ func (provider *Marathon) getLabel(application marathon.Application, label strin
|
||||||
return "", errors.New("Label not found:" + label)
|
return "", errors.New("Label not found:" + label)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (provider *Marathon) getEscapedName(name string) string {
|
func (provider *Marathon) getPort(task marathon.Task) string {
|
||||||
return strings.Replace(name, "/", "", -1)
|
for _, port := range task.Ports {
|
||||||
|
return strconv.Itoa(port)
|
||||||
|
}
|
||||||
|
return ""
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetFrontendValue returns the frontend value for the specified application, using
|
func (provider *Marathon) getWeight(task marathon.Task, applications []marathon.Application) string {
|
||||||
|
application, errApp := getApplication(task, applications)
|
||||||
|
if errApp != nil {
|
||||||
|
log.Errorf("Unable to get marathon application from task %s", task.AppID)
|
||||||
|
return "0"
|
||||||
|
}
|
||||||
|
if label, err := provider.getLabel(application, "traefik.weight"); err == nil {
|
||||||
|
return label
|
||||||
|
}
|
||||||
|
return "0"
|
||||||
|
}
|
||||||
|
|
||||||
|
func (provider *Marathon) getDomain(application marathon.Application) string {
|
||||||
|
if label, err := provider.getLabel(application, "traefik.domain"); err == nil {
|
||||||
|
return label
|
||||||
|
}
|
||||||
|
return provider.Domain
|
||||||
|
}
|
||||||
|
|
||||||
|
func (provider *Marathon) getProtocol(task marathon.Task, applications []marathon.Application) string {
|
||||||
|
application, errApp := getApplication(task, applications)
|
||||||
|
if errApp != nil {
|
||||||
|
log.Errorf("Unable to get marathon application from task %s", task.AppID)
|
||||||
|
return "http"
|
||||||
|
}
|
||||||
|
if label, err := provider.getLabel(application, "traefik.protocol"); err == nil {
|
||||||
|
return label
|
||||||
|
}
|
||||||
|
return "http"
|
||||||
|
}
|
||||||
|
|
||||||
|
func (provider *Marathon) getPassHostHeader(application marathon.Application) string {
|
||||||
|
if passHostHeader, err := provider.getLabel(application, "traefik.frontend.passHostHeader"); err == nil {
|
||||||
|
return passHostHeader
|
||||||
|
}
|
||||||
|
return "false"
|
||||||
|
}
|
||||||
|
|
||||||
|
// getFrontendValue returns the frontend value for the specified application, using
|
||||||
// it's label. It returns a default one if the label is not present.
|
// it's label. It returns a default one if the label is not present.
|
||||||
func (provider *Marathon) GetFrontendValue(application marathon.Application) string {
|
func (provider *Marathon) getFrontendValue(application marathon.Application) string {
|
||||||
if label, err := provider.getLabel(application, "traefik.frontend.value"); err == nil {
|
if label, err := provider.getLabel(application, "traefik.frontend.value"); err == nil {
|
||||||
return label
|
return label
|
||||||
}
|
}
|
||||||
return provider.getEscapedName(application.ID) + "." + provider.Domain
|
return getEscapedName(application.ID) + "." + provider.Domain
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetFrontendRule returns the frontend rule for the specified application, using
|
// getFrontendRule returns the frontend rule for the specified application, using
|
||||||
// it's label. It returns a default one (Host) if the label is not present.
|
// it's label. It returns a default one (Host) if the label is not present.
|
||||||
func (provider *Marathon) GetFrontendRule(application marathon.Application) string {
|
func (provider *Marathon) getFrontendRule(application marathon.Application) string {
|
||||||
if label, err := provider.getLabel(application, "traefik.frontend.rule"); err == nil {
|
if label, err := provider.getLabel(application, "traefik.frontend.rule"); err == nil {
|
||||||
return label
|
return label
|
||||||
}
|
}
|
||||||
|
|
656
provider/marathon_test.go
Normal file
656
provider/marathon_test.go
Normal file
|
@ -0,0 +1,656 @@
|
||||||
|
package provider
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"net/url"
|
||||||
|
"reflect"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/emilevauge/traefik/types"
|
||||||
|
"github.com/gambol99/go-marathon"
|
||||||
|
)
|
||||||
|
|
||||||
|
type fakeClient struct {
|
||||||
|
applicationsError bool
|
||||||
|
applications *marathon.Applications
|
||||||
|
tasksError bool
|
||||||
|
tasks *marathon.Tasks
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *fakeClient) Applications(url.Values) (*marathon.Applications, error) {
|
||||||
|
if c.applicationsError {
|
||||||
|
return nil, errors.New("error")
|
||||||
|
}
|
||||||
|
return c.applications, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *fakeClient) AllTasks() (*marathon.Tasks, error) {
|
||||||
|
if c.tasksError {
|
||||||
|
return nil, errors.New("error")
|
||||||
|
}
|
||||||
|
return c.tasks, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestMarathonLoadConfig(t *testing.T) {
|
||||||
|
cases := []struct {
|
||||||
|
applicationsError bool
|
||||||
|
applications *marathon.Applications
|
||||||
|
tasksError bool
|
||||||
|
tasks *marathon.Tasks
|
||||||
|
expectedNil bool
|
||||||
|
expectedFrontends map[string]*types.Frontend
|
||||||
|
expectedBackends map[string]*types.Backend
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
applications: &marathon.Applications{},
|
||||||
|
tasks: &marathon.Tasks{},
|
||||||
|
expectedFrontends: map[string]*types.Frontend{},
|
||||||
|
expectedBackends: map[string]*types.Backend{},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
applicationsError: true,
|
||||||
|
applications: &marathon.Applications{},
|
||||||
|
tasks: &marathon.Tasks{},
|
||||||
|
expectedNil: true,
|
||||||
|
expectedFrontends: map[string]*types.Frontend{},
|
||||||
|
expectedBackends: map[string]*types.Backend{},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
applications: &marathon.Applications{},
|
||||||
|
tasksError: true,
|
||||||
|
tasks: &marathon.Tasks{},
|
||||||
|
expectedNil: true,
|
||||||
|
expectedFrontends: map[string]*types.Frontend{},
|
||||||
|
expectedBackends: map[string]*types.Backend{},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
applications: &marathon.Applications{
|
||||||
|
Apps: []marathon.Application{
|
||||||
|
{
|
||||||
|
ID: "/test",
|
||||||
|
Ports: []int{80},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
tasks: &marathon.Tasks{
|
||||||
|
Tasks: []marathon.Task{
|
||||||
|
{
|
||||||
|
ID: "test",
|
||||||
|
AppID: "/test",
|
||||||
|
Host: "127.0.0.1",
|
||||||
|
Ports: []int{80},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expectedFrontends: map[string]*types.Frontend{
|
||||||
|
`frontend-test`: {
|
||||||
|
Backend: "backend-test",
|
||||||
|
Routes: map[string]types.Route{
|
||||||
|
`route-host-test`: {
|
||||||
|
Rule: "Host",
|
||||||
|
Value: "test.docker.localhost",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expectedBackends: map[string]*types.Backend{
|
||||||
|
"backend-test": {
|
||||||
|
Servers: map[string]types.Server{
|
||||||
|
"server-test": {
|
||||||
|
URL: "http://127.0.0.1:80",
|
||||||
|
Weight: 0,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
CircuitBreaker: nil,
|
||||||
|
LoadBalancer: nil,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, c := range cases {
|
||||||
|
provider := &Marathon{
|
||||||
|
Domain: "docker.localhost",
|
||||||
|
marathonClient: &fakeClient{
|
||||||
|
applicationsError: c.applicationsError,
|
||||||
|
applications: c.applications,
|
||||||
|
tasksError: c.tasksError,
|
||||||
|
tasks: c.tasks,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
actualConfig := provider.loadMarathonConfig()
|
||||||
|
if c.expectedNil {
|
||||||
|
if actualConfig != nil {
|
||||||
|
t.Fatalf("Should have been nil, got %v", actualConfig)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// Compare backends
|
||||||
|
if !reflect.DeepEqual(actualConfig.Backends, c.expectedBackends) {
|
||||||
|
t.Fatalf("expected %#v, got %#v", c.expectedBackends, actualConfig.Backends)
|
||||||
|
}
|
||||||
|
if !reflect.DeepEqual(actualConfig.Frontends, c.expectedFrontends) {
|
||||||
|
t.Fatalf("expected %#v, got %#v", c.expectedFrontends, actualConfig.Frontends)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestMarathonTaskFilter(t *testing.T) {
|
||||||
|
cases := []struct {
|
||||||
|
task marathon.Task
|
||||||
|
applications *marathon.Applications
|
||||||
|
expected bool
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
task: marathon.Task{},
|
||||||
|
applications: &marathon.Applications{},
|
||||||
|
expected: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
task: marathon.Task{
|
||||||
|
AppID: "test",
|
||||||
|
Ports: []int{80},
|
||||||
|
},
|
||||||
|
applications: &marathon.Applications{},
|
||||||
|
expected: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
task: marathon.Task{
|
||||||
|
AppID: "test",
|
||||||
|
Ports: []int{80},
|
||||||
|
},
|
||||||
|
applications: &marathon.Applications{
|
||||||
|
Apps: []marathon.Application{
|
||||||
|
{
|
||||||
|
ID: "foo",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
task: marathon.Task{
|
||||||
|
AppID: "foo",
|
||||||
|
Ports: []int{80},
|
||||||
|
},
|
||||||
|
applications: &marathon.Applications{
|
||||||
|
Apps: []marathon.Application{
|
||||||
|
{
|
||||||
|
ID: "foo",
|
||||||
|
Ports: []int{80, 443},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
task: marathon.Task{
|
||||||
|
AppID: "foo",
|
||||||
|
Ports: []int{80},
|
||||||
|
},
|
||||||
|
applications: &marathon.Applications{
|
||||||
|
Apps: []marathon.Application{
|
||||||
|
{
|
||||||
|
ID: "foo",
|
||||||
|
Ports: []int{80},
|
||||||
|
Labels: map[string]string{
|
||||||
|
"traefik.enable": "false",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
task: marathon.Task{
|
||||||
|
AppID: "foo",
|
||||||
|
Ports: []int{80},
|
||||||
|
},
|
||||||
|
applications: &marathon.Applications{
|
||||||
|
Apps: []marathon.Application{
|
||||||
|
{
|
||||||
|
ID: "foo",
|
||||||
|
Ports: []int{80},
|
||||||
|
HealthChecks: []*marathon.HealthCheck{
|
||||||
|
marathon.NewDefaultHealthCheck(),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
task: marathon.Task{
|
||||||
|
AppID: "foo",
|
||||||
|
Ports: []int{80},
|
||||||
|
HealthCheckResult: []*marathon.HealthCheckResult{
|
||||||
|
{
|
||||||
|
Alive: false,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
applications: &marathon.Applications{
|
||||||
|
Apps: []marathon.Application{
|
||||||
|
{
|
||||||
|
ID: "foo",
|
||||||
|
Ports: []int{80},
|
||||||
|
HealthChecks: []*marathon.HealthCheck{
|
||||||
|
marathon.NewDefaultHealthCheck(),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
task: marathon.Task{
|
||||||
|
AppID: "foo",
|
||||||
|
Ports: []int{80},
|
||||||
|
HealthCheckResult: []*marathon.HealthCheckResult{
|
||||||
|
{
|
||||||
|
Alive: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Alive: false,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
applications: &marathon.Applications{
|
||||||
|
Apps: []marathon.Application{
|
||||||
|
{
|
||||||
|
ID: "foo",
|
||||||
|
Ports: []int{80},
|
||||||
|
HealthChecks: []*marathon.HealthCheck{
|
||||||
|
marathon.NewDefaultHealthCheck(),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
task: marathon.Task{
|
||||||
|
AppID: "foo",
|
||||||
|
Ports: []int{80},
|
||||||
|
},
|
||||||
|
applications: &marathon.Applications{
|
||||||
|
Apps: []marathon.Application{
|
||||||
|
{
|
||||||
|
ID: "foo",
|
||||||
|
Ports: []int{80},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
task: marathon.Task{
|
||||||
|
AppID: "foo",
|
||||||
|
Ports: []int{80},
|
||||||
|
HealthCheckResult: []*marathon.HealthCheckResult{
|
||||||
|
{
|
||||||
|
Alive: true,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
applications: &marathon.Applications{
|
||||||
|
Apps: []marathon.Application{
|
||||||
|
{
|
||||||
|
ID: "foo",
|
||||||
|
Ports: []int{80},
|
||||||
|
HealthChecks: []*marathon.HealthCheck{
|
||||||
|
marathon.NewDefaultHealthCheck(),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: true,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, c := range cases {
|
||||||
|
actual := taskFilter(c.task, c.applications)
|
||||||
|
if actual != c.expected {
|
||||||
|
t.Fatalf("expected %v, got %v", c.expected, actual)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestMarathonApplicationFilter(t *testing.T) {
|
||||||
|
cases := []struct {
|
||||||
|
application marathon.Application
|
||||||
|
filteredTasks []marathon.Task
|
||||||
|
expected bool
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
application: marathon.Application{},
|
||||||
|
filteredTasks: []marathon.Task{},
|
||||||
|
expected: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
application: marathon.Application{
|
||||||
|
ID: "test",
|
||||||
|
},
|
||||||
|
filteredTasks: []marathon.Task{},
|
||||||
|
expected: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
application: marathon.Application{
|
||||||
|
ID: "foo",
|
||||||
|
},
|
||||||
|
filteredTasks: []marathon.Task{
|
||||||
|
{
|
||||||
|
AppID: "bar",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
application: marathon.Application{
|
||||||
|
ID: "foo",
|
||||||
|
},
|
||||||
|
filteredTasks: []marathon.Task{
|
||||||
|
{
|
||||||
|
AppID: "foo",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: true,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, c := range cases {
|
||||||
|
actual := applicationFilter(c.application, c.filteredTasks)
|
||||||
|
if actual != c.expected {
|
||||||
|
t.Fatalf("expected %v, got %v", c.expected, actual)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestMarathonGetPort(t *testing.T) {
|
||||||
|
provider := &Marathon{}
|
||||||
|
|
||||||
|
cases := []struct {
|
||||||
|
task marathon.Task
|
||||||
|
expected string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
task: marathon.Task{},
|
||||||
|
expected: "",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
task: marathon.Task{
|
||||||
|
Ports: []int{80},
|
||||||
|
},
|
||||||
|
expected: "80",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
task: marathon.Task{
|
||||||
|
Ports: []int{80, 443},
|
||||||
|
},
|
||||||
|
expected: "80",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, c := range cases {
|
||||||
|
actual := provider.getPort(c.task)
|
||||||
|
if actual != c.expected {
|
||||||
|
t.Fatalf("expected %q, got %q", c.expected, actual)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestMarathonGetWeigh(t *testing.T) {
|
||||||
|
provider := &Marathon{}
|
||||||
|
|
||||||
|
applications := []struct {
|
||||||
|
applications []marathon.Application
|
||||||
|
task marathon.Task
|
||||||
|
expected string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
applications: []marathon.Application{},
|
||||||
|
task: marathon.Task{},
|
||||||
|
expected: "0",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
applications: []marathon.Application{
|
||||||
|
{
|
||||||
|
ID: "test1",
|
||||||
|
Labels: map[string]string{
|
||||||
|
"traefik.weight": "10",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
task: marathon.Task{
|
||||||
|
AppID: "test2",
|
||||||
|
},
|
||||||
|
expected: "0",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
applications: []marathon.Application{
|
||||||
|
{
|
||||||
|
ID: "test",
|
||||||
|
Labels: map[string]string{
|
||||||
|
"traefik.test": "10",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
task: marathon.Task{
|
||||||
|
AppID: "test",
|
||||||
|
},
|
||||||
|
expected: "0",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
applications: []marathon.Application{
|
||||||
|
{
|
||||||
|
ID: "test",
|
||||||
|
Labels: map[string]string{
|
||||||
|
"traefik.weight": "10",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
task: marathon.Task{
|
||||||
|
AppID: "test",
|
||||||
|
},
|
||||||
|
expected: "10",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, a := range applications {
|
||||||
|
actual := provider.getWeight(a.task, a.applications)
|
||||||
|
if actual != a.expected {
|
||||||
|
t.Fatalf("expected %q, got %q", a.expected, actual)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestMarathonGetDomain(t *testing.T) {
|
||||||
|
provider := &Marathon{
|
||||||
|
Domain: "docker.localhost",
|
||||||
|
}
|
||||||
|
|
||||||
|
applications := []struct {
|
||||||
|
application marathon.Application
|
||||||
|
expected string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
application: marathon.Application{},
|
||||||
|
expected: "docker.localhost",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
application: marathon.Application{
|
||||||
|
Labels: map[string]string{
|
||||||
|
"traefik.domain": "foo.bar",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: "foo.bar",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, a := range applications {
|
||||||
|
actual := provider.getDomain(a.application)
|
||||||
|
if actual != a.expected {
|
||||||
|
t.Fatalf("expected %q, got %q", a.expected, actual)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestMarathonGetProtocol(t *testing.T) {
|
||||||
|
provider := &Marathon{}
|
||||||
|
|
||||||
|
applications := []struct {
|
||||||
|
applications []marathon.Application
|
||||||
|
task marathon.Task
|
||||||
|
expected string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
applications: []marathon.Application{},
|
||||||
|
task: marathon.Task{},
|
||||||
|
expected: "http",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
applications: []marathon.Application{
|
||||||
|
{
|
||||||
|
ID: "test1",
|
||||||
|
Labels: map[string]string{
|
||||||
|
"traefik.protocol": "https",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
task: marathon.Task{
|
||||||
|
AppID: "test2",
|
||||||
|
},
|
||||||
|
expected: "http",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
applications: []marathon.Application{
|
||||||
|
{
|
||||||
|
ID: "test",
|
||||||
|
Labels: map[string]string{
|
||||||
|
"traefik.foo": "bar",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
task: marathon.Task{
|
||||||
|
AppID: "test",
|
||||||
|
},
|
||||||
|
expected: "http",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
applications: []marathon.Application{
|
||||||
|
{
|
||||||
|
ID: "test",
|
||||||
|
Labels: map[string]string{
|
||||||
|
"traefik.protocol": "https",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
task: marathon.Task{
|
||||||
|
AppID: "test",
|
||||||
|
},
|
||||||
|
expected: "https",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, a := range applications {
|
||||||
|
actual := provider.getProtocol(a.task, a.applications)
|
||||||
|
if actual != a.expected {
|
||||||
|
t.Fatalf("expected %q, got %q", a.expected, actual)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestMarathonGetPassHostHeader(t *testing.T) {
|
||||||
|
provider := &Marathon{}
|
||||||
|
|
||||||
|
applications := []struct {
|
||||||
|
application marathon.Application
|
||||||
|
expected string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
application: marathon.Application{},
|
||||||
|
expected: "false",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
application: marathon.Application{
|
||||||
|
Labels: map[string]string{
|
||||||
|
"traefik.frontend.passHostHeader": "true",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: "true",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, a := range applications {
|
||||||
|
actual := provider.getPassHostHeader(a.application)
|
||||||
|
if actual != a.expected {
|
||||||
|
t.Fatalf("expected %q, got %q", a.expected, actual)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestMarathonGetFrontendValue(t *testing.T) {
|
||||||
|
provider := &Marathon{
|
||||||
|
Domain: "docker.localhost",
|
||||||
|
}
|
||||||
|
|
||||||
|
applications := []struct {
|
||||||
|
application marathon.Application
|
||||||
|
expected string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
application: marathon.Application{},
|
||||||
|
expected: ".docker.localhost",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
application: marathon.Application{
|
||||||
|
ID: "test",
|
||||||
|
},
|
||||||
|
expected: "test.docker.localhost",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
application: marathon.Application{
|
||||||
|
Labels: map[string]string{
|
||||||
|
"traefik.frontend.value": "foo.bar",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: "foo.bar",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, a := range applications {
|
||||||
|
actual := provider.getFrontendValue(a.application)
|
||||||
|
if actual != a.expected {
|
||||||
|
t.Fatalf("expected %q, got %q", a.expected, actual)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestMarathonGetFrontendRule(t *testing.T) {
|
||||||
|
provider := &Marathon{}
|
||||||
|
|
||||||
|
applications := []struct {
|
||||||
|
application marathon.Application
|
||||||
|
expected string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
application: marathon.Application{},
|
||||||
|
expected: "Host",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
application: marathon.Application{
|
||||||
|
Labels: map[string]string{
|
||||||
|
"traefik.frontend.rule": "Header",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expected: "Header",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, a := range applications {
|
||||||
|
actual := provider.getFrontendRule(a.application)
|
||||||
|
if actual != a.expected {
|
||||||
|
t.Fatalf("expected %q, got %q", a.expected, actual)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
|
@ -1,6 +1,15 @@
|
||||||
package provider
|
package provider
|
||||||
|
|
||||||
import "github.com/emilevauge/traefik/types"
|
import (
|
||||||
|
"bytes"
|
||||||
|
"io/ioutil"
|
||||||
|
"strings"
|
||||||
|
"text/template"
|
||||||
|
|
||||||
|
"github.com/BurntSushi/toml"
|
||||||
|
"github.com/emilevauge/traefik/autogen"
|
||||||
|
"github.com/emilevauge/traefik/types"
|
||||||
|
)
|
||||||
|
|
||||||
// Provider defines methods of a provider.
|
// Provider defines methods of a provider.
|
||||||
type Provider interface {
|
type Provider interface {
|
||||||
|
@ -8,3 +17,51 @@ type Provider interface {
|
||||||
// using the given configuration channel.
|
// using the given configuration channel.
|
||||||
Provide(configurationChan chan<- types.ConfigMessage) error
|
Provide(configurationChan chan<- types.ConfigMessage) error
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type baseProvider struct {
|
||||||
|
Watch bool
|
||||||
|
Filename string
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *baseProvider) getConfiguration(defaultTemplateFile string, funcMap template.FuncMap, templateObjects interface{}) (*types.Configuration, error) {
|
||||||
|
var (
|
||||||
|
buf []byte
|
||||||
|
err error
|
||||||
|
)
|
||||||
|
configuration := new(types.Configuration)
|
||||||
|
tmpl := template.New(p.Filename).Funcs(funcMap)
|
||||||
|
if len(p.Filename) > 0 {
|
||||||
|
buf, err = ioutil.ReadFile(p.Filename)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
buf, err = autogen.Asset(defaultTemplateFile)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_, err = tmpl.Parse(string(buf))
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
var buffer bytes.Buffer
|
||||||
|
err = tmpl.Execute(&buffer, templateObjects)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
if _, err := toml.Decode(buffer.String(), configuration); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return configuration, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func replace(s1 string, s2 string, s3 string) string {
|
||||||
|
return strings.Replace(s3, s1, s2, -1)
|
||||||
|
}
|
||||||
|
|
||||||
|
func getEscapedName(name string) string {
|
||||||
|
return strings.Replace(name, "/", "", -1)
|
||||||
|
}
|
||||||
|
|
170
provider/provider_test.go
Normal file
170
provider/provider_test.go
Normal file
|
@ -0,0 +1,170 @@
|
||||||
|
package provider
|
||||||
|
|
||||||
|
import (
|
||||||
|
"io/ioutil"
|
||||||
|
"os"
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
"text/template"
|
||||||
|
)
|
||||||
|
|
||||||
|
type myProvider struct {
|
||||||
|
baseProvider
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *myProvider) Foo() string {
|
||||||
|
return "bar"
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestConfigurationErrors(t *testing.T) {
|
||||||
|
templateErrorFile, err := ioutil.TempFile("", "provider-configuration-error")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
defer os.RemoveAll(templateErrorFile.Name())
|
||||||
|
data := []byte("Not a valid template {{ Bar }}")
|
||||||
|
err = ioutil.WriteFile(templateErrorFile.Name(), data, 0700)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
templateInvalidTOMLFile, err := ioutil.TempFile("", "provider-configuration-error")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
defer os.RemoveAll(templateInvalidTOMLFile.Name())
|
||||||
|
data = []byte(`Hello {{ .Name }}
|
||||||
|
{{ Foo }}`)
|
||||||
|
err = ioutil.WriteFile(templateInvalidTOMLFile.Name(), data, 0700)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
invalids := []struct {
|
||||||
|
provider *myProvider
|
||||||
|
defaultTemplate string
|
||||||
|
expectedError string
|
||||||
|
funcMap template.FuncMap
|
||||||
|
templateObjects interface{}
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
provider: &myProvider{
|
||||||
|
baseProvider{
|
||||||
|
Filename: "/non/existent/template.tmpl",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expectedError: "open /non/existent/template.tmpl: no such file or directory",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
provider: &myProvider{},
|
||||||
|
defaultTemplate: "non/existent/template.tmpl",
|
||||||
|
expectedError: "Asset non/existent/template.tmpl not found",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
provider: &myProvider{
|
||||||
|
baseProvider{
|
||||||
|
Filename: templateErrorFile.Name(),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expectedError: `function "Bar" not defined`,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
provider: &myProvider{
|
||||||
|
baseProvider{
|
||||||
|
Filename: templateInvalidTOMLFile.Name(),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expectedError: "Near line 1, key 'Hello': Near line 1: Expected key separator '=', but got '<' instead",
|
||||||
|
funcMap: template.FuncMap{
|
||||||
|
"Foo": func() string {
|
||||||
|
return "bar"
|
||||||
|
},
|
||||||
|
},
|
||||||
|
templateObjects: struct{ Name string }{Name: "bar"},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, invalid := range invalids {
|
||||||
|
configuration, err := invalid.provider.getConfiguration(invalid.defaultTemplate, invalid.funcMap, nil)
|
||||||
|
if err == nil || !strings.Contains(err.Error(), invalid.expectedError) {
|
||||||
|
t.Fatalf("should have generate an error with %q, got %v", invalid.expectedError, err)
|
||||||
|
}
|
||||||
|
if configuration != nil {
|
||||||
|
t.Fatalf("shouldn't have return a configuration object : %v", configuration)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestGetConfiguration(t *testing.T) {
|
||||||
|
templateFile, err := ioutil.TempFile("", "provider-configuration")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
defer os.RemoveAll(templateFile.Name())
|
||||||
|
data := []byte(`[backends]
|
||||||
|
[backends.backend1]
|
||||||
|
[backends.backend1.circuitbreaker]
|
||||||
|
expression = "NetworkErrorRatio() > 0.5"
|
||||||
|
[backends.backend1.servers.server1]
|
||||||
|
url = "http://172.17.0.2:80"
|
||||||
|
weight = 10
|
||||||
|
[backends.backend1.servers.server2]
|
||||||
|
url = "http://172.17.0.3:80"
|
||||||
|
weight = 1
|
||||||
|
|
||||||
|
[frontends]
|
||||||
|
[frontends.frontend1]
|
||||||
|
backend = "backend1"
|
||||||
|
passHostHeader = true
|
||||||
|
[frontends.frontend11.routes.test_2]
|
||||||
|
rule = "Path"
|
||||||
|
value = "/test"`)
|
||||||
|
err = ioutil.WriteFile(templateFile.Name(), data, 0700)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
provider := &myProvider{
|
||||||
|
baseProvider{
|
||||||
|
Filename: templateFile.Name(),
|
||||||
|
},
|
||||||
|
}
|
||||||
|
configuration, err := provider.getConfiguration(templateFile.Name(), nil, nil)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Shouldn't have error out, got %v", err)
|
||||||
|
}
|
||||||
|
if configuration == nil {
|
||||||
|
t.Fatalf("Configuration should not be nil, but was")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestReplace(t *testing.T) {
|
||||||
|
cases := []struct {
|
||||||
|
str string
|
||||||
|
expected string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
str: "",
|
||||||
|
expected: "",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
str: "foo",
|
||||||
|
expected: "bar",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
str: "foo foo",
|
||||||
|
expected: "bar bar",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
str: "somethingfoo",
|
||||||
|
expected: "somethingbar",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, c := range cases {
|
||||||
|
actual := replace("foo", "bar", c.str)
|
||||||
|
if actual != c.expected {
|
||||||
|
t.Fatalf("expected %q, got %q, for %q", c.expected, actual, c.str)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
|
@ -1,19 +1,20 @@
|
||||||
package provider
|
package provider
|
||||||
|
|
||||||
import "github.com/emilevauge/traefik/types"
|
import (
|
||||||
|
"github.com/docker/libkv/store"
|
||||||
|
"github.com/docker/libkv/store/zookeeper"
|
||||||
|
"github.com/emilevauge/traefik/types"
|
||||||
|
)
|
||||||
|
|
||||||
// Zookepper holds configurations of the Zookepper provider.
|
// Zookepper holds configurations of the Zookepper provider.
|
||||||
type Zookepper struct {
|
type Zookepper struct {
|
||||||
Watch bool
|
Kv
|
||||||
Endpoint string
|
|
||||||
Prefix string
|
|
||||||
Filename string
|
|
||||||
KvProvider *Kv
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Provide allows the provider to provide configurations to traefik
|
// Provide allows the provider to provide configurations to traefik
|
||||||
// using the given configuration channel.
|
// using the given configuration channel.
|
||||||
func (provider *Zookepper) Provide(configurationChan chan<- types.ConfigMessage) error {
|
func (provider *Zookepper) Provide(configurationChan chan<- types.ConfigMessage) error {
|
||||||
provider.KvProvider = NewZkProvider(provider)
|
provider.StoreType = store.ZK
|
||||||
return provider.KvProvider.provide(configurationChan)
|
zookeeper.Register()
|
||||||
|
return provider.provide(configurationChan)
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue