dandri/livekit
1
0
Fork 0
mirror of https://github.com/livekit/livekit.git synced 2026-03-30 19:55:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

improve reliability of data publish test

Browse Source
This commit is contained in:
David Zhao
2021-09-11 23:33:35 -07:00
parent 976e61f10c
commit 66997b37f9
1 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
test/client/client.go
View File

@@ -547,6 +547,14 @@ func (c *RTCClient) ensurePublisherConnected() error {
c.publisher.Negotiate()
}
dcOpen := utils.AtomicFlag{}
c.lossyDC.OnOpen(func() {
dcOpen.TrySet(true)
})
if c.lossyDC.ReadyState() == webrtc.DataChannelStateOpen {
dcOpen.TrySet(true)
}
// wait until connected
ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
defer cancel()
@@ -555,9 +563,7 @@ func (c *RTCClient) ensurePublisherConnected() error {
case <-ctx.Done():
return fmt.Errorf("could not connect publisher after timeout")
case <-time.After(10 * time.Millisecond):
if c.publisherConnected.Get() {
// TODO: pion seem to have a timing issue here. data channel is not actually open at this moment
time.Sleep(100 * time.Millisecond)
if c.publisherConnected.Get() && dcOpen.Get() {
return nil
}
}