mirror of
https://gitee.com/xia-chu/ZLMediaKit.git
synced 2026-05-08 11:57:48 +08:00
90 lines
2.5 KiB
C++
90 lines
2.5 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.
|
||
*/
|
||
|
||
#ifndef ZLMEDIAKIT_RTPSELECTOR_H
|
||
#define ZLMEDIAKIT_RTPSELECTOR_H
|
||
|
||
#if defined(ENABLE_RTPPROXY)
|
||
#include <stdint.h>
|
||
#include <mutex>
|
||
#include <unordered_map>
|
||
#include "RtpProcess.h"
|
||
#include "Common/MediaSource.h"
|
||
|
||
namespace mediakit{
|
||
|
||
class RtpSelector;
|
||
class RtpProcessHelper : public MediaSourceEvent , public std::enable_shared_from_this<RtpProcessHelper> {
|
||
public:
|
||
using Ptr = std::shared_ptr<RtpProcessHelper>;
|
||
RtpProcessHelper(const std::string &stream_id, const std::weak_ptr<RtpSelector > &parent);
|
||
~RtpProcessHelper();
|
||
void attachEvent();
|
||
RtpProcess::Ptr & getProcess();
|
||
|
||
protected:
|
||
// 通知其停止推流
|
||
bool close(MediaSource &sender) override;
|
||
|
||
private:
|
||
std::string _stream_id;
|
||
RtpProcess::Ptr _process;
|
||
std::weak_ptr<RtpSelector> _parent;
|
||
};
|
||
|
||
class RtpSelector : public std::enable_shared_from_this<RtpSelector>{
|
||
public:
|
||
RtpSelector() = default;
|
||
~RtpSelector() = default;
|
||
|
||
class ProcessExisted : public std::runtime_error {
|
||
public:
|
||
template<typename ...T>
|
||
ProcessExisted(T && ...args) : std::runtime_error(std::forward<T>(args)...) {}
|
||
~ProcessExisted() override = default;
|
||
};
|
||
|
||
static bool getSSRC(const char *data,size_t data_len, uint32_t &ssrc);
|
||
static RtpSelector &Instance();
|
||
|
||
/**
|
||
* 清空所有对象
|
||
*/
|
||
void clear();
|
||
|
||
/**
|
||
* 获取一个rtp处理器
|
||
* @param stream_id 流id
|
||
* @param makeNew 不存在时是否新建, 该参数为true时，必须确保之前未创建同名对象
|
||
* @return rtp处理器
|
||
*/
|
||
RtpProcess::Ptr getProcess(const std::string &stream_id, bool makeNew);
|
||
|
||
/**
|
||
* 删除rtp处理器
|
||
* @param stream_id 流id
|
||
* @param ptr rtp处理器指针
|
||
*/
|
||
void delProcess(const std::string &stream_id, const RtpProcess *ptr);
|
||
|
||
private:
|
||
void onManager();
|
||
void createTimer();
|
||
|
||
private:
|
||
toolkit::Timer::Ptr _timer;
|
||
std::recursive_mutex _mtx_map;
|
||
std::unordered_map<std::string,RtpProcessHelper::Ptr> _map_rtp_process;
|
||
};
|
||
|
||
}//namespace mediakit
|
||
#endif//defined(ENABLE_RTPPROXY)
|
||
#endif //ZLMEDIAKIT_RTPSELECTOR_H
|