mirror of
https://github.com/ZLMediaKit/ZLMediaKit.git
synced 2024-10-06 12:13:49 +08:00
754073918a
* 优化MultiMediaSourceMuxer头文件包含 * 将MediaSinkDelegate和Demux移到MediaSink中 * MediaSource头文件重构, 独立出PacketCache.h 精简Frame和Track的头文件 * Rtmp头文件重构 * Rtsp头文件重构 * webrtc头文件重构 * 规范.h头文件包含，并将其移到.cpp中: - 尽量不包含Common\config.h - Util\File.h - Rtsp/RtspPlayer.h - Rtmp/RtmpPlayer.h * 删除多余的Stamp.h和Base64包含
86 lines
2.2 KiB
C++
86 lines
2.2 KiB
C++
/*
|
||
* Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved.
|
||
*
|
||
* This file is part of ZLMediaKit(https://github.com/ZLMediaKit/ZLMediaKit).
|
||
*
|
||
* Use of this source code is governed by MIT license that can be found in the
|
||
* LICENSE file in the root of the source tree. All contributing project authors
|
||
* may be found in the AUTHORS file in the root of the source tree.
|
||
*/
|
||
|
||
#ifndef ZLMEDIAKIT_NACK_H
|
||
#define ZLMEDIAKIT_NACK_H
|
||
|
||
#include <set>
|
||
#include <map>
|
||
#include <deque>
|
||
#include <unordered_map>
|
||
#include "Rtsp/Rtsp.h"
|
||
#include "Rtcp/RtcpFCI.h"
|
||
|
||
namespace mediakit {
|
||
|
||
class NackList {
|
||
public:
|
||
NackList() = default;
|
||
~NackList() = default;
|
||
|
||
void pushBack(RtpPacket::Ptr rtp);
|
||
void forEach(const FCI_NACK &nack, const std::function<void(const RtpPacket::Ptr &rtp)> &cb);
|
||
|
||
private:
|
||
void popFront();
|
||
uint32_t getCacheMS();
|
||
int64_t getRtpStamp(uint16_t seq);
|
||
RtpPacket::Ptr *getRtp(uint16_t seq);
|
||
|
||
private:
|
||
uint32_t _cache_ms_check = 0;
|
||
std::deque<uint16_t> _nack_cache_seq;
|
||
std::unordered_map<uint16_t, RtpPacket::Ptr> _nack_cache_pkt;
|
||
};
|
||
|
||
class NackContext {
|
||
public:
|
||
using Ptr = std::shared_ptr<NackContext>;
|
||
using onNack = std::function<void(const FCI_NACK &nack)>;
|
||
//最大保留的rtp丢包状态个数
|
||
static constexpr auto kNackMaxSize = 2048;
|
||
// rtp丢包状态最长保留时间
|
||
static constexpr auto kNackMaxMS = 3 * 1000;
|
||
// nack最多请求重传10次
|
||
static constexpr auto kNackMaxCount = 10;
|
||
// nack重传频率，rtt的倍数
|
||
static constexpr auto kNackIntervalRatio = 1.0f;
|
||
|
||
NackContext() = default;
|
||
~NackContext() = default;
|
||
|
||
void received(uint16_t seq, bool is_rtx = false);
|
||
void setOnNack(onNack cb);
|
||
uint64_t reSendNack();
|
||
|
||
private:
|
||
void eraseFrontSeq();
|
||
void doNack(const FCI_NACK &nack, bool record_nack);
|
||
void recordNack(const FCI_NACK &nack);
|
||
void onRtx(uint16_t seq);
|
||
|
||
private:
|
||
int _rtt = 50;
|
||
onNack _cb;
|
||
std::set<uint16_t> _seq;
|
||
uint16_t _last_max_seq = 0;
|
||
|
||
struct NackStatus {
|
||
uint64_t first_stamp;
|
||
uint64_t update_stamp;
|
||
int nack_count = 0;
|
||
};
|
||
std::map<uint16_t /*seq*/, NackStatus> _nack_send_status;
|
||
};
|
||
|
||
} // namespace mediakit
|
||
|
||
#endif //ZLMEDIAKIT_NACK_H
|