Try removing buffered candidates

kylecarbs · kylecarbs · commit c5c8b6516455 · 2022-01-30T22:57:57.000Z
diff --git a/peer/conn.go b/peer/conn.go
@@ -77,7 +77,6 @@ func newWithClientOrServer(servers []webrtc.ICEServer, client bool, opts *ConnOp
 		localCandidateChannel:           make(chan webrtc.ICECandidateInit),
 		localSessionDescriptionChannel:  make(chan webrtc.SessionDescription),
 		remoteSessionDescriptionChannel: make(chan webrtc.SessionDescription),
-		pendingCandidatesToSend:         make([]webrtc.ICECandidateInit, 0),
 	}
 	if client {
 		// If we're the client, we want to flip the echo and
@@ -129,9 +128,6 @@ type Conn struct {
 
 	negotiateMutex sync.Mutex
 
-	pendingCandidatesToSend      []webrtc.ICECandidateInit
-	pendingCandidatesToSendMutex sync.Mutex
-
 	pingChannelID     uint16
 	pingEchoChannelID uint16
 
@@ -227,16 +223,6 @@ func (c *Conn) init() error {
 		// Run this in a goroutine so we don't block pion/webrtc
 		// from continuing.
 		go func() {
-			c.pendingCandidatesToSendMutex.Lock()
-			defer c.pendingCandidatesToSendMutex.Unlock()
-			// If the remote description hasn't been set yet, we queue the send of these candidates.
-			// It may work to send these immediately, but at the time of writing this package is
-			// unstable, so better being safe than sorry.
-			if c.rtc.RemoteDescription() == nil {
-				c.pendingCandidatesToSend = append(c.pendingCandidatesToSend, iceCandidate.ToJSON())
-				c.opts.Logger.Debug(context.Background(), "buffering local candidate")
-				return
-			}
 			c.opts.Logger.Debug(context.Background(), "sending local candidate")
 			select {
 			case <-c.closed:
@@ -336,24 +322,6 @@ func (c *Conn) negotiate() {
 		}
 		c.opts.Logger.Debug(context.Background(), "sent answer")
 	}
-
-	// Flush bufferred candidates after both sides have been negotiated!
-	go func() {
-		c.pendingCandidatesToSendMutex.Lock()
-		defer c.pendingCandidatesToSendMutex.Unlock()
-		for _, pendingCandidate := range c.pendingCandidatesToSend {
-			select {
-			case <-c.closed:
-				return
-			case c.localCandidateChannel <- pendingCandidate:
-			}
-			c.opts.Logger.Debug(context.Background(), "flushed buffered local candidate")
-		}
-		c.opts.Logger.Debug(context.Background(), "flushed buffered local candidates",
-			slog.F("count", len(c.pendingCandidatesToSend)),
-		)
-		c.pendingCandidatesToSend = make([]webrtc.ICECandidateInit, 0)
-	}()
 }
 
 // AddRemoteCandidate adds a remote candidate to the RTC connection.
@@ -366,12 +334,12 @@ func (c *Conn) AddRemoteCandidate(i webrtc.ICECandidateInit) {
 	go func() {
 		c.negotiateMutex.Lock()
 		defer c.negotiateMutex.Unlock()
-		if c.isClosed() {
-			return
-		}
 		c.opts.Logger.Debug(context.Background(), "accepting candidate", slog.F("length", len(i.Candidate)))
 		err := c.rtc.AddICECandidate(i)
 		if err != nil {
+			if c.rtc.ConnectionState() == webrtc.PeerConnectionStateClosed {
+				return
+			}
 			_ = c.CloseWithError(xerrors.Errorf("accept candidate: %w", err))
 		}
 	}()
