traefik/docker.go

package main
import(
	"github.com/fsouza/go-dockerclient"
	"github.com/leekchan/gtf"
	"bytes"
	"github.com/BurntSushi/toml"
	"log"
)

type DockerProvider struct {
	Watch bool
	Endpoint string
	dockerClient *docker.Client
}

func (provider *DockerProvider) Provide(serviceChan chan<- *Service){
	provider.dockerClient, _ = docker.NewClient(provider.Endpoint)
	dockerEvents := make(chan *docker.APIEvents)
	if(provider.Watch) {
		provider.dockerClient.AddEventListener(dockerEvents)
	}
	go func() {
		for {
			event := <-dockerEvents
			log.Println("Event receveived", event)
			service:= provider.loadDockerConfig()
			serviceChan <- service
		}
	}()

	service:= provider.loadDockerConfig()
	serviceChan <- service
}

func (provider *DockerProvider) loadDockerConfig() *Service {
	service := new(Service)
	containerList, _ := provider.dockerClient.ListContainers(docker.ListContainersOptions{})
	containersInspected := []docker.Container{}
	for _, container := range containerList {
		containerInspected, _ := provider.dockerClient.InspectContainer(container.ID)
		containersInspected = append(containersInspected, *containerInspected)
	}
	containers := struct {
		Containers []docker.Container
	}{
		containersInspected,
	}
	tmpl, err := gtf.New("docker.tmpl").ParseFiles("docker.tmpl")
	if err != nil {
		log.Println("Error reading file:", err)
		return nil
	}

	var buffer bytes.Buffer

	err = tmpl.Execute(&buffer, containers)
	if err != nil {
		log.Println("Error with docker template:", err)
		return nil
	}

	if _, err := toml.Decode(buffer.String(), service); err != nil {
		log.Println("Error creating docker service:", err)
		return nil
	}
	return service
}