Merge pull request #27 from JustAdam/container-restart

Added flag -restart-container to allow restarting of docker containers.

Author: jwilder

README.md

@@ -40,7 +40,7 @@ $ docker-gen
 Usage: docker-gen [options] <template> [<dest>]
 ```
 
-[-config file] [-watch=false] [-notify="restart xyz"] [-interval=0] [-endpoint tcp|unix://..]
+[-config file] [-watch=false] [-notify="restart xyz"] [-notify-sighup="nginx-proxy"] [-interval=0] [-endpoint tcp|unix://..]
 
 *Options:*
 ```
@@ -50,6 +50,7 @@ Usage: docker-gen [options] <template> [<dest>]
   -interval=0:run notify command interval (s). Useful for service registration use cases.
   -notify="": run command after template is regenerated ["restart xyz"]. Useful for restarting nginx,
               reloading haproxy, etc..
+  -notify-sighup="": send HUP signal to container.  Equivalent to `docker kill -s HUP container-ID`
   -only-exposed=false: only include containers with exposed ports
   -only-published=false: only include containers with published ports (implies -only-exposed)
   -version=false: show version

docker-gen.go

@@ -15,17 +15,18 @@ import (
 )
 
 var (
-	buildVersion  string
-	version       bool
-	watch         bool
-	notifyCmd     string
-	onlyExposed   bool
-	onlyPublished bool
-	configFile    string
-	configs       ConfigFile
-	interval      int
-	endpoint      string
-	wg            sync.WaitGroup
+	buildVersion            string
+	version                 bool
+	watch                   bool
+	notifyCmd               string
+	notifySigHUPContainerID string
+	onlyExposed             bool
+	onlyPublished           bool
+	configFile              string
+	configs                 ConfigFile
+	interval                int
+	endpoint                string
+	wg                      sync.WaitGroup
 )
 
 type Event struct {
@@ -66,13 +67,14 @@ func (i *DockerImage) String() string {
 }
 
 type Config struct {
-	Template      string
-	Dest          string
-	Watch         bool
-	NotifyCmd     string
-	OnlyExposed   bool
-	OnlyPublished bool
-	Interval      int
+	Template         string
+	Dest             string
+	Watch            bool
+	NotifyCmd        string
+	NotifyContainers map[string]docker.Signal
+	OnlyExposed      bool
+	OnlyPublished    bool
+	Interval         int
 }
 
 type ConfigFile struct {
@@ -120,7 +122,7 @@ func (r *RuntimeContainer) PublishedAddresses() []Address {
 }
 
 func usage() {
-	println("Usage: docker-gen [-config file] [-watch=false] [-notify=\"restart xyz\"] [-interval=0] [-endpoint tcp|unix://..] <template> [<dest>]")
+	println("Usage: docker-gen [-config file] [-watch=false] [-notify=\"restart xyz\"] [-notify-sighup=\"container-ID\"] [-interval=0] [-endpoint tcp|unix://..] <template> [<dest>]")
 }
 
 func generateFromContainers(client *docker.Client) {
@@ -136,6 +138,7 @@ func generateFromContainers(client *docker.Client) {
 			continue
 		}
 		runNotifyCmd(config)
+		sendSignalToContainer(client, config)
 	}
 }
 
@@ -153,6 +156,23 @@ func runNotifyCmd(config Config) {
 	}
 }
 
+func sendSignalToContainer(client *docker.Client, config Config) {
+	if len(config.NotifyContainers) < 1 {
+		return
+	}
+
+	for container, signal := range config.NotifyContainers {
+		log.Printf("Sending container '%s' signal '%v'", container, signal)
+		killOpts := docker.KillContainerOptions{
+			ID:     container,
+			Signal: signal,
+		}
+		if err := client.KillContainer(killOpts); err != nil {
+			log.Printf("Error sending signal to container: %s", err)
+		}
+	}
+}
+
 func loadConfig(file string) error {
 	_, err := toml.DecodeFile(file, &configs)
 	if err != nil {
@@ -186,6 +206,7 @@ func generateAtInterval(client *docker.Client, configs ConfigFile) {
 					// ignore changed return value. always run notify command
 					generateFile(configCopy, containers)
 					runNotifyCmd(configCopy)
+					sendSignalToContainer(client, configCopy)
 				case <-quit:
 					ticker.Stop()
 					return
@@ -225,6 +246,7 @@ func initFlags() {
 	flag.BoolVar(&onlyExposed, "only-exposed", false, "only include containers with exposed ports")
 	flag.BoolVar(&onlyPublished, "only-published", false, "only include containers with published ports (implies -only-exposed)")
 	flag.StringVar(&notifyCmd, "notify", "", "run command after template is regenerated")
+	flag.StringVar(&notifySigHUPContainerID, "notify-sighup", "", "send HUP signal to container.  Equivalent to `docker kill -s HUP container-ID`")
 	flag.StringVar(&configFile, "config", "", "config file with template directives")
 	flag.IntVar(&interval, "interval", 0, "notify command interval (s)")
 	flag.StringVar(&endpoint, "endpoint", "", "docker api endpoint")
@@ -252,13 +274,17 @@ func main() {
 		}
 	} else {
 		config := Config{
-			Template:      flag.Arg(0),
-			Dest:          flag.Arg(1),
-			Watch:         watch,
-			NotifyCmd:     notifyCmd,
-			OnlyExposed:   onlyExposed,
-			OnlyPublished: onlyPublished,
-			Interval:      interval,
+			Template:         flag.Arg(0),
+			Dest:             flag.Arg(1),
+			Watch:            watch,
+			NotifyCmd:        notifyCmd,
+			NotifyContainers: make(map[string]docker.Signal),
+			OnlyExposed:      onlyExposed,
+			OnlyPublished:    onlyPublished,
+			Interval:         interval,
+		}
+		if notifySigHUPContainerID != "" {
+			config.NotifyContainers[notifySigHUPContainerID] = docker.SIGHUP
 		}
 		configs = ConfigFile{
 			Config: []Config{config}}