amass/ZLMediaKit
1
0
Fork 0
mirror of https://github.com/ZLMediaKit/ZLMediaKit.git synced 2024-11-22 19:00:01 +08:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

Add optional customized_path parameter for startRecord

Browse Source
This commit is contained in:
Gemfield 2020-01-02 12:47:12 +08:00
parent 84de9720b5
commit e31c1ee207
7 changed files with 29 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/include/mk_recorder.h
View File

@ -83,7 +83,7 @@ API_EXPORT int API_CALL mk_recorder_status(int type, const char *vhost, const ch
* @param continue_record 流注销时是否继续等待录制还是立即停止录制 * @param continue_record 流注销时是否继续等待录制还是立即停止录制
* @return 0代表成功，负数代表失败 * @return 0代表成功，负数代表失败
*/ */
API_EXPORT int API_CALL mk_recorder_start(int type, const char *vhost, const char *app, const char *stream,int wait_for_record, int continue_record); API_EXPORT int API_CALL mk_recorder_start(int type, const char *vhost, const char *app, const char *stream, const char *customized_path, int wait_for_record, int continue_record);
/** /**
* 停止录制 * 停止录制

4
api/source/mk_recorder.cpp
View File

@ -57,9 +57,9 @@ API_EXPORT int API_CALL mk_recorder_status(int type, const char *vhost, const ch
return Recorder::getRecordStatus((Recorder::type)type,vhost,app,stream); return Recorder::getRecordStatus((Recorder::type)type,vhost,app,stream);
} }
API_EXPORT int API_CALL mk_recorder_start(int type, const char *vhost, const char *app, const char *stream,int wait_for_record, int continue_record){ API_EXPORT int API_CALL mk_recorder_start(int type, const char *vhost, const char *app, const char *stream,const char *customized_path,int wait_for_record, int continue_record){
assert(vhost && app && stream); assert(vhost && app && stream);
return Recorder::startRecord((Recorder::type)type,vhost,app,stream,wait_for_record,continue_record); return Recorder::startRecord((Recorder::type)type,vhost,app,stream,customized_path,wait_for_record,continue_record);
} }
API_EXPORT int API_CALL mk_recorder_stop(int type, const char *vhost, const char *app, const char *stream){ API_EXPORT int API_CALL mk_recorder_stop(int type, const char *vhost, const char *app, const char *stream){

2
server/WebApi.cpp
View File

@ -745,10 +745,12 @@ void installWebApi() {
API_REGIST(api,startRecord,{ API_REGIST(api,startRecord,{
CHECK_SECRET(); CHECK_SECRET();
CHECK_ARGS("type","vhost","app","stream","wait_for_record","continue_record"); CHECK_ARGS("type","vhost","app","stream","wait_for_record","continue_record");
int result = Recorder::startRecord((Recorder::type)allArgs["type"].as<int>(), int result = Recorder::startRecord((Recorder::type)allArgs["type"].as<int>(),
allArgs["vhost"], allArgs["vhost"],
allArgs["app"], allArgs["app"],
allArgs["stream"], allArgs["stream"],
allArgs["customized_path"],
allArgs["wait_for_record"], allArgs["wait_for_record"],
allArgs["continue_record"]); allArgs["continue_record"]);
val["result"] = result; val["result"] = result;

4
src/Common/MultiMediaSourceMuxer.h
View File

@ -53,11 +53,11 @@ public:
} }
if(enable_hls){ if(enable_hls){
Recorder::startRecord(Recorder::type_hls,vhost, app, stream, true, false); Recorder::startRecord(Recorder::type_hls,vhost, app, stream, "", true, false);
} }
if(enable_mp4){ if(enable_mp4){
Recorder::startRecord(Recorder::type_mp4,vhost, app, stream, true, false); Recorder::startRecord(Recorder::type_mp4,vhost, app, stream, "", true, false);
} }
_get_hls_media_source = [vhost,app,stream](){ _get_hls_media_source = [vhost,app,stream](){

5
src/Record/HlsMakerImp.cpp
View File

@ -58,8 +58,9 @@ string HlsMakerImp::onOpenSegment(int index) {
string segment_name , segment_path; string segment_name , segment_path;
{ {
auto strDate = getTimeStr("%Y-%m-%d"); auto strDate = getTimeStr("%Y-%m-%d");
auto strTime = getTimeStr("%H-%M-%S"); auto strHour = getTimeStr("%H");
segment_name = StrPrinter << strDate + "/" + strTime << "_" << index << ".ts"; auto strTime = getTimeStr("%M-%S");
segment_name = StrPrinter << strDate + "/" + strHour + "/" + strTime << "_" << index << ".ts";
segment_path = _path_prefix + "/" + segment_name; segment_path = _path_prefix + "/" + segment_name;
if(isLive()){ if(isLive()){
_segment_file_paths.emplace(index,segment_path); _segment_file_paths.emplace(index,segment_path);

28
src/Record/Recorder.cpp
View File

@ -34,7 +34,7 @@ using namespace toolkit;
namespace mediakit { namespace mediakit {
MediaSinkInterface *createHlsRecorder(const string &strVhost_tmp, const string &strApp, const string &strId) { MediaSinkInterface *createHlsRecorder(const string &strVhost_tmp, const string &strApp, const string &strId, const string &customized_path) {
#if defined(ENABLE_HLS) #if defined(ENABLE_HLS)
GET_CONFIG(bool, enableVhost, General::kEnableVhost); GET_CONFIG(bool, enableVhost, General::kEnableVhost);
GET_CONFIG(string, hlsPath, Hls::kFilePath); GET_CONFIG(string, hlsPath, Hls::kFilePath);
@ -53,6 +53,10 @@ MediaSinkInterface *createHlsRecorder(const string &strVhost_tmp, const string &
} else { } else {
m3u8FilePath = strApp + "/" + strId + "/hls.m3u8"; m3u8FilePath = strApp + "/" + strId + "/hls.m3u8";
} }
//Here we use the customized file path.
if(!customized_path.empty()){
m3u8FilePath = customized_path + "/hls.m3u8";
}
m3u8FilePath = File::absolutePath(m3u8FilePath, hlsPath); m3u8FilePath = File::absolutePath(m3u8FilePath, hlsPath);
auto ret = new HlsRecorder(m3u8FilePath, params); auto ret = new HlsRecorder(m3u8FilePath, params);
ret->setMediaSource(strVhost, strApp, strId); ret->setMediaSource(strVhost, strApp, strId);
@ -62,7 +66,7 @@ MediaSinkInterface *createHlsRecorder(const string &strVhost_tmp, const string &
#endif //defined(ENABLE_HLS) #endif //defined(ENABLE_HLS)
} }
MediaSinkInterface *createMP4Recorder(const string &strVhost_tmp, const string &strApp, const string &strId) { MediaSinkInterface *createMP4Recorder(const string &strVhost_tmp, const string &strApp, const string &strId, const string &customized_path) {
#if defined(ENABLE_MP4RECORD) #if defined(ENABLE_MP4RECORD)
GET_CONFIG(bool, enableVhost, General::kEnableVhost); GET_CONFIG(bool, enableVhost, General::kEnableVhost);
GET_CONFIG(string, recordPath, Record::kFilePath); GET_CONFIG(string, recordPath, Record::kFilePath);
@ -80,6 +84,10 @@ MediaSinkInterface *createMP4Recorder(const string &strVhost_tmp, const string &
} else { } else {
mp4FilePath = recordAppName + "/" + strApp + "/" + strId + "/"; mp4FilePath = recordAppName + "/" + strApp + "/" + strId + "/";
} }
//Here we use the customized file path.
if(!customized_path.empty()){
mp4FilePath = customized_path + "/";
}
mp4FilePath = File::absolutePath(mp4FilePath, recordPath); mp4FilePath = File::absolutePath(mp4FilePath, recordPath);
return new MP4Recorder(mp4FilePath, strVhost, strApp, strId); return new MP4Recorder(mp4FilePath, strVhost, strApp, strId);
#else #else
@ -193,7 +201,7 @@ public:
return it->second->getRecorder(); return it->second->getRecorder();
} }
int startRecord(const string &vhost, const string &app, const string &stream_id, bool waitForRecord, bool continueRecord) { int startRecord(const string &vhost, const string &app, const string &stream_id, const string &customized_path, bool waitForRecord, bool continueRecord) {
auto key = getRecorderKey(vhost, app, stream_id); auto key = getRecorderKey(vhost, app, stream_id);
lock_guard<decltype(_recorder_mtx)> lck(_recorder_mtx); lock_guard<decltype(_recorder_mtx)> lck(_recorder_mtx);
if (getRecordStatus_l(key) != Recorder::status_not_record) { if (getRecordStatus_l(key) != Recorder::status_not_record) {
@ -207,7 +215,7 @@ public:
return -1; return -1;
} }
auto recorder = MediaSinkInterface::Ptr(createRecorder(vhost, app, stream_id)); auto recorder = MediaSinkInterface::Ptr(createRecorder(vhost, app, stream_id, customized_path));
if (!recorder) { if (!recorder) {
// 创建录制器失败 // 创建录制器失败
return -2; return -2;
@ -326,14 +334,14 @@ private:
return vhost + "/" + app + "/" + stream_id; return vhost + "/" + app + "/" + stream_id;
} }
MediaSinkInterface *createRecorder(const string &vhost, const string &app, const string &stream_id) { MediaSinkInterface *createRecorder(const string &vhost, const string &app, const string &stream_id, const string &customized_path) {
MediaSinkInterface *ret = nullptr; MediaSinkInterface *ret = nullptr;
switch (type) { switch (type) {
case Recorder::type_hls: case Recorder::type_hls:
ret = createHlsRecorder(vhost, app, stream_id); ret = createHlsRecorder(vhost, app, stream_id,customized_path);
break; break;
case Recorder::type_mp4: case Recorder::type_mp4:
ret = createMP4Recorder(vhost, app, stream_id); ret = createMP4Recorder(vhost, app, stream_id,customized_path);
break; break;
default: default:
break; break;
@ -385,12 +393,12 @@ std::shared_ptr<MediaSinkInterface> Recorder::getRecorder(type type, const strin
} }
int Recorder::startRecord(Recorder::type type, const string &vhost, const string &app, const string &stream_id, bool waitForRecord, bool continueRecord) { int Recorder::startRecord(Recorder::type type, const string &vhost, const string &app, const string &stream_id, const string &customized_path, bool waitForRecord, bool continueRecord) {
switch (type){ switch (type){
case type_mp4: case type_mp4:
return MediaSourceWatcher<type_mp4>::Instance().startRecord(vhost,app,stream_id,waitForRecord,continueRecord); return MediaSourceWatcher<type_mp4>::Instance().startRecord(vhost,app,stream_id,customized_path,waitForRecord,continueRecord);
case type_hls: case type_hls:
return MediaSourceWatcher<type_hls>::Instance().startRecord(vhost,app,stream_id,waitForRecord,continueRecord); return MediaSourceWatcher<type_hls>::Instance().startRecord(vhost,app,stream_id,customized_path,waitForRecord,continueRecord);
} }
WarnL << "unknown record type: " << type; WarnL << "unknown record type: " << type;
return -3; return -3;

2
src/Record/Recorder.h
View File

@ -73,7 +73,7 @@ public:
* @param continueRecord 流注销时是否继续等待录制还是立即停止录制 * @param continueRecord 流注销时是否继续等待录制还是立即停止录制
* @return 0代表成功，负数代表失败 * @return 0代表成功，负数代表失败
*/ */
static int startRecord(type type, const string &vhost, const string &app, const string &stream_id,bool waitForRecord, bool continueRecord); static int startRecord(type type, const string &vhost, const string &app, const string &stream_id,const string &customized_path,bool waitForRecord, bool continueRecord);
/** /**
* 停止录制 * 停止录制