aunix/ZLMediaKit
1
0
Fork 0
mirror of https://gitee.com/xia-chu/ZLMediaKit.git synced 2026-05-06 10:57:50 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

rtp tcp级联速度太慢时主动限制上游接收速度

Browse Source 
防止发送速度跟不上接受速度导致发送超时
This commit is contained in:
xia-chu 2025-04-20 18:53:02 +08:00
parent 986a63bb73
commit 9e6b33b02e
3 changed files with 25 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/Common/MultiMediaSourceMuxer.cpp
View File

@ -426,7 +426,7 @@ void MultiMediaSourceMuxer::startSendRtp(MediaSource &sender, const MediaSourceE
}
});
rtp_sender->startSend(args, [ssrc,ssrc_multi_send, weak_self, rtp_sender, cb, tracks, ring, poller](uint16_t local_port, const SockException &ex) mutable {
rtp_sender->startSend(sender, args, [ssrc,ssrc_multi_send, weak_self, rtp_sender, cb, tracks, ring, poller](uint16_t local_port, const SockException &ex) mutable {
cb(local_port, ex);
auto strong_self = weak_self.lock();
if (!strong_self || ex) {

23
src/Rtp/RtpSender.cpp
View File

@ -35,7 +35,16 @@ RtpSender::~RtpSender() {
}
}
void RtpSender::startSend(const MediaSourceEvent::SendRtpArgs &args, const function<void(uint16_t local_port, const SockException &ex)> &cb){
void RtpSender::startSend(const MediaSource &sender, const MediaSourceEvent::SendRtpArgs &args, const function<void(uint16_t local_port, const SockException &ex)> &cb){
auto origin_socket = sender.getOriginSock();
_origin_socket = dynamic_pointer_cast<Socket>(origin_socket);
if (!_origin_socket) {
auto process = dynamic_pointer_cast<RtpProcess>(origin_socket);
if (process) {
_origin_socket = process->getSock();
}
}
_args = args;
if (!_interface) {
// 重连时不重新创建对象 [AUTO-TRANSLATED:b788cd5d]
@ -313,6 +322,15 @@ void RtpSender::onConnect() {
}
});
}
if (_socket_rtp->sockType() == toolkit::SockNum::Sock_TCP && _origin_socket) {
// rtp 端口是TCP端口，转发速度应当控制收流速度
auto origin_socket = _origin_socket;
_socket_rtp->setOnFlush([origin_socket]() {
origin_socket->enableRecv(true);
return true;
});
}
InfoL << "startSend rtp success: " << _socket_rtp->get_peer_ip() << ":" << _socket_rtp->get_peer_port() << ", data_type: " << _args.data_type << ", con_type: " << _args.con_type;
}
@ -433,6 +451,9 @@ void RtpSender::onFlushRtpList(shared_ptr<List<Buffer::Ptr>> rtp_list) {
}
default: CHECK(0);
}
if (_socket_rtp->sockType() == toolkit::SockNum::Sock_TCP && _socket_rtp->isSocketBusy() && _origin_socket) {
_origin_socket->enableRecv(false);
}
});
};
if (_args.con_type != MediaSourceEvent::SendRtpArgs::kVoiceTalk) {

3
src/Rtp/RtpSender.h
View File

@ -40,7 +40,7 @@ public:
* [AUTO-TRANSLATED:c31bd9b3]
*/
void startSend(const MediaSourceEvent::SendRtpArgs &args, const std::function<void(uint16_t local_port, const toolkit::SockException &ex)> &cb);
void startSend(const MediaSource &sender, const MediaSourceEvent::SendRtpArgs &args, const std::function<void(uint16_t local_port, const toolkit::SockException &ex)> &cb);
/**
* 输入帧数据
@ -116,6 +116,7 @@ private:
private:
bool _is_connect = false;
toolkit::Socket::Ptr _origin_socket;
MediaSourceEvent::SendRtpArgs _args;
toolkit::Socket::Ptr _socket_rtp;
toolkit::Socket::Ptr _socket_rtcp;