From 4132df89cdab9b068bb374c8e5781afb87960b75 Mon Sep 17 00:00:00 2001
From: qiuzhouwei <chatop@qq.com>
Date: Mon, 28 Sep 2020 10:45:59 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E4=BA=86=E6=9C=8D=E5=8A=A1?=
 =?UTF-8?q?=E5=99=A8id,=E4=BF=AE=E6=94=B9=E4=BA=86record.cpp=E4=B8=ADmp4?=
 =?UTF-8?q?=E5=BD=95=E5=88=B6=E7=9A=84=E8=87=AA=E5=AE=9A=E4=B9=89=E5=BD=95?=
 =?UTF-8?q?=E5=88=B6=E8=B7=AF=E5=BE=84=EF=BC=8Cffmpegsource=E4=B8=AD?=
 =?UTF-8?q?=E7=9A=84=E8=B6=85=E6=97=B6=E6=97=B6=E9=97=B4=EF=BC=8Cwebhook?=
 =?UTF-8?q?=E4=B8=AD=E7=9A=84=E5=94=AF=E4=B8=80id=E7=AD=89?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 server/FFmpegSource.cpp |  3 ++-
 server/WebHook.cpp      | 12 ++++++++++++
 src/Common/config.cpp   | 21 +++++++++++++++++++++
 src/Common/config.h     |  2 ++
 src/Record/Recorder.cpp |  9 ++++++++-
 5 files changed, 45 insertions(+), 2 deletions(-)

