添加录像查询和录像裁剪合并

panlinlin
1 parent aad3e404
Showing 4 changed files with 205 additions and 12 deletions
src/main/java/top/panll/assist/config/StartConfig.java
src/main/java/top/panll/assist/service/FFmpegExecUtils.java
src/main/java/top/panll/assist/service/VideoFileService.java
src/main/java/top/panll/assist/utils/DateUtils.java
@@ -39,6 +39,14 @@ public class StartConfig implements CommandLineRunner {
             logger.error("[userSettings.record]配置错误，请检查路径是否存在");
             System.exit(1);
         }
+        if (!recordFile.canRead()) {
+            logger.error("[userSettings.record]路径无法读取");
+            System.exit(1);
+        }
+        if (!recordFile.canWrite()) {
+            logger.error("[userSettings.record]路径无法写入");
+            System.exit(1);
+        }
         try {
             String ffmpegPath = userSettings.getFfmpeg();
             String ffprobePath = userSettings.getFfprobe();
 package top.panll.assist.service;
  
 import net.bramp.ffmpeg.FFmpeg;
+import net.bramp.ffmpeg.FFmpegExecutor;
+import net.bramp.ffmpeg.FFmpegUtils;
 import net.bramp.ffmpeg.FFprobe;
+import net.bramp.ffmpeg.builder.FFmpegBuilder;
+import net.bramp.ffmpeg.job.FFmpegJob;
+import net.bramp.ffmpeg.probe.FFmpegProbeResult;
+import net.bramp.ffmpeg.progress.Progress;
+import net.bramp.ffmpeg.progress.ProgressListener;
+import org.springframework.util.DigestUtils;
+
+import java.io.BufferedWriter;
+import java.io.File;
+import java.io.FileWriter;
+import java.io.IOException;
+import java.util.List;
+import java.util.concurrent.TimeUnit;
  
 public class FFmpegExecUtils {
  
@@ -24,4 +39,82 @@ public class FFmpegExecUtils {
     public FFprobe ffprobe;
     public FFmpeg ffmpeg;
  
+    public interface VideoHandEndCallBack {
+        public void run(String status, double percentage, String result);
+    }
+
+    public static void mergeOrCutFile(List<File> fils, File dest, VideoHandEndCallBack callBack){
+        FFmpeg ffmpeg = FFmpegExecUtils.getInstance().ffmpeg;
+        FFprobe ffprobe = FFmpegExecUtils.getInstance().ffprobe;
+        if (fils == null || fils.size() == 0 || ffmpeg == null || ffprobe == null || dest== null || !dest.exists()){
+            callBack.run("error", 0.0, null);
+            return;
+        }
+
+        String temp = DigestUtils.md5DigestAsHex(String.valueOf(System.currentTimeMillis()).getBytes());
+        File tempFile = new File(dest.getAbsolutePath() + File.separator + temp);
+        if (!tempFile.exists()) {
+            tempFile.mkdirs();
+        }
+        FFmpegExecutor executor = new FFmpegExecutor(ffmpeg, ffprobe);
+        String fileListName = tempFile.getName() + File.separator + "fileList";
+        double durationAll = 0.0;
+        try {
+            BufferedWriter bw =new BufferedWriter(new FileWriter(fileListName));
+            for (File file : fils) {
+                FFmpegProbeResult in = ffprobe.probe(file.getAbsolutePath());
+                double duration = in.getFormat().duration;
+                System.out.println(duration);
+                bw.write("file " + file.getAbsolutePath());
+                bw.newLine();
+                durationAll += duration;
+            }
+            bw.flush();
+            bw.close();
+            System.out.println(durationAll);
+        } catch (IOException e) {
+            e.printStackTrace();
+            callBack.run("error", 0.0, null);
+        }
+        long startTime = System.currentTimeMillis();
+        FFmpegBuilder builder = new FFmpegBuilder()
+
+                .setFormat("concat")
+                .overrideOutputFiles(true)
+                .setInput(fileListName) // Or filename
+                .addExtraArgs("-safe", "0")
+                .addOutput(dest.getAbsolutePath() + File.separator + temp + ".mp4")
+                .setVideoCodec("copy")
+                .setFormat("mp4")
+
+                .done();
+
+
+        double finalDurationAll = durationAll;
+        FFmpegJob job = executor.createJob(builder, (Progress progress) -> {
+            final double duration_ns = finalDurationAll * TimeUnit.SECONDS.toNanos(1);
+            double percentage = progress.out_time_ns / duration_ns;
+
+            // Print out interesting information about the progress
+            System.out.println(String.format(
+                    "[%.0f%%] status:%s frame:%d time:%s ms fps:%.0f speed:%.2fx",
+                    percentage * 100,
+                    progress.status,
+                    progress.frame,
+                    FFmpegUtils.toTimecode(progress.out_time_ns, TimeUnit.NANOSECONDS),
+                    progress.fps.doubleValue(),
+                    progress.speed
+            ));
+            if (progress.status.equals(Progress.Status.END)){
+                callBack.run(progress.status.name(), percentage,
+                        dest.getAbsolutePath() + File.separator + temp + ".mp4");
+                System.out.println(System.currentTimeMillis() - startTime);
+            }else {
+                callBack.run(progress.status.name(), percentage, null);
+            }
+        });
+
+        job.run();
+    }
+
 }
@@ -8,15 +8,15 @@ import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import top.panll.assist.config.StartConfig;
 import top.panll.assist.dto.UserSettings;
+import top.panll.assist.utils.DateUtils;
  
 import java.io.File;
 import java.io.FilenameFilter;
 import java.io.IOException;
+import java.lang.reflect.Array;
 import java.text.ParseException;
 import java.text.SimpleDateFormat;
-import java.util.ArrayList;
-import java.util.Date;
-import java.util.List;
+import java.util.*;
  
 @Service
 public class VideoFileService {
@@ -27,6 +27,11 @@ public class VideoFileService {
     private UserSettings userSettings;
  
  
+    /**
+     * 对视频文件重命名， 00：00：00-00：00：00
+     * @param file
+     * @throws ParseException
+     */
     public void handFile(File file) throws ParseException {
         FFprobe ffprobe = FFmpegExecUtils.getInstance().ffprobe;
         if(file.isFile() && !file.getName().startsWith(".") && file.getName().indexOf(":") < 0) {
@@ -54,6 +59,14 @@ public class VideoFileService {
         }
     }
  
+    /**
+     * 获取制定推流的指定时间段内的推流
+     * @param app
+     * @param stream
+     * @param startTime
+     * @param endTime
+     * @return
+     */
     public List<File> getFilesInTime(String app, String stream, Date startTime, Date endTime){
  
         List<File> result = new ArrayList<>();
@@ -63,17 +76,15 @@ public class VideoFileService {
         }
  
         SimpleDateFormat formatter = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
+        SimpleDateFormat formatterForDate = new SimpleDateFormat("yyyy-MM-dd");
         String startTimeStr = formatter.format(startTime);
         String endTimeStr = formatter.format(endTime);
         logger.debug("获取[app: {}, stream: {}, statime: {}, endTime: {}]的视频", app, stream,
                 startTimeStr, endTimeStr);
  
         File file = new File(userSettings.getRecord());
-        File[] appFiles = file.listFiles(new FilenameFilter() {
-            @Override
-            public boolean accept(File dir, String name) {
+        File[] appFiles = file.listFiles((File dir, String name) -> {
                 return app.equals(name);
-            }
         });
         if (appFiles.length == 0) {
             logger.warn("获取[app: {}, stream: {}, statime: {}, endTime: {}]的视频时未找到目录： {}", app, stream,
@@ -81,11 +92,8 @@ public class VideoFileService {
             return result;
         }
         File appFile = appFiles[0];
-        File[] streamFiles = appFile.listFiles(new FilenameFilter() {
-            @Override
-            public boolean accept(File dir, String name) {
+        File[] streamFiles = appFile.listFiles((File dir, String name)-> {
                 return stream.equals(name);
-            }
         });
         if (streamFiles.length == 0) {
             logger.warn("获取[app: {}, stream: {}, statime: {}, endTime: {}]的视频时未找到目录： {}", app, stream,
@@ -93,10 +101,69 @@ public class VideoFileService {
             return result;
         }
         File streamFile = streamFiles[0];
-        // TODO 按时间获取文件
  
+        File[] dateFiles = streamFile.listFiles((File dir, String name) -> {
+            Date fileDate = null;
+            try {
+                fileDate = formatterForDate.parse(name);
+            } catch (ParseException e) {
+                logger.error("过滤日期文件时异常： {}-{}", name, e.getMessage());
+                return false;
+            }
+            return DateUtils.getStartOfDay(fileDate).after(startTime) && DateUtils.getEndOfDay(fileDate).before(endTime);
+        });
  
+        if (dateFiles != null && dateFiles.length > 0) {
+            for (File dateFile : dateFiles) {
+                // TODO 按时间获取文件
+                File[] files = dateFile.listFiles((File dir, String name) ->{
+                    boolean filterResult = false;
+                    if (name.contains(":") && name.endsWith(".mp4") && !name.startsWith(".")){
+                        String[] timeArray = name.split("-");
+                        if (timeArray.length == 2){
+                            String fileStartTimeStr = dateFile.getName() + " " + timeArray[0];
+                            String fileEndTimeStr = dateFile.getName() + " " + timeArray[1];
+                            try {
+                                filterResult = formatter.parse(fileStartTimeStr).after(startTime) && formatter.parse(fileEndTimeStr).before(endTime);
+                            } catch (ParseException e) {
+                                logger.error("过滤视频文件时异常： {}-{}", name, e.getMessage());
+                                return false;
+                            }
+                        }
+                    }
+                    return filterResult;
+                });
+
+                List<File> fileList = Arrays.asList(files);
+                result.addAll(fileList);
+            }
+        }
+        if (result.size() > 0) {
+            result.sort((File f1, File f2) -> {
+                boolean sortResult = false;
+                String[] timeArray1 = f1.getName().split("-");
+                String[] timeArray2 = f2.getName().split("-");
+                if (timeArray1.length == 2 && timeArray2.length == 2){
+                    File dateFile1 = f1.getParentFile();
+                    File dateFile2 = f2.getParentFile();
+                    String fileStartTimeStr1 = dateFile1.getName() + " " + timeArray1[0];
+                    String fileStartTimeStr2 = dateFile2.getName() + " " + timeArray1[0];
+                    try {
+                        sortResult = formatter.parse(fileStartTimeStr1).before(formatter.parse(fileStartTimeStr2)) ;
+                    } catch (ParseException e) {
+                        logger.error("排序视频文件时异常： {}-{}", fileStartTimeStr1, fileStartTimeStr2);
+                    }
+                }
+                return sortResult?1 : 0;
+            });
+        }
         return result;
     }
  
+
+    public void mergeOrCut(String app, String stream, Date startTime, Date endTime) {
+        List<File> filesInTime = this.getFilesInTime(app, stream, startTime, endTime);
+
+    }
+
 }
+package top.panll.assist.utils;
+
+import java.time.Instant;
+import java.time.LocalDateTime;
+import java.time.LocalTime;
+import java.time.ZoneId;
+import java.util.Date;
+
+public class DateUtils {
+
+    // 获得某天最大时间 2020-02-19 23:59:59
+    public static Date getEndOfDay(Date date) {
+        LocalDateTime localDateTime = LocalDateTime.ofInstant(Instant.ofEpochMilli(date.getTime()), ZoneId.systemDefault());;
+        LocalDateTime endOfDay = localDateTime.with(LocalTime.MAX);
+        return Date.from(endOfDay.atZone(ZoneId.systemDefault()).toInstant());
+    }
+
+    // 获得某天最小时间 2020-02-17 00:00:00
+    public static Date getStartOfDay(Date date) {
+        LocalDateTime localDateTime = LocalDateTime.ofInstant(Instant.ofEpochMilli(date.getTime()), ZoneId.systemDefault());
+        LocalDateTime startOfDay = localDateTime.with(LocalTime.MIN);
+        return Date.from(startOfDay.atZone(ZoneId.systemDefault()).toInstant());
+    }
+
+}