添加录像巡查超过保存时间的录像文件进行移除

648540858
1 parent 3c59c549
Showing 5 changed files with 161 additions and 35 deletions
src/main/java/top/panll/assist/config/TaskConfig.java
src/main/java/top/panll/assist/controller/RecordController.java
src/main/java/top/panll/assist/service/FFmpegExecUtils.java
src/main/java/top/panll/assist/service/FileManagerTimer.java
src/main/java/top/panll/assist/service/VideoFileService.java
@@ -31,13 +31,13 @@ public class TaskConfig {
     @Scheduled(cron = "0 0 0 * * ?")
     private void configureTasks() {
         logger.info("录像过期自检任务执行");
-        List<File> appList = videoFileService.getAppList();
+        List<File> appList = videoFileService.getAppList(false);
         Calendar calendar = Calendar.getInstance();
         calendar.add(Calendar.DATE, - 7);
         Date monday = calendar.getTime();
         if (appList != null && appList.size() > 0) {
             for (File appFile : appList) {
-                List<File> streamList = videoFileService.getStreamList(appFile.getName());
+                List<File> streamList = videoFileService.getStreamList(appFile.getName(), false);
                 if (streamList != null && streamList.size() > 0) {
                     for (File streamFile : streamList) {
                         File[] recordDateFileList = streamFile.listFiles();
@@ -79,9 +79,11 @@ public class RecordController {
                                                   @RequestParam int count){
         WVPResult<PageInfo<String>> result = new WVPResult<>();
         List<String> resultData = new ArrayList<>();
-        List<File> appList = videoFileService.getAppList();
-        for (File file : appList) {
-            resultData.add(file.getName());
+        List<File> appList = videoFileService.getAppList(true);
+        if (appList != null) {
+            for (File file : appList) {
+                resultData.add(file.getName());
+            }
         }
         result.setCode(0);
         result.setMsg("success");
@@ -114,9 +116,11 @@ public class RecordController {
             result.setMsg("app不能为空");
             return result;
         }
-        List<File> streamList = videoFileService.getStreamList(app);
-        for (File file : streamList) {
-            resultData.add(file.getName());
+        List<File> streamList = videoFileService.getStreamList(app, true);
+        if (streamList != null) {
+            for (File file : streamList) {
+                resultData.add(file.getName());
+            }
         }
         result.setCode(0);
         result.setMsg("success");
@@ -155,7 +159,7 @@ public class RecordController {
             result.setMsg("stream不能为空");
             return result;
         }
-        List<File> dateList = videoFileService.getDateList(app, stream, year, month);
+        List<File> dateList = videoFileService.getDateList(app, stream, year, month, true);
         for (File file : dateList) {
             resultData.add(file.getName());
         }
@@ -52,8 +52,16 @@ public class FFmpegExecUtils implements InitializingBean{
     @Autowired
     private RedisUtil redisUtil;
  
-    public FFprobe ffprobe;
-    public FFmpeg ffmpeg;
+    private FFprobe ffprobe;
+    private FFmpeg ffmpeg;
+
+    public FFprobe getFfprobe() {
+        return ffprobe;
+    }
+
+    public FFmpeg getFfmpeg() {
+        return ffmpeg;
+    }
  
     @Override
     public void afterPropertiesSet() throws Exception {
+package top.panll.assist.service;
+
+import org.apache.commons.io.FileUtils;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.scheduling.annotation.Scheduled;
+import org.springframework.stereotype.Component;
+import top.panll.assist.dto.UserSettings;
+
+import java.io.File;
+import java.text.ParseException;
+import java.text.SimpleDateFormat;
+import java.util.Calendar;
+import java.util.Date;
+import java.util.List;
+
+@Component
+public class FileManagerTimer {
+
+    private SimpleDateFormat formatter = new SimpleDateFormat("yyyy-MM-dd");
+
+    private final static Logger logger = LoggerFactory.getLogger(FileManagerTimer.class);
+
+    @Autowired
+    private UserSettings userSettings;
+
+    @Autowired
+    private VideoFileService videoFileService;
+
+//    @Scheduled(fixedDelay = 20000)   //测试 20秒执行一次
+    @Scheduled(cron = "0 0 0 * * ?")   //每天的0点执行
+    public void execute(){
+        int recordDay = userSettings.getRecordDay();
+        Date date=new Date();
+        Calendar calendar = Calendar.getInstance();
+        calendar.setTime(date);
+        calendar.add(Calendar.DAY_OF_MONTH, 0 - recordDay);
+        date = calendar.getTime();
+        logger.info("[录像巡查]移除 {} 之前的文件", formatter.format(date));
+        File recordFileDir = new File(userSettings.getRecord());
+        if (recordFileDir.canWrite()) {
+            List<File> appList = videoFileService.getAppList(false);
+            if (appList != null && appList.size() > 0) {
+                for (File appFile : appList) {
+                    if ("download.html".equals(appFile.getName())) {
+                        continue;
+                    }
+                    List<File> streamList = videoFileService.getStreamList(appFile, false);
+                    if (streamList != null && streamList.size() > 0) {
+                        for (File streamFile : streamList) {
+                            List<File> dateList = videoFileService.getDateList(streamFile, null, null, false);
+                            if (dateList != null && dateList.size() > 0) {
+                                for (File dateFile : dateList) {
+                                    try {
+                                        Date parse = formatter.parse(dateFile.getName());
+                                        if (parse.before(date)) {
+                                            boolean result = FileUtils.deleteQuietly(dateFile);
+                                            if (result) {
+                                                logger.info("[录像巡查]成功移除 {} ", dateFile.getAbsolutePath());
+                                            }else {
+                                                logger.info("[录像巡查]移除失败 {} ", dateFile.getAbsolutePath());
+                                            }
+                                        }
+                                    } catch (ParseException e) {
+                                        e.printStackTrace();
+                                    }
+                                }
+                            }
+                            if (streamFile.listFiles() == null &&  streamFile.listFiles().length == 0) {
+                                boolean result = FileUtils.deleteQuietly(streamFile);
+                                if (result) {
+                                    logger.info("[录像巡查]成功移除 {} ", streamFile.getAbsolutePath());
+                                }else {
+                                    logger.info("[录像巡查]移除失败 {} ", streamFile.getAbsolutePath());
+                                }
+                            }
+                        }
+                    }
+                    if (appFile.listFiles() == null || appFile.listFiles().length == 0) {
+                        boolean result = FileUtils.deleteQuietly(appFile);
+                        if (result) {
+                            logger.info("[录像巡查]成功移除 {} ", appFile.getAbsolutePath());
+                        }else {
+                            logger.info("[录像巡查]移除失败 {} ", appFile.getAbsolutePath());
+                        }
+                    }
+
+                }
+            }
+        }
+    }
+}
@@ -40,7 +40,7 @@ public class VideoFileService {
     private RedisTemplate redisTemplate;
  
     @Autowired
-    private FFmpegExecUtils fFmpegExecUtils;
+    private FFmpegExecUtils ffmpegExecUtils;
  
  
  
@@ -49,12 +49,14 @@ public class VideoFileService {
  
     private final String keyStr = "MERGEORCUT";
  
-    public List<File> getAppList() {
+    public List<File> getAppList(Boolean sort) {
         File recordFile = new File(userSettings.getRecord());
-        if (recordFile != null) {
+        if (recordFile != null && recordFile.isDirectory()) {
             File[] files = recordFile.listFiles();
             List<File> result = Arrays.asList(files);
-            Collections.sort(result);
+            if (sort != null && sort) {
+                Collections.sort(result);
+            }
             return result;
         }else {
             return null;
@@ -98,12 +100,18 @@ public class VideoFileService {
 //        }
 //    }
  
-    public List<File> getStreamList(String app) {
+    public List<File> getStreamList(String app, Boolean sort) {
         File appFile = new File(userSettings.getRecord() + File.separator + app);
-        if (appFile != null) {
+        return getStreamList(appFile, sort);
+    }
+
+    public List<File> getStreamList(File appFile, Boolean sort) {
+        if (appFile != null && appFile.isDirectory()) {
             File[] files = appFile.listFiles();
             List<File> result = Arrays.asList(files);
-            Collections.sort(result);
+            if (sort != null && sort) {
+                Collections.sort(result);
+            }
             return result;
         }else {
             return null;
@@ -116,7 +124,7 @@ public class VideoFileService {
      * @throws ParseException
      */
     public void handFile(File file) {
-        FFprobe ffprobe = fFmpegExecUtils.ffprobe;
+        FFprobe ffprobe = ffmpegExecUtils.getFfprobe();
         if(file.exists() && file.isFile() && !file.getName().startsWith(".")&& file.getName().endsWith(".mp4") && file.getName().indexOf(":") < 0) {
             try {
                 FFmpegProbeResult in = null;
@@ -137,7 +145,7 @@ public class VideoFileService {
                 String newName = file.getAbsolutePath().replace(file.getName(),
                         simpleDateFormat.format(startTime) + "-" + simpleDateFormat.format(endTime) + "-" + durationLong + ".mp4");
                 file.renameTo(new File(newName));
-                System.out.println(newName);
+                logger.debug("[处理文件] {}", file.getName());
             } catch (IOException e) {
                 logger.warn("文件可能以损坏[{}]", file.getAbsolutePath());
 //                e.printStackTrace();
@@ -151,11 +159,11 @@ public class VideoFileService {
  
         List<Map<String, String>> result = new ArrayList<>();
  
-        List<File> appList = getAppList();
+        List<File> appList = getAppList(true);
         if (appList != null && appList.size() > 0) {
             for (File appFile : appList) {
                 if (appFile.isDirectory()) {
-                    List<File> streamList = getStreamList(appFile.getName());
+                    List<File> streamList = getStreamList(appFile.getName(), true);
                     if (streamList != null && streamList.size() > 0) {
                         for (File streamFile : streamList) {
                             Map<String, String> data = new HashMap<>();
@@ -340,7 +348,7 @@ public class VideoFileService {
             mergeOrCutTaskInfo.setEndTime(endTimeInFile);
         }
  
-        fFmpegExecUtils.mergeOrCutFile(filesInTime, recordFile, stream, (status, percentage, result)->{
+        ffmpegExecUtils.mergeOrCutFile(filesInTime, recordFile, stream, (status, percentage, result)->{
             // 发出redis通知
             if (status.equals(Progress.Status.END.name())) {
                 mergeOrCutTaskInfo.setPercentage("1");
@@ -363,11 +371,22 @@ public class VideoFileService {
         return taskId;
     }
  
-    public List<File> getDateList(String app, String stream, Integer year, Integer month) {
+    /**
+     * 获取指定时间的日期文件夹
+     * @param app
+     * @param stream
+     * @param year
+     * @param month
+     * @return
+     */
+    public List<File> getDateList(String app, String stream, Integer year, Integer month, Boolean sort) {
         File recordFile = new File(userSettings.getRecord());
         File streamFile = new File(recordFile.getAbsolutePath() + File.separator + app + File.separator + stream);
-        if (!streamFile.exists()) {
-            logger.warn("获取[app: {}, stream: {}]的视频时未找到目录： {}", app, stream, stream);
+        return getDateList(streamFile, year, month, sort);
+    }
+    public List<File> getDateList(File streamFile, Integer year, Integer month, Boolean sort) {
+        if (!streamFile.exists() && streamFile.isDirectory()) {
+            logger.warn("获取[]的视频时未找到目录： {}",streamFile.getName());
             return null;
         }
         File[] dateFiles = streamFile.listFiles((File dir, String name)->{
@@ -393,17 +412,19 @@ public class VideoFileService {
  
         });
         List<File> dateFileList = Arrays.asList(dateFiles);
+        if (sort != null && sort) {
+            dateFileList.sort((File f1, File f2)->{
+                int sortResult = 0;
  
-        dateFileList.sort((File f1, File f2)->{
-            int sortResult = 0;
+                try {
+                    sortResult = simpleDateFormat.parse(f1.getName()).compareTo(simpleDateFormat.parse(f2.getName()));
+                } catch (ParseException e) {
+                    logger.error("格式化时间{}/{}错误", f1.getName(), f2.getName());
+                }
+                return sortResult;
+            });
+        }
  
-            try {
-                sortResult = simpleDateFormat.parse(f1.getName()).compareTo(simpleDateFormat.parse(f2.getName()));
-            } catch (ParseException e) {
-                logger.error("格式化时间{}/{}错误", f1.getName(), f2.getName());
-            }
-            return sortResult;
-        });
         return dateFileList;
     }