优化rtp统计性能

src/Rtcp/RtcpContext.cpp

@@ -9,6 +9,7 @@
  */
 
 #include "RtcpContext.h"
+#include "Util/logger.h"
 using namespace toolkit;
 
 namespace mediakit {
@@ -17,18 +18,19 @@ void RtcpContext::clear(){
     memset(this, 0, sizeof(RtcpContext));
 }
 
-RtcpContext::RtcpContext(uint32_t sample_rate){
+RtcpContext::RtcpContext(uint32_t sample_rate, bool is_receiver) {
     _sample_rate = sample_rate;
+    _is_receiver = is_receiver;
 }
 
 void RtcpContext::onRtp(uint16_t seq, uint32_t stamp, size_t bytes) {
-    _bytes += bytes;
-    ++_packets;
-
+    if (_is_receiver) {
+        //接收者才做复杂的统计运算
         auto sys_stamp = getCurrentMillisecond();
         if (_last_rtp_sys_stamp) {
             //计算时间戳抖动值
-        double diff = double(int64_t(sys_stamp) - int64_t(_last_rtp_sys_stamp) - int64_t(stamp) + int64_t(_last_rtp_stamp));
+            double diff = double(
+                    int64_t(sys_stamp) - int64_t(_last_rtp_sys_stamp) - int64_t(stamp) + int64_t(_last_rtp_stamp));
             if (diff < 0) {
                 diff = -diff;
             }
@@ -58,15 +60,23 @@ void RtcpContext::onRtp(uint16_t seq, uint32_t stamp, size_t bytes) {
             _seq_base = seq;
         }
 
-    _last_rtp_stamp = stamp;
-    _last_rtp_sys_stamp = sys_stamp;
         _last_rtp_seq = seq;
+        _last_rtp_sys_stamp = sys_stamp;
+    }
+
+    ++_packets;
+    _bytes += bytes;
+    _last_rtp_stamp = stamp;
 }
 
 void RtcpContext::onRtcp(RtcpHeader *rtcp) {
     if ((RtcpType) rtcp->pt != RtcpType::RTCP_SR) {
         return;
     }
+    if (!_is_receiver) {
+        WarnL << "rtp发送者收到sr包";
+        return;
+    }
     auto rtcp_sr = (RtcpSR *) (rtcp);
     /**
      last SR timestamp (LSR): 32 bits
@@ -80,6 +90,9 @@ void RtcpContext::onRtcp(RtcpHeader *rtcp){
 }
 
 size_t RtcpContext::getExpectedPackets() const {
+    if (!_is_receiver) {
+        throw std::runtime_error("rtp发送者无法统计应收包数");
+    }
     return (_seq_cycles << 16) + _seq_max - _seq_base + 1;
 }
 
@@ -91,6 +104,9 @@ size_t RtcpContext::getExpectedPacketsInterval(){
 }
 
 size_t RtcpContext::getLost() {
+    if (!_is_receiver) {
+        throw std::runtime_error("rtp发送者无法统计丢包率");
+    }
     return getExpectedPackets() - _packets;
 }
 
@@ -102,6 +118,9 @@ size_t RtcpContext::geLostInterval(){
 }
 
 Buffer::Ptr RtcpContext::createRtcpSR(uint32_t rtcp_ssrc) {
+    if (_is_receiver) {
+        throw std::runtime_error("rtp接收者尝试发送sr包");
+    }
     auto rtcp = RtcpSR::create(0);
     rtcp->ssrc = htonl(rtcp_ssrc);
 
@@ -117,6 +136,9 @@ Buffer::Ptr RtcpContext::createRtcpSR(uint32_t rtcp_ssrc){
 }
 
 Buffer::Ptr RtcpContext::createRtcpRR(uint32_t rtcp_ssrc, uint32_t rtp_ssrc) {
+    if (!_is_receiver) {
+        throw std::runtime_error("rtp发送者尝试发送rr包");
+    }
     auto rtcp = RtcpRR::create(1);
     rtcp->ssrc = htonl(rtcp_ssrc);
 

src/Rtcp/RtcpContext.h

@@ -23,8 +23,9 @@ public:
     /**
      * 创建rtcp上下文
      * @param sample_rate 音频采用率，视频一般为90000
+     * @param is_receiver 是否为rtp接收者，接收者更消耗性能
      */
-    RtcpContext(uint32_t sample_rate);
+    RtcpContext(uint32_t sample_rate, bool is_receiver);
 
     /**
      * 输出或输入rtp时调用
@@ -82,6 +83,8 @@ private:
     size_t geLostInterval();
 
 private:
+    //是否为接收者
+    bool _is_receiver;
     //时间戳抖动值
     double _jitter = 0;
     //视频默认90000,音频为采样率

src/Rtsp/RtspPlayer.cpp

@@ -204,7 +204,7 @@ void RtspPlayer::handleResDESCRIBE(const Parser& parser) {
         throw std::runtime_error("onCheckSDP faied");
     }
     for (auto &track : _sdp_track) {
-        _rtcp_context.emplace_back(std::make_shared<RtcpContext>(track->_samplerate));
+        _rtcp_context.emplace_back(std::make_shared<RtcpContext>(track->_samplerate, true));
     }
     sendSetup(0);
 }

src/Rtsp/RtspPusher.cpp

@@ -177,7 +177,7 @@ void RtspPusher::sendAnnounce() {
         throw std::runtime_error("无有效的Sdp Track");
     }
     for (auto &track : _track_vec) {
-        _rtcp_context.emplace_back(std::make_shared<RtcpContext>(track->_samplerate));
+        _rtcp_context.emplace_back(std::make_shared<RtcpContext>(track->_samplerate, false));
     }
     _on_res_func = std::bind(&RtspPusher::handleResAnnounce, this, placeholders::_1);
     sendRtspRequest("ANNOUNCE", _url, {}, src->getSdp());

src/Rtsp/RtspSession.cpp

@@ -235,8 +235,9 @@ void RtspSession::handleReq_ANNOUNCE(const Parser &parser) {
         sendRtspResponse("403 Forbidden", {"Content-Type", "text/plain"}, err);
         throw SockException(Err_shutdown,StrPrinter << err << ":" << full_url);
     }
+    _rtcp_context.clear();
     for (auto &track : _sdp_track) {
-        _rtcp_context.emplace_back(std::make_shared<RtcpContext>(track->_samplerate));
+        _rtcp_context.emplace_back(std::make_shared<RtcpContext>(track->_samplerate, true));
     }
     _push_src = std::make_shared<RtspMediaSourceImp>(_media_info._vhost, _media_info._app, _media_info._streamid);
     _push_src->setListener(dynamic_pointer_cast<MediaSourceEvent>(shared_from_this()));
@@ -406,7 +407,7 @@ void RtspSession::onAuthSuccess() {
             return;
         }
         for (auto &track : strongSelf->_sdp_track) {
-            strongSelf->_rtcp_context.emplace_back(std::make_shared<RtcpContext>(track->_samplerate));
+            strongSelf->_rtcp_context.emplace_back(std::make_shared<RtcpContext>(track->_samplerate, false));
         }
         strongSelf->_sessionid = makeRandStr(12);
         strongSelf->_play_src = rtsp_src;