整理文件录制

src/Record/MP4Recorder.cpp

@@ -0,0 +1,166 @@
+/*
+ * MIT License
+ *
+ * Copyright (c) 2016-2019 xiongziliang <771730766@qq.com>
+ *
+ * This file is part of ZLMediaKit(https://github.com/xiongziliang/ZLMediaKit).
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to deal
+ * in the Software without restriction, including without limitation the rights
+ * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ * copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in all
+ * copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+ * SOFTWARE.
+ */
+
+#ifdef ENABLE_MP4RECORD
+#include <ctime>
+#include <sys/stat.h>
+#include "Common/config.h"
+#include "MP4Recorder.h"
+#include "Util/util.h"
+#include "Util/NoticeCenter.h"
+#include "Thread/WorkThreadPool.h"
+
+using namespace toolkit;
+
+namespace mediakit {
+
+string timeStr(const char *fmt) {
+	std::tm tm_snapshot;
+	auto time = ::time(NULL);
+#if defined(_WIN32)
+	localtime_s(&tm_snapshot, &time); // thread-safe
+#else
+	localtime_r(&time, &tm_snapshot); // POSIX
+#endif
+	const size_t size = 1024;
+	char buffer[size];
+	auto success = std::strftime(buffer, size, fmt, &tm_snapshot);
+	if (0 == success)
+		return string(fmt);
+	return buffer;
+}
+
+MP4Recorder::MP4Recorder(const string& strPath,
+				   const string &strVhost,
+				   const string &strApp,
+				   const string &strStreamId) {
+	_strPath = strPath;
+	/////record 业务逻辑//////
+	_info.strAppName = strApp;
+	_info.strStreamId = strStreamId;
+	_info.strVhost = strVhost;
+	_info.strFolder = strPath;
+}
+MP4Recorder::~MP4Recorder() {
+	closeFile();
+}
+
+void MP4Recorder::createFile() {
+	closeFile();
+	auto strDate = timeStr("%Y-%m-%d");
+	auto strTime = timeStr("%H-%M-%S");
+	auto strFileTmp = _strPath + strDate + "/." + strTime + ".mp4";
+	auto strFile =	_strPath + strDate + "/" + strTime + ".mp4";
+
+	/////record 业务逻辑//////
+	_info.ui64StartedTime = ::time(NULL);
+	_info.strFileName = strTime + ".mp4";
+	_info.strFilePath = strFile;
+    GET_CONFIG(string,appName,Record::kAppName);
+    _info.strUrl = appName + "/"
+                   + _info.strAppName + "/"
+                   + _info.strStreamId + "/"
+                   + strDate + "/"
+                   + strTime + ".mp4";
+
+	try {
+		_muxer = std::make_shared<MP4MuxerFile>(strFileTmp.data());
+		for(auto &track :_tracks){
+            //添加track
+            _muxer->addTrack(track);
+		}
+		_strFileTmp = strFileTmp;
+		_strFile = strFile;
+		_createFileTicker.resetTime();
+	}catch(std::exception &ex) {
+		WarnL << ex.what();
+	}
+}
+
+void MP4Recorder::asyncClose() {
+	auto muxer = _muxer;
+	auto strFileTmp = _strFileTmp;
+	auto strFile = _strFile;
+	auto info = _info;
+	WorkThreadPool::Instance().getExecutor()->async([muxer,strFileTmp,strFile,info]() {
+		//获取文件录制时间，放在关闭mp4之前是为了忽略关闭mp4执行时间
+		const_cast<Mp4Info&>(info).ui64TimeLen = ::time(NULL) - info.ui64StartedTime;
+		//关闭mp4非常耗时，所以要放在后台线程执行
+		const_cast<MP4MuxerFile::Ptr &>(muxer).reset();
+		//临时文件名改成正式文件名，防止mp4未完成时被访问
+		rename(strFileTmp.data(),strFile.data());
+		//获取文件大小
+		struct stat fileData;
+		stat(strFile.data(), &fileData);
+		const_cast<Mp4Info&>(info).ui64FileSize = fileData.st_size;
+		/////record 业务逻辑//////
+		NoticeCenter::Instance().emitEvent(Broadcast::kBroadcastRecordMP4,info);
+	});
+}
+
+void MP4Recorder::closeFile() {
+	if (_muxer) {
+		asyncClose();
+		_muxer = nullptr;
+	}
+}
+
+void MP4Recorder::inputFrame(const Frame::Ptr &frame) {
+	GET_CONFIG(uint32_t,recordSec,Record::kFileSecond);
+	if(!_muxer || ((_createFileTicker.elapsedTime() > recordSec * 1000) &&
+			      (!_haveVideo || (_haveVideo && frame->keyFrame()))) ){
+		//成立条件
+		//1、_muxer为空
+		//2、到了切片时间，并且只有音频
+		//3、到了切片时间，有视频并且遇到视频的关键帧
+		createFile();
+	}
+
+	if(_muxer){
+		//生成mp4文件
+		_muxer->inputFrame(frame);
+	}
+}
+
+void MP4Recorder::addTrack(const Track::Ptr & track){
+	//保存所有的track，为创建MP4MuxerFile做准备
+	_tracks.emplace_back(track);
+	if(track->getTrackType() == TrackVideo){
+		_haveVideo = true;
+	}
+}
+
+void MP4Recorder::resetTracks() {
+	closeFile();
+	_tracks.clear();
+	_haveVideo = false;
+	_createFileTicker.resetTime();
+}
+
+} /* namespace mediakit */
+
+
+#endif //ENABLE_MP4RECORD