generated from zhangwei/Matrixai
105 lines
2.5 KiB
C++
105 lines
2.5 KiB
C++
#include "RtspServer.h"
|
|
#include "RtspConnection.h"
|
|
#include "SocketUtil.h"
|
|
#include "Logger.h"
|
|
|
|
using namespace xop;
|
|
using namespace std;
|
|
|
|
RtspServer::RtspServer(EventLoop* loop)
|
|
: TcpServer(loop)
|
|
{
|
|
|
|
}
|
|
|
|
RtspServer::~RtspServer()
|
|
{
|
|
|
|
}
|
|
|
|
std::shared_ptr<RtspServer> RtspServer::Create(xop::EventLoop* loop)
|
|
{
|
|
std::shared_ptr<RtspServer> server(new RtspServer(loop));
|
|
return server;
|
|
}
|
|
|
|
MediaSessionId RtspServer::AddSession(MediaSession* session)
|
|
{
|
|
std::lock_guard<std::mutex> locker(mutex_);
|
|
|
|
if (rtsp_suffix_map_.find(session->GetRtspUrlSuffix()) != rtsp_suffix_map_.end()) {
|
|
return 0;
|
|
}
|
|
|
|
std::shared_ptr<MediaSession> media_session(session);
|
|
MediaSessionId sessionId = media_session->GetMediaSessionId();
|
|
rtsp_suffix_map_.emplace(std::move(media_session->GetRtspUrlSuffix()), sessionId);
|
|
media_sessions_.emplace(sessionId, std::move(media_session));
|
|
|
|
return sessionId;
|
|
}
|
|
|
|
void RtspServer::RemoveSession(MediaSessionId sessionId)
|
|
{
|
|
std::lock_guard<std::mutex> locker(mutex_);
|
|
|
|
auto iter = media_sessions_.find(sessionId);
|
|
if(iter != media_sessions_.end()) {
|
|
rtsp_suffix_map_.erase(iter->second->GetRtspUrlSuffix());
|
|
media_sessions_.erase(sessionId);
|
|
}
|
|
}
|
|
|
|
MediaSession::Ptr RtspServer::LookMediaSession(const std::string& suffix)
|
|
{
|
|
std::lock_guard<std::mutex> locker(mutex_);
|
|
|
|
auto iter = rtsp_suffix_map_.find(suffix);
|
|
if(iter != rtsp_suffix_map_.end()) {
|
|
MediaSessionId id = iter->second;
|
|
return media_sessions_[id];
|
|
}
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
MediaSession::Ptr RtspServer::LookMediaSession(MediaSessionId session_Id)
|
|
{
|
|
std::lock_guard<std::mutex> locker(mutex_);
|
|
|
|
auto iter = media_sessions_.find(session_Id);
|
|
if(iter != media_sessions_.end()) {
|
|
return iter->second;
|
|
}
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
bool RtspServer::PushFrame(MediaSessionId session_id, MediaChannelId channel_id, AVFrame frame)
|
|
{
|
|
std::shared_ptr<MediaSession> sessionPtr = nullptr;
|
|
|
|
{
|
|
std::lock_guard<std::mutex> locker(mutex_);
|
|
auto iter = media_sessions_.find(session_id);
|
|
if (iter != media_sessions_.end()) {
|
|
sessionPtr = iter->second;
|
|
}
|
|
else {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
if (sessionPtr!=nullptr && sessionPtr->GetNumClient()!=0) {
|
|
return sessionPtr->HandleFrame(channel_id, frame);
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
TcpConnection::Ptr RtspServer::OnConnect(SOCKET sockfd)
|
|
{
|
|
return std::make_shared<RtspConnection>(shared_from_this(), event_loop_->GetTaskScheduler().get(), sockfd);
|
|
}
|
|
|