From 022857c3d7534d07b95f4a788bf4269988cc7cd4 Mon Sep 17 00:00:00 2001
From: dexter <178529795@qq.com>
Date: Tue, 6 Sep 2022 18:56:19 +0800
Subject: [PATCH] =?UTF-8?q?=F0=9F=91=8C=20IMPROVE:=20=E5=A2=9E=E5=8A=A0?=
 =?UTF-8?q?=E4=B8=80=E4=BA=9B=E9=94=99=E8=AF=AF=E4=BF=A1=E6=81=AF=E8=BE=93?=
 =?UTF-8?q?=E5=87=BA?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 client.go | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/client.go b/client.go
index 1a9ac2b..550c148 100644
--- a/client.go
+++ b/client.go
@@ -3,6 +3,7 @@ package rtsp
 import (
 	"github.com/aler9/gortsplib"
 	"github.com/aler9/gortsplib/pkg/url"
+	"go.uber.org/zap"
 	"m7s.live/engine/v4"
 )
 
@@ -50,16 +51,19 @@ func (p *RTSPPuller) Connect() error {
 func (p *RTSPPuller) Pull() {
 	u, _ := url.Parse(p.RemoteURL)
 	if _, err := p.Options(u); err != nil {
+		p.Error("Options", zap.Error(err))
 		return
 	}
 	// find published tracks
 	tracks, baseURL, _, err := p.Describe(u)
 	if err != nil {
+		p.Error("Describe", zap.Error(err))
 		return
 	}
 	p.tracks = tracks
 	p.SetTracks()
 	if err = p.SetupAndPlay(tracks, baseURL); err != nil {
+		p.Error("SetupAndPlay", zap.Error(err))
 		return
 	}
 	p.Wait()
@@ -95,10 +99,12 @@ func (p *RTSPPusher) Connect() error {
 	// parse URL
 	u, err := url.Parse(p.RemoteURL)
 	if err != nil {
+		p.Error("url.Parse", zap.Error(err))
 		return err
 	}
 	// connect to the server
 	if err = p.Client.Start(u.Scheme, u.Host); err != nil {
+		p.Error("Client.Start", zap.Error(err))
 		return err
 	}
 	_, err = p.Client.Options(u)
@@ -119,15 +125,18 @@ func (p *RTSPPusher) Push() (err error) {
 	// 	}
 	// }
 	if _, err = p.Announce(u, p.tracks); err != nil {
+		p.Error("Announce", zap.Error(err))
 		return
 	}
 	for _, track := range p.tracks {
 		_, err = p.Setup(false, track, u, 0, 0)
 		if err != nil {
+			p.Error("Setup", zap.Error(err))
 			return
 		}
 	}
 	if _, err = p.Record(); err != nil {
+		p.Error("Record", zap.Error(err))
 		return
 	}
 	p.PlayRTP()