Merge pull request #16284 from vrothberg/fix-16076

notifyproxy: fix container watcher
containers · Oct 25, 2022 · 045a190 · 045a190
2 parents 86f7b99 + c4ebe9e
commit 045a190
Showing 1 changed file with 13 additions and 12 deletions.
diff --git a/pkg/systemd/notifyproxy/notifyproxy.go b/pkg/systemd/notifyproxy/notifyproxy.go
@@ -183,20 +183,21 @@ func (p *NotifyProxy) WaitAndClose() error {
 		ctx, cancel := context.WithCancel(context.Background())
 		defer cancel()
 		go func() {
-			select {
-			case <-ctx.Done():
-				return
-			default:
-				state, err := p.container.State()
-				if err != nil {
-					p.errorChan <- err
-					return
-				}
-				if state != define.ContainerStateRunning {
-					p.errorChan <- fmt.Errorf("%w: %s", ErrNoReadyMessage, p.container.ID())
+			for {
+				select {
+				case <-ctx.Done():
 					return
+				case <-time.After(time.Second):
+					state, err := p.container.State()
+					if err != nil {
+						p.errorChan <- err
+						return
+					}
+					if state != define.ContainerStateRunning {
+						p.errorChan <- fmt.Errorf("%w: %s", ErrNoReadyMessage, p.container.ID())
+						return
+					}
 				}
-				time.Sleep(time.Second)
 			}
 		}()
 	}