handle kill event

Author: Dustin Blackman

context.go

@@ -78,6 +78,7 @@ type Volume struct {
 }
 
 type State struct {
+	Killing bool
 	Running bool
 }
 

generator.go

@@ -191,12 +191,21 @@ func (g *generator) generateFromEvents() {
 			watchers = append(watchers, watcher)
 
 			debouncedChan := newDebounceChannel(watcher, config.Wait)
-			for _ = range debouncedChan {
+			for event := range debouncedChan {
 				containers, err := g.getContainers()
 				if err != nil {
 					log.Printf("Error listing containers: %s\n", err)
 					continue
 				}
+
+				if event.Status == "kill" {
+					for i := len(containers) - 1; i >= 0; i-- {
+						if containers[i].ID == event.ID {
+							containers[i].State.Killing = true
+						}
+					}
+				}
+
 				changed := GenerateFile(config, containers)
 				if !changed {
 					log.Printf("Contents of %s did not change. Skipping notification '%s'", config.Dest, config.NotifyCmd)
@@ -270,7 +279,7 @@ func (g *generator) generateFromEvents() {
 						time.Sleep(10 * time.Second)
 						break
 					}
-					if event.Status == "start" || event.Status == "stop" || event.Status == "die" {
+					if event.Status == "start" || event.Status == "stop" || event.Status == "die" || event.Status == "kill" {
 						log.Printf("Received event %s for container %s", event.Status, event.ID[:12])
 						// fanout event to all watchers
 						for _, watcher := range watchers {
@@ -375,6 +384,7 @@ func (g *generator) getContainers() ([]*RuntimeContainer, error) {
 			},
 			State: State{
 				Running: container.State.Running,
+				Killing: false,
 			},
 			Name:         strings.TrimLeft(container.Name, "/"),
 			Hostname:     container.Config.Hostname,

template.go

@@ -459,7 +459,7 @@ func filterRunning(config Config, containers Context) Context {
 	} else {
 		filteredContainers := Context{}
 		for _, container := range containers {
-			if container.State.Running {
+			if container.State.Running && !container.State.Killing {
 				filteredContainers = append(filteredContainers, container)
 			}
 		}