diff --git a/server/FFmpegSource.cpp b/server/FFmpegSource.cpp
index 21e87b41..4c3a1df2 100644
--- a/server/FFmpegSource.cpp
+++ b/server/FFmpegSource.cpp
@@ -198,7 +198,8 @@ void FFmpegSource::startTimer(int timeout_ms) {
                 //同步查找流
                 if (!src) {
                     //流不在线，重新拉流
-                    if(strongSelf->_replay_ticker.elapsedTime() > 10 * 1000){
+					//@子悦，这里原先是10秒超时，实际发现10秒不够，我改成20秒了
+                    if(strongSelf->_replay_ticker.elapsedTime() > 20 * 1000){
                         //上次重试时间超过10秒，那么再重试FFmpeg拉流
                         strongSelf->_replay_ticker.resetTime();
                         strongSelf->play(strongSelf->_src_url, strongSelf->_dst_url, timeout_ms, [](const SockException &) {});
diff --git a/server/WebHook.cpp b/server/WebHook.cpp
index e2d1adf1..9f359d20 100644
--- a/server/WebHook.cpp
+++ b/server/WebHook.cpp
@@ -158,6 +158,8 @@ static void do_http_hook(const string &url,const ArgsType &body,const function<v
 
 static ArgsType make_json(const MediaInfo &args){
     ArgsType body;
+    GET_CONFIG(string,mediaServerId,General::kMediaServerId);
+    body["mediaserverid"] =  mediaServerId;
     body["schema"] = args._schema;
     body["vhost"] = args._vhost;
     body["app"] = args._app;
@@ -319,6 +321,8 @@ void installWebHook(){
             return;
         }
         ArgsType body;
+        GET_CONFIG(string,mediaServerId,General::kMediaServerId);
+        body["mediaserverid"] =  mediaServerId;
         body["regist"] = bRegist;
         body["schema"] = sender.getSchema();
         body["vhost"] = sender.getVhost();
@@ -344,6 +348,8 @@ void installWebHook(){
 
     static auto getRecordInfo = [](const RecordInfo &info) {
         ArgsType body;
+		GET_CONFIG(string,mediaServerId,General::kMediaServerId);
+    	body["mediaserverid"] =  mediaServerId;
         body["start_time"] = (Json::UInt64) info.start_time;
         body["file_size"] = (Json::UInt64) info.file_size;
         body["time_len"] = info.time_len;
@@ -382,6 +388,8 @@ void installWebHook(){
             return;
         }
         ArgsType body;
+        GET_CONFIG(string,mediaServerId,General::kMediaServerId);
+        body["mediaserverid"] =  mediaServerId;
         body["ip"] = sender.get_peer_ip();
         body["port"] = sender.get_peer_port();
         body["id"] = sender.getIdentifier();
@@ -400,6 +408,8 @@ void installWebHook(){
         }
 
         ArgsType body;
+        GET_CONFIG(string,mediaServerId,General::kMediaServerId);
+        body["mediaserverid"] =  mediaServerId;
         body["schema"] = sender.getSchema();
         body["vhost"] = sender.getVhost();
         body["app"] = sender.getApp();
@@ -445,6 +455,8 @@ void installWebHook(){
         }
 
         ArgsType body;
+        GET_CONFIG(string,mediaServerId,General::kMediaServerId);
+        body["mediaserverid"] =  mediaServerId;
         body["ip"] = sender.get_peer_ip();
         body["port"] = sender.get_peer_port();
         body["id"] = sender.getIdentifier();
diff --git a/src/Common/config.cpp b/src/Common/config.cpp
index f26c504d..b94dbbed 100644
--- a/src/Common/config.cpp
+++ b/src/Common/config.cpp
@@ -18,6 +18,25 @@
 using namespace toolkit;
 
 namespace mediakit {
+    string generalGuid() {
+        srand(time(0));
+        std::string random_str("");
+        for (int i = 0; i < 6; ++i) {
+            for (int j = 0; j < 8; j++)
+                switch (rand() % 2) {
+                    case 1:
+                        random_str += ('A' + rand() % 26);
+                        break;
+                    default:
+                        random_str += ('0' + rand() % 10);
+                        break;
+                }
+            if (i < 5)
+                random_str += "-";
+        }
+
+        return random_str;
+    }
 
 bool loadIniConfig(const char *ini_path){
     string ini;
@@ -58,6 +77,7 @@ const string kBroadcastHttpBeforeAccess = "kBroadcastHttpBeforeAccess";
 //通用配置项目
 namespace General{
 #define GENERAL_FIELD "general."
+const string kMediaServerId = GENERAL_FIELD"mediaServerId";
 const string kFlowThreshold = GENERAL_FIELD"flowThreshold";
 const string kStreamNoneReaderDelayMS = GENERAL_FIELD"streamNoneReaderDelayMS";
 const string kMaxStreamWaitTimeMS = GENERAL_FIELD"maxStreamWaitMS";
@@ -80,6 +100,7 @@ onceToken token([](){
     mINI::Instance()[kPublishToMP4] = 0;
     mINI::Instance()[kMergeWriteMS] = 0;
     mINI::Instance()[kModifyStamp] = 0;
+    mINI::Instance()[kMediaServerId] = generalGuid();
 },nullptr);
 
 }//namespace General
diff --git a/src/Common/config.h b/src/Common/config.h
index c352b1f7..bc11d9b6 100644
--- a/src/Common/config.h
+++ b/src/Common/config.h
@@ -154,6 +154,8 @@ extern const string kBroadcastReloadConfig;
 
 ////////////通用配置///////////
 namespace General{
+//每个流媒体服务器的ID（GUID）
+extern const string kMediaServerId;
 //流量汇报事件流量阈值,单位KB，默认1MB
 extern const string kFlowThreshold;
 //流无人观看并且超过若干时间后才触发kBroadcastStreamNoneReader事件
diff --git a/src/Record/Recorder.cpp b/src/Record/Recorder.cpp
index 100a2951..9f6300f6 100644
--- a/src/Record/Recorder.cpp
+++ b/src/Record/Recorder.cpp
@@ -46,8 +46,15 @@ string Recorder::getRecordPath(Recorder::type type, const string &vhost, const s
             }
             //Here we use the customized file path.
             if (!customized_path.empty()) {
-                mp4FilePath = customized_path + "/";
+                /*开始删除*/
+                // mp4FilePath = customized_path + "/";
+                /*删除结束*/
+                /*开始添加*/
+				//@子悦，你上次说这里为了安全不能跳出目录，但实际操作过程中因为存储挂载位置不由流媒体决定，为了方便保存到挂载存储上，我这边做成可以跳出自已目录，你看是否合适，或者有其他办法可以处理这事
+                return  customized_path + "/"+mp4FilePath;
+                /*开始添加*/
             }
+
             return File::absolutePath(mp4FilePath, recordPath);
         }
         default: