mirror of
https://gitee.com/xia-chu/ZLMediaKit.git
synced 2026-05-23 01:57:50 +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包含
105 lines
3.1 KiB
C++
105 lines
3.1 KiB
C++
/*
|
|
* 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 "HlsMediaSource.h"
|
|
#include "Common/config.h"
|
|
|
|
using namespace toolkit;
|
|
|
|
namespace mediakit {
|
|
|
|
HlsCookieData::HlsCookieData(const MediaInfo &info, const std::shared_ptr<SockInfo> &sock_info) {
|
|
_info = info;
|
|
_sock_info = sock_info;
|
|
_added = std::make_shared<bool>(false);
|
|
addReaderCount();
|
|
}
|
|
|
|
void HlsCookieData::addReaderCount() {
|
|
if (!*_added) {
|
|
auto src = getMediaSource();
|
|
if (src) {
|
|
*_added = true;
|
|
_ring_reader = src->getRing()->attach(EventPollerPool::Instance().getPoller());
|
|
auto added = _added;
|
|
_ring_reader->setDetachCB([added]() {
|
|
// HlsMediaSource已经销毁
|
|
*added = false;
|
|
});
|
|
}
|
|
}
|
|
}
|
|
|
|
HlsCookieData::~HlsCookieData() {
|
|
if (*_added) {
|
|
uint64_t duration = (_ticker.createdTime() - _ticker.elapsedTime()) / 1000;
|
|
WarnL << _sock_info->getIdentifier() << "(" << _sock_info->get_peer_ip() << ":" << _sock_info->get_peer_port()
|
|
<< ") " << "HLS播放器(" << _info.shortUrl() << ")断开,耗时(s):" << duration;
|
|
|
|
GET_CONFIG(uint32_t, iFlowThreshold, General::kFlowThreshold);
|
|
uint64_t bytes = _bytes.load();
|
|
if (bytes >= iFlowThreshold * 1024) {
|
|
NoticeCenter::Instance().emitEvent(Broadcast::kBroadcastFlowReport, _info, bytes, duration, true, static_cast<SockInfo &>(*_sock_info));
|
|
}
|
|
}
|
|
}
|
|
|
|
void HlsCookieData::addByteUsage(size_t bytes) {
|
|
addReaderCount();
|
|
_bytes += bytes;
|
|
_ticker.resetTime();
|
|
}
|
|
|
|
void HlsCookieData::setMediaSource(const HlsMediaSource::Ptr &src) {
|
|
_src = src;
|
|
}
|
|
|
|
HlsMediaSource::Ptr HlsCookieData::getMediaSource() const {
|
|
return _src.lock();
|
|
}
|
|
|
|
void HlsMediaSource::setIndexFile(std::string index_file)
|
|
{
|
|
if (!_ring) {
|
|
std::weak_ptr<HlsMediaSource> weakSelf = std::dynamic_pointer_cast<HlsMediaSource>(shared_from_this());
|
|
auto lam = [weakSelf](int size) {
|
|
auto strongSelf = weakSelf.lock();
|
|
if (!strongSelf) {
|
|
return;
|
|
}
|
|
strongSelf->onReaderChanged(size);
|
|
};
|
|
_ring = std::make_shared<RingType>(0, std::move(lam));
|
|
regist();
|
|
}
|
|
|
|
//赋值m3u8索引文件内容
|
|
std::lock_guard<std::mutex> lck(_mtx_index);
|
|
_index_file = std::move(index_file);
|
|
|
|
if (!_index_file.empty()) {
|
|
_list_cb.for_each([&](const std::function<void(const std::string& str)>& cb) { cb(_index_file); });
|
|
_list_cb.clear();
|
|
}
|
|
}
|
|
|
|
void HlsMediaSource::getIndexFile(std::function<void(const std::string& str)> cb)
|
|
{
|
|
std::lock_guard<std::mutex> lck(_mtx_index);
|
|
if (!_index_file.empty()) {
|
|
cb(_index_file);
|
|
return;
|
|
}
|
|
//等待生成m3u8文件
|
|
_list_cb.emplace_back(std::move(cb));
|
|
}
|
|
|
|
} // namespace mediakit
|