提取webrtc推流、播放代码为单独的派生类

2026-06-15 04:25:58 +08:00 · 2021-10-15 16:27:17 +08:00
parent 8531b5e1cb
commit 7f3f47abbb
7 changed files with 379 additions and 229 deletions
--- a/webrtc/WebRtcPlayer.cpp
+++ b/webrtc/WebRtcPlayer.cpp
@@ -0,0 +1,101 @@
+/*
+ * Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved.
+ *
+ * This file is part of ZLMediaKit(https://github.com/xia-chu/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.
+ */
+
+#include "WebRtcPlayer.h"
+
+WebRtcPlayer::Ptr WebRtcPlayer::create(const EventPoller::Ptr &poller,
+                                       const RtspMediaSource::Ptr &src,
+                                       const MediaInfo &info) {
+    WebRtcPlayer::Ptr ret(new WebRtcPlayer(poller, src, info), [](WebRtcPlayer *ptr) {
+        ptr->onDestory();
+        delete ptr;
+    });
+    ret->onCreate();
+    return ret;
+}
+
+WebRtcPlayer::WebRtcPlayer(const EventPoller::Ptr &poller,
+                           const RtspMediaSource::Ptr &src,
+                           const MediaInfo &info) : WebRtcTransportImp(poller) {
+    InfoL << this;
+    _media_info = info;
+    _play_src = src;
+    CHECK(_play_src);
+}
+
+void WebRtcPlayer::onStartWebRTC() {
+    CHECK(_play_src);
+    WebRtcTransportImp::onStartWebRTC();
+    if (canSendRtp()) {
+        _play_src->pause(false);
+        _reader = _play_src->getRing()->attach(getPoller(), true);
+        weak_ptr<WebRtcPlayer> weak_self = static_pointer_cast<WebRtcPlayer>(shared_from_this());
+        _reader->setReadCB([weak_self](const RtspMediaSource::RingDataType &pkt) {
+            auto strongSelf = weak_self.lock();
+            if (!strongSelf) {
+                return;
+            }
+            size_t i = 0;
+            pkt->for_each([&](const RtpPacket::Ptr &rtp) {
+                strongSelf->beforeSendRtp(rtp, ++i == pkt->size());
+            });
+        });
+        _reader->setDetachCB([weak_self]() {
+            auto strongSelf = weak_self.lock();
+            if (!strongSelf) {
+                return;
+            }
+            strongSelf->onShutdown(SockException(Err_shutdown, "rtsp ring buffer detached"));
+        });
+
+        //确保该rtp codec类型对方支持
+        memset(_can_send_rtp, 0, sizeof(_can_send_rtp));
+        for (auto &m : _answer_sdp->media) {
+            _can_send_rtp[m.type] = m.direction == RtpDirection::sendonly || m.direction == RtpDirection::sendrecv;
+        }
+    }
+    //使用完毕后，释放强引用，这样确保推流器断开后能及时注销媒体
+    _play_src = nullptr;
+}
+
+void WebRtcPlayer::onDestory() {
+    WebRtcTransportImp::onDestory();
+
+    auto duration = getDuration();
+    auto bytes_usage = getBytesUsage();
+    //流量统计事件广播
+    GET_CONFIG(uint32_t, iFlowThreshold, General::kFlowThreshold);
+    if (_reader && getSession()) {
+        WarnL << "RTC播放器("
+              << _media_info._vhost << "/"
+              << _media_info._app << "/"
+              << _media_info._streamid
+              << ")结束播放,耗时(s):" << duration;
+        if (bytes_usage >= iFlowThreshold * 1024) {
+            NoticeCenter::Instance().emitEvent(Broadcast::kBroadcastFlowReport, _media_info, bytes_usage, duration,
+                                               true, static_cast<SockInfo &>(*getSession()));
+        }
+    }
+}
+
+void WebRtcPlayer::onRtcConfigure(RtcConfigure &configure) const {
+    CHECK(_play_src);
+    WebRtcTransportImp::onRtcConfigure(configure);
+    //这是播放
+    configure.audio.direction = configure.video.direction = RtpDirection::sendonly;
+    configure.setPlayRtspInfo(_play_src->getSdp());
+}
+
+void WebRtcPlayer::beforeSendRtp(const RtpPacket::Ptr &rtp, bool flush, bool rtx) {
+    if (!_can_send_rtp[rtp->type]) {
+        return;
+    }
+    onSendRtp(rtp, flush, rtx);
+}