添加获取视频时长接口

648540858
1 parent 6f091112
Showing 4 changed files with 93 additions and 8 deletions
pom.xml
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/VideoFileService.java
@@ -44,12 +44,6 @@
         </dependency>
  
         <dependency>
-            <groupId>log4j</groupId>
-            <artifactId>log4j</artifactId>
-            <version>1.2.17</version>
-        </dependency>
-
-        <dependency>
             <groupId>net.bramp.ffmpeg</groupId>
             <artifactId>ffmpeg</artifactId>
             <version>0.6.2</version>
@@ -411,4 +411,25 @@ public class RecordController {
         ret.put("data", JSON.toJSON(spaceInfo));
         return new ResponseEntity<>(ret.toString(), HttpStatus.OK);
     }
+
+    /**
+     * 录像文件的时长
+     */
+    @ApiOperation("录像文件的时长")
+    @ApiImplicitParams({
+            @ApiImplicitParam(name="app", value = "应用名", required = true, dataTypeClass = String.class),
+            @ApiImplicitParam(name="stream", value = "流ID", required = true, dataTypeClass = String.class),
+            @ApiImplicitParam(name="recordIng", value = "是否录制中", required = false, dataTypeClass = String.class),
+    })
+    @ResponseBody
+    @GetMapping(value = "/file/duration", produces = "application/json;charset=UTF-8")
+    public ResponseEntity<String> fileDuration( @RequestParam String app, @RequestParam String stream, Boolean recordIng) {
+        if (recordIng == null) recordIng = true;
+        JSONObject ret = new JSONObject();
+        ret.put("code", 0);
+        ret.put("msg", "success");
+        Double duration = videoFileService.fileDuration(app, stream, recordIng);
+        ret.put("data", duration);
+        return new ResponseEntity<>(ret.toString(), HttpStatus.OK);
+    }
 }
@@ -72,6 +72,8 @@ public class FFmpegExecUtils implements InitializingBean{
         logger.info("wvp-pro辅助程序启动成功。 \n{}\n{} ", this.ffmpeg.version(), this.ffprobe.version());
     }
  
+
+
     public interface VideoHandEndCallBack {
         void run(String status, double percentage, String result);
     }
@@ -149,4 +151,9 @@ public class FFmpegExecUtils implements InitializingBean{
         job.run();
     }
  
+    public double duration(File file) throws IOException {
+        FFmpegProbeResult in = ffprobe.probe(file.getAbsolutePath());
+        return in.getFormat().duration * 1000;
+    }
+
 }
@@ -135,8 +135,7 @@ public class VideoFileService {
         FFprobe ffprobe = ffmpegExecUtils.getFfprobe();
         if(file.exists() && file.isFile() && !file.getName().startsWith(".")&& file.getName().endsWith(".mp4") && file.getName().indexOf(":") < 0) {
             try {
-                FFmpegProbeResult in = null;
-                in = ffprobe.probe(file.getAbsolutePath());
+                FFmpegProbeResult in = ffprobe.probe(file.getAbsolutePath());
                 double duration = in.getFormat().duration * 1000;
                 String endTimeStr = file.getName().replace(".mp4", "");
  
@@ -206,6 +205,55 @@ public class VideoFileService {
         return result;
     }
  
+    public List<File> getAllFiles(String app, String stream, Boolean recording){
+
+        List<File> result = new ArrayList<>();
+        if (app == null || stream == null) {
+            return result;
+        }
+
+        SimpleDateFormat formatter = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
+        SimpleDateFormat formatterForDate = new SimpleDateFormat("yyyy-MM-dd");
+
+        File recordFile = new File(userSettings.getRecord());
+        File streamFile = new File(recordFile.getAbsolutePath() + File.separator + app + File.separator + stream + File.separator);
+        if (!streamFile.exists()) {
+            logger.warn("获取[app: {}, stream: {}]的视频时未找到目录： {}", app, stream, stream);
+            return null;
+        }
+
+        File[] dateFiles = streamFile.listFiles((File dir, String name) -> {
+            if (new File(dir + File.separator + name).isFile()) {
+                return false;
+            }
+            try {
+                formatterForDate.parse(name);
+            } catch (ParseException e) {
+                logger.error("过滤日期文件时异常： {}-{}", name, e.getMessage());
+                return false;
+            }
+
+            return true ;
+        });
+
+        if (dateFiles != null && dateFiles.length > 0) {
+            for (File dateFile : dateFiles) {
+                File[] files = dateFile.listFiles((File dir, String name) ->{
+                    File currentFile = new File(dir + File.separator + name);
+                    if (recording) {
+                        return currentFile.isFile()  && name.contains(":") && name.endsWith(".mp4") && currentFile.length() > 0;
+                    }else {
+                        return currentFile.isFile()  && name.contains(":") && name.endsWith(".mp4") && currentFile.length() > 0 && !name.startsWith(".");
+                    }
+                });
+                List<File> fileList = Arrays.asList(files);
+                result.addAll(fileList);
+            }
+        }
+        return result;
+    }
+
+
     /**
      * 获取制定推流的指定时间段内的推流
      * @param app
@@ -604,4 +652,19 @@ public class VideoFileService {
         }
         return result;
     }
+
+    public Double fileDuration(String app, String stream, Boolean recordIng) {
+        List<File> allFiles = getAllFiles(app, stream, recordIng);
+        Double durationResult = 0.0;
+        if (allFiles.size() > 0) {
+            for (File file : allFiles) {
+                try {
+                    durationResult += ffmpegExecUtils.duration(file);
+                } catch (IOException e) {
+                    logger.error("获取{}视频时长错误：{}", file.getAbsolutePath(), e.getMessage());
+                }
+            }
+        }
+        return durationResult;
+    }
 }