Flv: Allow stream restart when closed

prsyahmi · Dec 5, 2020 · e06ae42 · e06ae42
1 parent fe2070c
commit e06ae42
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/v380/FlvStream.cpp b/v380/FlvStream.cpp
@@ -204,6 +204,7 @@ FlvStream::FlvStream()
 	, m_LastTimestamp(0)
 	, m_EnableVideo(false)
 	, m_EnableAudio(false)
+	, m_Inited(false)
 	, m_Exit(false)
 {
 	m_Thread = std::thread([&]() {
@@ -248,6 +249,11 @@ void FlvStream::Init(bool enableVideo, bool enableAudio)
 {
 	TFlvHeader header;
 
+	if (m_Inited) {
+		return;
+	}
+	m_Inited = true;
+
 	m_EnableVideo = enableVideo;
 	m_EnableAudio = enableAudio;
 

diff --git a/v380/FlvStream.h b/v380/FlvStream.h
@@ -24,6 +24,7 @@ class FlvStream
 	bool m_EnableVideo;
 	bool m_EnableAudio;
 
+	bool m_Inited;
 	bool m_Exit;
 	std::thread m_Thread;
 	Semaphore m_Semaphore;

diff --git a/v380/v380.cpp b/v380/v380.cpp
@@ -547,6 +547,8 @@ int main(int argc, const char* argv[])
 					socketStream.Send(ptzcmd);
 				}
 			}
+
+			fprintf(stderr, "Stream stopped, restarting stream\n");
 		}
 		catch (const std::exception& ex)
 		{