Force stop after second signal

wish · Feb 4, 2019 · 3920be4 · 3920be4
1 parent d50de15
commit 3920be4
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/cmd/katalog-sync-sidecar/main.go b/cmd/katalog-sync-sidecar/main.go
@@ -104,8 +104,18 @@ WAITLOOP:
 		}
 	}
 
+	go func() {
+		<-sigs
+		cancel()
+	}()
+
 	// Send deregister request
 	for {
+		select {
+		case <-ctx.Done():
+			return
+		default:
+		}
 		logrus.Infof("deregister attempt")
 		_, err := client.Deregister(ctx, &katalogsync.DeregisterQuery{Namespace: opts.Namespace, PodName: opts.PodName, ContainerName: opts.ContainerName})
 		if err == nil {