remove ConnClient.CloseUDPListeners

This commit is contained in:
aler9
2020-11-08 19:48:18 +01:00
parent 4aa5848fba
commit 7c91f02459
3 changed files with 9 additions and 25 deletions

View File

@@ -184,17 +184,6 @@ func (c *ConnClient) checkState(allowed map[connClientState]struct{}) error {
allowed, c.state)
}
// CloseUDPListeners closes any open UDP listener.
func (c *ConnClient) CloseUDPListeners() {
for _, l := range c.udpRtpListeners {
l.close()
}
for _, l := range c.udpRtcpListeners {
l.close()
}
}
// NetConn returns the underlying net.Conn.
func (c *ConnClient) NetConn() net.Conn {
return c.conf.Conn

View File

@@ -79,13 +79,13 @@ func TestDialReadUDP(t *testing.T) {
time.Sleep(1 * time.Second)
loopDone := make(chan struct{})
defer func() { <-loopDone }()
conn, err := DialRead("rtsp://localhost:8554/teststream", StreamProtocolUDP)
require.NoError(t, err)
defer conn.Close()
loopDone := make(chan struct{})
defer func() { <-loopDone }()
go func() {
defer close(loopDone)
conn.LoopUDP()
@@ -93,8 +93,6 @@ func TestDialReadUDP(t *testing.T) {
_, err = conn.ReadFrameUDP(0, StreamTypeRtp)
require.NoError(t, err)
conn.CloseUDPListeners()
}
func TestDialReadTCP(t *testing.T) {
@@ -156,13 +154,13 @@ func TestDialReadRedirect(t *testing.T) {
time.Sleep(1 * time.Second)
loopDone := make(chan struct{})
defer func() { <-loopDone }()
conn, err := DialRead("rtsp://localhost:8554/path1", StreamProtocolUDP)
require.NoError(t, err)
defer conn.Close()
loopDone := make(chan struct{})
defer func() { <-loopDone }()
go func() {
defer close(loopDone)
conn.LoopUDP()
@@ -170,8 +168,6 @@ func TestDialReadRedirect(t *testing.T) {
_, err = conn.ReadFrameUDP(0, StreamTypeRtp)
require.NoError(t, err)
conn.CloseUDPListeners()
}
func TestDialPublishUDP(t *testing.T) {

View File

@@ -13,6 +13,9 @@ import (
// read all tracks with the UDP protocol.
func main() {
var wg sync.WaitGroup
defer wg.Wait()
// connect to the server and start reading all tracks
conn, err := gortsplib.DialRead("rtsp://localhost:8554/mystream", gortsplib.StreamProtocolUDP)
if err != nil {
@@ -20,10 +23,6 @@ func main() {
}
defer conn.Close()
var wg sync.WaitGroup
defer wg.Wait()
defer conn.CloseUDPListeners()
for _, track := range conn.Tracks() {
// read RTP frames
wg.Add(1)