mirrors/docker-etchosts
1
0
Fork 0
mirror of https://github.com/costela/docker-etchosts.git synced 2025-05-11 18:05:42 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

add acutal etchosts syncing

Browse source
This commit is contained in:
Leo Antunes 2018-08-22 00:09:55 +02:00
parent 323ad6a54e
commit 03a5f533ca
3 changed files with 115 additions and 50 deletions
Download patch file Download diff file Expand all files Collapse all files

51
docker.go
View file

@ -57,9 +57,10 @@ func getIPsToNames(client dockerClienter, id string) (ipsToNamesMap, error) {
}
for netName, netInfo := range containerFull.NetworkSettings.Networks {
names := make([]string, 4) // 4 is worst-case size if container in a compose project (see below)
names := make([]string, 0, 4) // 4 is worst-case size if container in a compose project (see below)
appendNames := func(names []string, name string) []string {
log.Debugf("found base name %s with IP %s", name, netInfo.IPAddress)
names = append(names, fmt.Sprintf("%s", name))
names = append(names, fmt.Sprintf("%s.%s", name, netName))
if proj, ok := containerFull.Config.Labels["com.docker.compose.project"]; ok {
@ -80,40 +81,30 @@ func getIPsToNames(client dockerClienter, id string) (ipsToNamesMap, error) {
return ipsToNames, nil
}
func listenForEvents(client dockerClienter) {
func syncAndListenForEvents(client dockerClienter) {
eventOpts := types.EventsOptions{
Filters: filters.NewArgs(
filters.Arg("type", "container"),
filters.Arg("event", "start"),
filters.Arg("event", "die"),
filters.Arg("event", "destroy"),
),
}
// helper channel to ensure we run once without
kickoff := make(chan bool, 1)
kickoff <- true
events, errors := client.Events(context.Background(), eventOpts)
loop:
for {
select {
case <-kickoff:
log.Infof("running initial sync")
getAndWrite(client)
case event := <-events:
switch event.Action {
case "start":
ipsToNames, err := getIPsToNames(client, event.Actor.Attributes["name"])
if err != nil {
log.WithFields(
log.Fields{"container": event.Actor.ID},
).Errorf("could not get info for container: %s", err)
continue
}
err = addToEtcHosts(ipsToNames)
if err != nil {
log.WithFields(
log.Fields{"container": event.Actor.ID},
).Errorf("could not add container to hosts file: %s", err)
}
case "die":
// We remove by name because we cannot get the container's IP after it has been stopped.
// Names are supposed to be kept unique by the docker deamon.
removeFromEtcHosts(event.Actor.Attributes["name"])
}
log.Infof("got %s event for %s", event.Action, event.Actor.Attributes["name"])
getAndWrite(client)
case err := <-errors:
log.Errorf("error fetching event: %s", err)
break loop
@ -121,6 +112,20 @@ loop:
}
}
func getAndWrite(client dockerClienter) {
log.Info("fetching container infos")
currentContent, err := getAllIPsToNames(client)
if err != nil {
log.Errorf("error getting container infos: %s", err)
}
log.Info("writing current state")
err = writeToEtcHosts(currentContent)
if err != nil {
log.Errorf("error syncing hosts: %s", err)
}
}
func waitForConnection(client dockerClienter) {
err := backoff.Retry(func() error {
log.Info("retrying connection to docker")