WithShutdownNotifier: chan<- interface{}

vbauerster · Feb 11, 2023 · 4c2e771 · 4c2e771
1 parent 93889df
commit 4c2e771
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/container_option.go b/container_option.go
@@ -56,15 +56,11 @@ func WithRenderDelay(ch <-chan struct{}) ContainerOption {
 	}
 }
 
-// WithShutdownNotifier provided chanel will be closed, after all bars
-// have been rendered.
-func WithShutdownNotifier(ch chan struct{}) ContainerOption {
+// WithShutdownNotifier value of type `[]*mpb.Bar` will be send into provided
+// channel upon container shutdown.
+func WithShutdownNotifier(ch chan<- interface{}) ContainerOption {
 	return func(s *pState) {
-		select {
-		case <-ch:
-		default:
-			s.shutdownNotifier = ch
-		}
+		s.shutdownNotifier = ch
 	}
 }