From f93ee1a42e8c47e472332287b7350b66a6b0fa11 Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期五, 24 七月 2020 18:28:57 +0800
Subject: [PATCH] 保存触发id之后的视频作为触发视频

---
 csrc/buz/recorder.cpp |   24 +++++++++++++++++++++---
 1 files changed, 21 insertions(+), 3 deletions(-)

diff --git a/csrc/buz/recorder.cpp b/csrc/buz/recorder.cpp
index 2d00c57..beda224 100644
--- a/csrc/buz/recorder.cpp
+++ b/csrc/buz/recorder.cpp
@@ -5,6 +5,8 @@
 #include <unistd.h>
 #include <chrono>
 
+#include <sys/stat.h>
+
 extern "C"{
 #include <libavcodec/avcodec.h>
 }
@@ -82,6 +84,12 @@
             std::string filename(sole::uuid4().base62() + "-" + std::to_string(pid) + ".mp4");
             file_path_ = dir_ + "/" + filename;
 
+            std::string backup_dir("./video");
+            size_t pos = dir_.rfind("/");
+            if (pos != std::string::npos){
+                backup_dir = dir_.substr(0, pos);
+            }
+
             auto v = in_->getStream(AVMEDIA_TYPE_VIDEO);
             if (!v){
                 return -2;
@@ -95,9 +103,11 @@
                 logIt("start record h264 file: %s", file_path_.c_str());                
                 return 0;
             }else{
-                file_path_ = "./" + filename;
-                ret = out_->JustWriter(v, a, file_path_.c_str());
+                mkdir(backup_dir.c_str(), 0777);
+                file_path_ = backup_dir + "/" + filename;
+
                 logIt("failed in dir %s, try file %s to start record file", dir_.c_str(), file_path_.c_str());                
+                ret = out_->JustWriter(v, a, file_path_.c_str());
                 if (ret){
                     logIt("start record h264 file: %s", file_path_.c_str());                
                     return 0;
@@ -118,9 +128,17 @@
             std::string filename(sole::uuid4().base62() + "-" + std::to_string(pid) + ".hevc");
             file_path_ = dir_ + "/" + filename;
 
+            std::string backup_dir("./video");
+            size_t pos = dir_.rfind("/");
+            if (pos != std::string::npos){
+                backup_dir = dir_.substr(0, pos);
+            }
+            
             fp_ = fopen(file_path_.c_str(), "wb");
             if (!fp_){
-                file_path_ = "./" + filename;
+                mkdir(backup_dir.c_str(), 0777);
+                file_path_ = backup_dir + "/" + filename;
+
                 logIt("failed in dir %s, try file %s to start record hevc file", dir_.c_str(), file_path_.c_str());                
                 fp_ = fopen(file_path_.c_str(), "wb");
                 if (!fp_){

--
Gitblit v1.8.0