54 lines
1.3 KiB
C++
54 lines
1.3 KiB
C++
// rtmp_manager.hpp
|
|
#pragma once
|
|
|
|
#include <atomic>
|
|
#include <condition_variable>
|
|
#include <future>
|
|
#include <memory>
|
|
#include <mutex>
|
|
#include <queue>
|
|
#include <string>
|
|
#include <thread>
|
|
#include <unordered_map>
|
|
#include <vector>
|
|
|
|
#include "app_config.hpp"
|
|
#include "mqtt_client_wrapper.hpp"
|
|
|
|
class RTMPManager
|
|
{
|
|
public:
|
|
struct StreamResultInfo
|
|
{
|
|
int loc{-1};
|
|
std::string url;
|
|
int result{1}; // 0=成功, 1=失败
|
|
std::string reason;
|
|
};
|
|
|
|
static void init();
|
|
static void enqueue_video_push_request(const VideoPushRequest &req);
|
|
static void stop_all();
|
|
|
|
private:
|
|
struct StreamContext
|
|
{
|
|
std::atomic<bool> running{false};
|
|
std::thread thread;
|
|
std::promise<StreamResultInfo> start_result;
|
|
};
|
|
|
|
static void rtmp_worker_thread();
|
|
static StreamResultInfo start_camera(const Camera &cam, int streamType);
|
|
static StreamResultInfo stop_camera(const Camera &cam, int streamType);
|
|
|
|
static std::unordered_map<std::string, std::unique_ptr<StreamContext>> streams;
|
|
static std::mutex streams_mutex;
|
|
|
|
static std::queue<VideoPushRequest> request_queue;
|
|
static std::mutex queue_mutex;
|
|
static std::condition_variable queue_cv;
|
|
|
|
static std::thread worker_thread;
|
|
static std::atomic<bool> running;
|
|
}; |