360 lines
11 KiB
C++
360 lines
11 KiB
C++
/*
|
||
* Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved.
|
||
*
|
||
* This file is part of ZLMediaKit(https://github.com/xiongziliang/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_MEDIASOURCE_H
|
||
#define ZLMEDIAKIT_MEDIASOURCE_H
|
||
|
||
#include <mutex>
|
||
#include <string>
|
||
#include <memory>
|
||
#include <functional>
|
||
#include <unordered_map>
|
||
#include "Common/config.h"
|
||
#include "Common/Parser.h"
|
||
#include "Util/logger.h"
|
||
#include "Util/TimeTicker.h"
|
||
#include "Util/NoticeCenter.h"
|
||
#include "Util/List.h"
|
||
#include "Network/Socket.h"
|
||
#include "Rtsp/Rtsp.h"
|
||
#include "Rtmp/Rtmp.h"
|
||
#include "Extension/Track.h"
|
||
#include "Record/Recorder.h"
|
||
|
||
using namespace std;
|
||
using namespace toolkit;
|
||
|
||
namespace toolkit{
|
||
class TcpSession;
|
||
}// namespace toolkit
|
||
|
||
namespace mediakit {
|
||
|
||
enum class MediaOriginType : uint8_t {
|
||
unknown = 0,
|
||
rtmp_push ,
|
||
rtsp_push,
|
||
rtp_push,
|
||
pull,
|
||
ffmpeg_pull,
|
||
mp4_vod,
|
||
device_chn
|
||
};
|
||
|
||
string getOriginTypeString(MediaOriginType type);
|
||
|
||
class MediaSource;
|
||
class MediaSourceEvent{
|
||
public:
|
||
friend class MediaSource;
|
||
MediaSourceEvent(){};
|
||
virtual ~MediaSourceEvent(){};
|
||
|
||
// 获取媒体源类型
|
||
virtual MediaOriginType getOriginType(MediaSource &sender) const { return MediaOriginType::unknown; }
|
||
// 获取媒体源url或者文件路径
|
||
virtual string getOriginUrl(MediaSource &sender) const { return ""; }
|
||
// 获取媒体源客户端相关信息
|
||
virtual std::shared_ptr<SockInfo> getOriginSock(MediaSource &sender) const { return nullptr; }
|
||
|
||
// 通知拖动进度条
|
||
virtual bool seekTo(MediaSource &sender, uint32_t stamp) { return false; }
|
||
// 通知其停止产生流
|
||
virtual bool close(MediaSource &sender, bool force) { return false; }
|
||
// 获取观看总人数
|
||
virtual int totalReaderCount(MediaSource &sender) = 0;
|
||
// 通知观看人数变化
|
||
virtual void onReaderChanged(MediaSource &sender, int size);
|
||
//流注册或注销事件
|
||
virtual void onRegist(MediaSource &sender, bool regist) {};
|
||
|
||
////////////////////////仅供MultiMediaSourceMuxer对象继承////////////////////////
|
||
// 开启或关闭录制
|
||
virtual bool setupRecord(MediaSource &sender, Recorder::type type, bool start, const string &custom_path) { return false; };
|
||
// 获取录制状态
|
||
virtual bool isRecording(MediaSource &sender, Recorder::type type) { return false; };
|
||
// 获取所有track相关信息
|
||
virtual vector<Track::Ptr> getTracks(MediaSource &sender, bool trackReady = true) const { return vector<Track::Ptr>(); };
|
||
// 开始发送ps-rtp
|
||
virtual void startSendRtp(MediaSource &sender, const string &dst_url, uint16_t dst_port, uint32_t ssrc, bool is_udp, const function<void(const SockException &ex)> &cb) { cb(SockException(Err_other, "not implemented"));};
|
||
// 停止发送ps-rtp
|
||
virtual bool stopSendRtp(MediaSource &sender) {return false; }
|
||
|
||
private:
|
||
Timer::Ptr _async_close_timer;
|
||
};
|
||
|
||
//该对象用于拦截感兴趣的MediaSourceEvent事件
|
||
class MediaSourceEventInterceptor : public MediaSourceEvent{
|
||
public:
|
||
MediaSourceEventInterceptor(){}
|
||
~MediaSourceEventInterceptor() override {}
|
||
|
||
MediaOriginType getOriginType(MediaSource &sender) const override;
|
||
string getOriginUrl(MediaSource &sender) const override;
|
||
std::shared_ptr<SockInfo> getOriginSock(MediaSource &sender) const override;
|
||
|
||
bool seekTo(MediaSource &sender, uint32_t stamp) override;
|
||
bool close(MediaSource &sender, bool force) override;
|
||
int totalReaderCount(MediaSource &sender) override;
|
||
void onReaderChanged(MediaSource &sender, int size) override;
|
||
void onRegist(MediaSource &sender, bool regist) override;
|
||
bool setupRecord(MediaSource &sender, Recorder::type type, bool start, const string &custom_path) override;
|
||
bool isRecording(MediaSource &sender, Recorder::type type) override;
|
||
vector<Track::Ptr> getTracks(MediaSource &sender, bool trackReady = true) const override;
|
||
void startSendRtp(MediaSource &sender, const string &dst_url, uint16_t dst_port, uint32_t ssrc, bool is_udp, const function<void(const SockException &ex)> &cb) override;
|
||
bool stopSendRtp(MediaSource &sender) override;
|
||
|
||
protected:
|
||
std::weak_ptr<MediaSourceEvent> _listener;
|
||
};
|
||
|
||
/**
|
||
* 解析url获取媒体相关信息
|
||
*/
|
||
class MediaInfo{
|
||
public:
|
||
~MediaInfo() {}
|
||
MediaInfo() {}
|
||
MediaInfo(const string &url) { parse(url); }
|
||
void parse(const string &url);
|
||
|
||
public:
|
||
string _full_url;
|
||
string _schema;
|
||
string _host;
|
||
string _port;
|
||
string _vhost;
|
||
string _app;
|
||
string _streamid;
|
||
string _param_strs;
|
||
};
|
||
|
||
class BytesSpeed {
|
||
public:
|
||
BytesSpeed() = default;
|
||
~BytesSpeed() = default;
|
||
|
||
/**
|
||
* 添加统计字节
|
||
*/
|
||
BytesSpeed& operator += (uint64_t bytes) {
|
||
_bytes += bytes;
|
||
if (_bytes > 1024 * 1024) {
|
||
//数据大于1MB就计算一次网速
|
||
computeSpeed();
|
||
}
|
||
return *this;
|
||
}
|
||
|
||
/**
|
||
* 获取速度，单位bytes/s
|
||
*/
|
||
int getSpeed() {
|
||
if (_ticker.elapsedTime() < 1000) {
|
||
//获取频率小于1秒，那么返回上次计算结果
|
||
return _speed;
|
||
}
|
||
return computeSpeed();
|
||
}
|
||
|
||
private:
|
||
uint64_t computeSpeed() {
|
||
auto elapsed = _ticker.elapsedTime();
|
||
if (!elapsed) {
|
||
return _speed;
|
||
}
|
||
_speed = _bytes * 1000 / elapsed;
|
||
_ticker.resetTime();
|
||
_bytes = 0;
|
||
return _speed;
|
||
}
|
||
|
||
private:
|
||
int _speed = 0;
|
||
uint64_t _bytes = 0;
|
||
Ticker _ticker;
|
||
};
|
||
|
||
/**
|
||
* 媒体源，任何rtsp/rtmp的直播流都源自该对象
|
||
*/
|
||
class MediaSource: public TrackSource, public enable_shared_from_this<MediaSource> {
|
||
public:
|
||
typedef std::shared_ptr<MediaSource> Ptr;
|
||
typedef unordered_map<string, weak_ptr<MediaSource> > StreamMap;
|
||
typedef unordered_map<string, StreamMap > AppStreamMap;
|
||
typedef unordered_map<string, AppStreamMap > VhostAppStreamMap;
|
||
typedef unordered_map<string, VhostAppStreamMap > SchemaVhostAppStreamMap;
|
||
|
||
MediaSource(const string &schema, const string &vhost, const string &app, const string &stream_id) ;
|
||
virtual ~MediaSource() ;
|
||
|
||
////////////////获取MediaSource相关信息////////////////
|
||
|
||
// 获取协议类型
|
||
const string& getSchema() const;
|
||
// 虚拟主机
|
||
const string& getVhost() const;
|
||
// 应用名
|
||
const string& getApp() const;
|
||
// 流id
|
||
const string& getId() const;
|
||
|
||
// 获取所有Track
|
||
vector<Track::Ptr> getTracks(bool ready = true) const override;
|
||
|
||
// 获取流当前时间戳
|
||
virtual uint32_t getTimeStamp(TrackType type) { return 0; };
|
||
// 设置时间戳
|
||
virtual void setTimeStamp(uint32_t stamp) {};
|
||
|
||
// 获取数据速率，单位bytes/s
|
||
int getBytesSpeed();
|
||
|
||
////////////////MediaSourceEvent相关接口实现////////////////
|
||
|
||
// 设置监听者
|
||
void setListener(const std::weak_ptr<MediaSourceEvent> &listener);
|
||
// 获取监听者
|
||
const std::weak_ptr<MediaSourceEvent>& getListener() const;
|
||
|
||
// 本协议获取观看者个数，可能返回本协议的观看人数，也可能返回总人数
|
||
virtual int readerCount() = 0;
|
||
// 观看者个数，包括(hls/rtsp/rtmp)
|
||
virtual int totalReaderCount();
|
||
|
||
// 获取媒体源类型
|
||
MediaOriginType getOriginType() const;
|
||
// 获取媒体源url或者文件路径
|
||
string getOriginUrl() const;
|
||
// 获取媒体源客户端相关信息
|
||
std::shared_ptr<SockInfo> getOriginSock() const;
|
||
|
||
// 拖动进度条
|
||
bool seekTo(uint32_t stamp);
|
||
// 关闭该流
|
||
bool close(bool force);
|
||
// 该流观看人数变化
|
||
void onReaderChanged(int size);
|
||
// 开启或关闭录制
|
||
bool setupRecord(Recorder::type type, bool start, const string &custom_path);
|
||
// 获取录制状态
|
||
bool isRecording(Recorder::type type);
|
||
// 开始发送ps-rtp
|
||
void startSendRtp(const string &dst_url, uint16_t dst_port, uint32_t ssrc, bool is_udp, const function<void(const SockException &ex)> &cb);
|
||
// 停止发送ps-rtp
|
||
bool stopSendRtp();
|
||
|
||
////////////////static方法，查找或生成MediaSource////////////////
|
||
|
||
// 同步查找流
|
||
static Ptr find(const string &schema, const string &vhost, const string &app, const string &id);
|
||
|
||
// 忽略类型，同步查找流，可能返回rtmp/rtsp/hls类型
|
||
static Ptr find(const string &vhost, const string &app, const string &stream_id);
|
||
|
||
// 异步查找流
|
||
static void findAsync(const MediaInfo &info, const std::shared_ptr<TcpSession> &session, const function<void(const Ptr &src)> &cb);
|
||
// 遍历所有流
|
||
static void for_each_media(const function<void(const Ptr &src)> &cb);
|
||
// 从mp4文件生成MediaSource
|
||
static MediaSource::Ptr createFromMP4(const string &schema, const string &vhost, const string &app, const string &stream, const string &file_path = "", bool check_app = true);
|
||
|
||
protected:
|
||
//媒体注册
|
||
void regist();
|
||
|
||
private:
|
||
//媒体注销
|
||
bool unregist();
|
||
//触发媒体事件
|
||
void emitEvent(bool regist);
|
||
|
||
protected:
|
||
BytesSpeed _speed;
|
||
|
||
private:
|
||
string _schema;
|
||
string _vhost;
|
||
string _app;
|
||
string _stream_id;
|
||
std::weak_ptr<MediaSourceEvent> _listener;
|
||
};
|
||
|
||
///缓存刷新策略类
|
||
class FlushPolicy {
|
||
public:
|
||
FlushPolicy() = default;
|
||
~FlushPolicy() = default;
|
||
|
||
uint32_t getStamp(const RtpPacket::Ptr &packet) {
|
||
return packet->timeStamp;
|
||
}
|
||
|
||
uint32_t getStamp(const RtmpPacket::Ptr &packet) {
|
||
return packet->time_stamp;
|
||
}
|
||
|
||
bool isFlushAble(bool is_video, bool is_key, uint32_t new_stamp, int cache_size);
|
||
|
||
private:
|
||
uint32_t _last_stamp[2] = {0, 0};
|
||
};
|
||
|
||
/// 合并写缓存模板
|
||
/// \tparam packet 包类型
|
||
/// \tparam policy 刷新缓存策略
|
||
/// \tparam packet_list 包缓存类型
|
||
template<typename packet, typename policy = FlushPolicy, typename packet_list = List<std::shared_ptr<packet> > >
|
||
class PacketCache {
|
||
public:
|
||
PacketCache(){
|
||
_cache = std::make_shared<packet_list>();
|
||
}
|
||
|
||
virtual ~PacketCache() = default;
|
||
|
||
void inputPacket(bool is_video, const std::shared_ptr<packet> &pkt, bool key_pos) {
|
||
if (_policy.isFlushAble(is_video, key_pos, _policy.getStamp(pkt), _cache->size())) {
|
||
flushAll();
|
||
}
|
||
|
||
//追加数据到最后
|
||
_cache->emplace_back(pkt);
|
||
if (key_pos) {
|
||
_key_pos = key_pos;
|
||
}
|
||
}
|
||
|
||
virtual void clearCache() {
|
||
_cache->clear();
|
||
}
|
||
|
||
virtual void onFlush(std::shared_ptr<packet_list> &, bool key_pos) = 0;
|
||
|
||
private:
|
||
void flushAll() {
|
||
if (_cache->empty()) {
|
||
return;
|
||
}
|
||
onFlush(_cache, _key_pos);
|
||
_cache = std::make_shared<packet_list>();
|
||
_key_pos = false;
|
||
}
|
||
|
||
private:
|
||
bool _key_pos = false;
|
||
policy _policy;
|
||
std::shared_ptr<packet_list> _cache;
|
||
};
|
||
|
||
} /* namespace mediakit */
|
||
#endif //ZLMEDIAKIT_MEDIASOURCE_H
|