cut blockSetState short if it hangs (#407)

livekit · Jun 30, 2023 · 37299d7 · 37299d7
1 parent 249a7d2
commit 37299d7
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/pkg/pipeline/pipeline.go b/pkg/pipeline/pipeline.go
@@ -520,7 +520,20 @@ func (p *Pipeline) stop() {
 		return
 	}
 
-	_ = p.pipeline.BlockSetState(gst.StateNull)
+	stateChange := make(chan error, 1)
+	go func() {
+		stateChange <- p.pipeline.BlockSetState(gst.StateNull)
+	}()
+
+	select {
+	case err := <-stateChange:
+		if err != nil {
+			logger.Errorw("SetStateNull failed", err)
+		}
+	case <-time.After(eosTimeout):
+		logger.Errorw("SetStateNull timed out", nil)
+	}
+
 	endedAt := time.Now().UnixNano()
 	logger.Infow("pipeline stopped")