mirror of
https://github.com/ZLMediaKit/ZLMediaKit.git
synced 2024-11-26 12:37:09 +08:00
132 lines
3.3 KiB
C++
132 lines
3.3 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.
|
||
*/
|
||
|
||
#ifndef SRC_PUSHER_PUSHERBASE_H_
|
||
#define SRC_PUSHER_PUSHERBASE_H_
|
||
|
||
#include <map>
|
||
#include <memory>
|
||
#include <string>
|
||
#include <functional>
|
||
#include "Network/Socket.h"
|
||
#include "Util/mini.h"
|
||
#include "Common/MediaSource.h"
|
||
|
||
namespace mediakit {
|
||
|
||
class PusherBase : public toolkit::mINI {
|
||
public:
|
||
using Ptr = std::shared_ptr<PusherBase>;
|
||
using Event = std::function<void(const toolkit::SockException &ex)>;
|
||
|
||
static Ptr createPusher(const toolkit::EventPoller::Ptr &poller,
|
||
const MediaSource::Ptr &src,
|
||
const std::string &strUrl);
|
||
|
||
PusherBase();
|
||
virtual ~PusherBase() = default;
|
||
|
||
/**
|
||
* 开始推流
|
||
* @param strUrl 视频url，支持rtsp/rtmp
|
||
*/
|
||
virtual void publish(const std::string &strUrl) {};
|
||
|
||
/**
|
||
* 中断推流
|
||
*/
|
||
virtual void teardown() {};
|
||
|
||
/**
|
||
* 摄像推流结果回调
|
||
*/
|
||
virtual void setOnPublished(const Event &cb) = 0;
|
||
|
||
/**
|
||
* 设置断开回调
|
||
*/
|
||
virtual void setOnShutdown(const Event &cb) = 0;
|
||
|
||
protected:
|
||
virtual void onShutdown(const toolkit::SockException &ex) = 0;
|
||
virtual void onPublishResult(const toolkit::SockException &ex) = 0;
|
||
};
|
||
|
||
template<typename Parent, typename Delegate>
|
||
class PusherImp : public Parent {
|
||
public:
|
||
using Ptr = std::shared_ptr<PusherImp>;
|
||
|
||
template<typename ...ArgsType>
|
||
PusherImp(ArgsType &&...args) : Parent(std::forward<ArgsType>(args)...) {}
|
||
|
||
/**
|
||
* 开始推流
|
||
* @param url 推流url，支持rtsp/rtmp
|
||
*/
|
||
void publish(const std::string &url) override {
|
||
return _delegate ? _delegate->publish(url) : Parent::publish(url);
|
||
}
|
||
|
||
/**
|
||
* 中断推流
|
||
*/
|
||
void teardown() override {
|
||
return _delegate ? _delegate->teardown() : Parent::teardown();
|
||
}
|
||
|
||
std::shared_ptr<toolkit::SockInfo> getSockInfo() const {
|
||
return std::dynamic_pointer_cast<toolkit::SockInfo>(_delegate);
|
||
}
|
||
|
||
/**
|
||
* 摄像推流结果回调
|
||
*/
|
||
void setOnPublished(const PusherBase::Event &cb) override {
|
||
if (_delegate) {
|
||
_delegate->setOnPublished(cb);
|
||
}
|
||
_on_publish = cb;
|
||
}
|
||
|
||
/**
|
||
* 设置断开回调
|
||
*/
|
||
void setOnShutdown(const PusherBase::Event &cb) override {
|
||
if (_delegate) {
|
||
_delegate->setOnShutdown(cb);
|
||
}
|
||
_on_shutdown = cb;
|
||
}
|
||
|
||
protected:
|
||
void onShutdown(const toolkit::SockException &ex) override {
|
||
if (_on_shutdown) {
|
||
_on_shutdown(ex);
|
||
_on_shutdown = nullptr;
|
||
}
|
||
}
|
||
|
||
void onPublishResult(const toolkit::SockException &ex) override {
|
||
if (_on_publish) {
|
||
_on_publish(ex);
|
||
_on_publish = nullptr;
|
||
}
|
||
}
|
||
|
||
protected:
|
||
PusherBase::Event _on_shutdown;
|
||
PusherBase::Event _on_publish;
|
||
std::shared_ptr<Delegate> _delegate;
|
||
};
|
||
|
||
} /* namespace mediakit */
|
||
#endif /* SRC_PUSHER_PUSHERBASE_H_ */
|