VideoFileService.java
6.87 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
package top.panll.assist.service;
import net.bramp.ffmpeg.FFprobe;
import net.bramp.ffmpeg.probe.FFmpegProbeResult;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
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.*;
@Service
public class VideoFileService {
private final static Logger logger = LoggerFactory.getLogger(VideoFileService.class);
@Autowired
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) {
try {
FFmpegProbeResult in = null;
in = ffprobe.probe(file.getAbsolutePath());
double duration = in.getFormat().duration * 1000;
String endTimeStr = file.getName().replace(".mp4", "");
SimpleDateFormat formatter = new SimpleDateFormat("yyyy-MM-dd HH-mm-ss");
SimpleDateFormat simpleDateFormat = new SimpleDateFormat("HH:mm:ss");
File dateFile = new File(file.getParent());
long endTime = formatter.parse(dateFile.getName() + " " + endTimeStr).getTime();
long startTime = endTime - new Double(duration).longValue();
String newName = file.getAbsolutePath().replace(file.getName(),
simpleDateFormat.format(startTime) + "-" + simpleDateFormat.format(endTime) + ".mp4");
file.renameTo(new File(newName));
System.out.println(newName);
} catch (IOException exception) {
exception.printStackTrace();
}
}
}
/**
* 获取制定推流的指定时间段内的推流
* @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<>();
if (app == null || stream == null) {
return result;
}
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((File dir, String name) -> {
return app.equals(name);
});
if (appFiles.length == 0) {
logger.warn("获取[app: {}, stream: {}, statime: {}, endTime: {}]的视频时未找到目录： {}", app, stream,
startTimeStr, endTimeStr, app);
return result;
}
File appFile = appFiles[0];
File[] streamFiles = appFile.listFiles((File dir, String name)-> {
return stream.equals(name);
});
if (streamFiles.length == 0) {
logger.warn("获取[app: {}, stream: {}, statime: {}, endTime: {}]的视频时未找到目录： {}", app, stream,
startTimeStr, endTimeStr, stream);
return result;
}
File streamFile = streamFiles[0];
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);
}
}