apps/lkm
1
0
Fork 0
mirror of https://github.com/lkmio/lkm.git synced 2025-10-16 04:00:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

rtc支持单端口

Browse Source
This commit is contained in:
yangjiechina
2024-07-13 13:53:26 +08:00
parent 0114a36249
commit 27da874258
7 changed files with 114 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
stream/sink.go
View File

@@ -72,7 +72,11 @@ type Sink interface {
func GenerateSinkId(addr net.Addr) SinkId {
network := addr.Network()
if "tcp" == network {
id := uint64(binary.BigEndian.Uint32(addr.(*net.TCPAddr).IP.To4()))
to4 := addr.(*net.TCPAddr).IP.To4()
if to4 == nil {
to4 = make([]byte, 4)
}
id := uint64(binary.BigEndian.Uint32(to4))
id <<= 32
id |= uint64(addr.(*net.TCPAddr).Port << 16)
@@ -186,7 +190,9 @@ func (s *BaseSink) DesiredVideoCodecId() utils.AVCodecID {
// 拉流断开连接,不需要考虑线程安全
// 踢流走source管道删除,并且关闭Conn
func (s *BaseSink) Close() {
utils.Assert(SessionStateClose != s.State_)
if SessionStateClosed != s.State_ {
return
}
if s.Conn != nil {
s.Conn.Close()
@@ -202,12 +208,12 @@ func (s *BaseSink) Close() {
{
s.Lock()
defer s.UnLock()
if s.State_ == SessionStateClose {
if s.State_ == SessionStateClosed {
return
}
state = s.State_
s.State_ = SessionStateClose
s.State_ = SessionStateClosed
}
if state == SessionStateTransferring {