diff --git a/pkg/webrtc/conn.go b/pkg/webrtc/conn.go index f853bf43..063831f0 100644 --- a/pkg/webrtc/conn.go +++ b/pkg/webrtc/conn.go @@ -161,21 +161,21 @@ func (c *Conn) AddCandidate(candidate string) error { return c.pc.AddICECandidate(webrtc.ICECandidateInit{Candidate: candidate}) } -func (c *Conn) GetSenderTrack(mid string) *Track { - if tr := c.getTranseiver(mid); tr != nil { - if s := tr.Sender(); s != nil { - if t := s.Track().(*Track); t != nil { - return t - } +func (c *Conn) getTranseiver(mid string) *webrtc.RTPTransceiver { + for _, tr := range c.pc.GetTransceivers() { + if tr.Mid() == mid { + return tr } } return nil } -func (c *Conn) getTranseiver(mid string) *webrtc.RTPTransceiver { - for _, tr := range c.pc.GetTransceivers() { - if tr.Mid() == mid { - return tr +func (c *Conn) getSenderTrack(mid string) *Track { + if tr := c.getTranseiver(mid); tr != nil { + if s := tr.Sender(); s != nil { + if t := s.Track().(*Track); t != nil { + return t + } } } return nil @@ -209,7 +209,7 @@ func (c *Conn) getMediaCodec(remote *webrtc.TrackRemote) (*core.Media, *core.Cod // check GetTrack panic(core.Caller()) - // return nil, nil + return nil, nil } func sanitizeIP6(host string) string { @@ -217,4 +217,4 @@ func sanitizeIP6(host string) string { return "[" + host + "]" } return host -} +} \ No newline at end of file diff --git a/pkg/webrtc/consumer.go b/pkg/webrtc/consumer.go index 767394df..1efb1507 100644 --- a/pkg/webrtc/consumer.go +++ b/pkg/webrtc/consumer.go @@ -32,7 +32,7 @@ func (c *Conn) AddTrack(media *core.Media, codec *core.Codec, track *core.Receiv panic(core.Caller()) } - localTrack := c.GetSenderTrack(media.ID) + localTrack := c.getSenderTrack(media.ID) if localTrack == nil { return errors.New("webrtc: can't get track") } @@ -87,4 +87,4 @@ func (c *Conn) AddTrack(media *core.Media, codec *core.Codec, track *core.Receiv c.Senders = append(c.Senders, sender) return nil -} +} \ No newline at end of file