76 lines
2.2 KiB
C++
76 lines
2.2 KiB
C++
// rtsp_manager.hpp
|
|
#pragma once
|
|
|
|
#include <gst/gst.h>
|
|
#include <unordered_map>
|
|
#include <string>
|
|
#include <mutex>
|
|
#include <thread>
|
|
#include <atomic>
|
|
#include <functional>
|
|
#include <memory>
|
|
#include <condition_variable>
|
|
#include <queue>
|
|
#include "app_config.hpp"
|
|
|
|
class RTMPManager
|
|
{
|
|
public:
|
|
enum class StreamResult
|
|
{
|
|
OK,
|
|
PIPELINE_ERROR,
|
|
CONNECTION_FAIL,
|
|
EOS_RECEIVED,
|
|
UNKNOWN
|
|
};
|
|
|
|
struct StreamStatus
|
|
{
|
|
bool running;
|
|
StreamResult last_result;
|
|
std::string last_error;
|
|
};
|
|
|
|
using StreamCallback = std::function<void(const std::string &key, StreamStatus status)>;
|
|
|
|
static void init();
|
|
static void start_camera(const Camera &cam, StreamType type);
|
|
static void stop_camera(const std::string &cam_name, StreamType type);
|
|
static void stop_all();
|
|
static bool is_streaming(const std::string &cam_name, StreamType type);
|
|
static bool is_any_streaming();
|
|
static std::string get_stream_url(const std::string &cam_name, StreamType type);
|
|
|
|
static void set_status_callback(StreamCallback cb);
|
|
|
|
private:
|
|
struct StreamContext
|
|
{
|
|
std::atomic<bool> running{false};
|
|
std::thread thread;
|
|
StreamStatus status;
|
|
StreamContext() = default;
|
|
// non-copyable
|
|
StreamContext(const StreamContext &) = delete;
|
|
StreamContext &operator=(const StreamContext &) = delete;
|
|
// movable? we avoid moving by using unique_ptr
|
|
};
|
|
|
|
// RTMPManager 内部增加
|
|
static std::mutex stop_queue_mutex;
|
|
static std::condition_variable stop_cv;
|
|
static std::queue<std::unique_ptr<StreamContext>> stop_queue;
|
|
static std::atomic<bool> stop_thread_running{false};
|
|
static std::thread stop_thread;
|
|
// store unique_ptr to avoid copy/move issues with atomic/thread
|
|
static std::unordered_map<std::string, std::unique_ptr<StreamContext>> streams;
|
|
static std::mutex streams_mutex;
|
|
static StreamCallback status_callback;
|
|
|
|
static void stream_loop(Camera cam, StreamType type);
|
|
static GstElement *create_pipeline(const Camera &cam, StreamType type);
|
|
static std::string make_stream_key(const std::string &cam_name, StreamType type);
|
|
static void update_status(const std::string &key, const StreamStatus &status);
|
|
};
|