118 lines
3.7 KiB
C++
118 lines
3.7 KiB
C++
/*
|
|
* Copyright (c) 2016-present 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-like 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已经销毁 [AUTO-TRANSLATED:bedb0385]
|
|
// HlsMediaSource has been destroyed
|
|
*added = false;
|
|
});
|
|
auto info = _sock_info;
|
|
_ring_reader->setGetInfoCB([info]() {
|
|
Any ret;
|
|
ret.set(info);
|
|
return ret;
|
|
});
|
|
}
|
|
}
|
|
}
|
|
|
|
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) {
|
|
try {
|
|
NOTICE_EMIT(BroadcastFlowReportArgs, Broadcast::kBroadcastFlowReport, _info, bytes, duration, true, *_sock_info);
|
|
} catch (std::exception &ex) {
|
|
WarnL << "Exception occurred: " << ex.what();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
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::static_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索引文件内容 [AUTO-TRANSLATED:c11882b5]
|
|
// Assign m3u8 index file content
|
|
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文件 [AUTO-TRANSLATED:c3ae3286]
|
|
// Waiting for m3u8 file generation
|
|
_list_cb.emplace_back(std::move(cb));
|
|
}
|
|
|
|
} // namespace mediakit
|