From e65f970097ada7d1c826631da76108bd80ddb5a3 Mon Sep 17 00:00:00 2001
From: langhuihui <178529795@qq.com>
Date: Sun, 12 Nov 2023 12:07:14 +0800
Subject: [PATCH] fix: replace client wait

---
 client.go | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/client.go b/client.go
index 03996d5..7885ac9 100644
--- a/client.go
+++ b/client.go
@@ -91,7 +91,8 @@ func (p *RTSPPuller) Pull() (err error) {
 		p.Error("Play", zap.Error(err))
 		return err
 	}
-	return p.Wait()
+	<-p.Done()
+	return p.Err()
 }
 
 type RTSPPusher struct {