traefik/docker.go

67 lines
1.6 KiB
Go
Raw Normal View History

2015-09-07 08:38:58 +00:00
package main
import(
"github.com/fsouza/go-dockerclient"
"github.com/leekchan/gtf"
"bytes"
"github.com/BurntSushi/toml"
"log"
2015-09-07 08:38:58 +00:00
)
type DockerProvider struct {
2015-09-07 16:10:33 +00:00
Watch bool
2015-09-07 15:39:22 +00:00
Endpoint string
2015-09-07 08:38:58 +00:00
dockerClient *docker.Client
}
func (provider *DockerProvider) Provide(serviceChan chan<- *Service){
2015-09-07 15:39:22 +00:00
provider.dockerClient, _ = docker.NewClient(provider.Endpoint)
2015-09-07 08:38:58 +00:00
dockerEvents := make(chan *docker.APIEvents)
2015-09-07 16:10:33 +00:00
if(provider.Watch) {
provider.dockerClient.AddEventListener(dockerEvents)
}
2015-09-07 08:38:58 +00:00
go func() {
for {
event := <-dockerEvents
log.Println("Event receveived", event)
2015-09-07 08:38:58 +00:00
service:= provider.loadDockerConfig()
serviceChan <- service
2015-09-07 08:38:58 +00:00
}
}()
service:= provider.loadDockerConfig()
serviceChan <- service
2015-09-07 08:38:58 +00:00
}
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
2015-09-07 08:38:58 +00:00
}
var buffer bytes.Buffer
err = tmpl.Execute(&buffer, containers)
if err != nil {
log.Println("Error with docker template:", err)
return nil
2015-09-07 08:38:58 +00:00
}
if _, err := toml.Decode(buffer.String(), service); err != nil {
log.Println("Error creating docker service:", err)
2015-09-07 08:38:58 +00:00
return nil
}
return service
}