Shikong/gb28181-docking-platform-mocking
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

视频缓存获取方式优化

Browse Source
This commit is contained in:
shikong 2024-02-07 09:02:16 +08:00
parent f4c687b403
commit dead56fc66
1 changed files with 14 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
gb28181-mocking-service/src/main/java/cn/skcks/docking/gb28181/mocking/core/sip/service/VideoCacheManager.java
View File

@ -23,8 +23,6 @@ import org.springframework.stereotype.Service;
import java.io.File; import java.io.File;
import java.io.FileOutputStream; import java.io.FileOutputStream;
import java.io.InputStream; import java.io.InputStream;
import java.nio.channels.FileChannel;
import java.nio.channels.FileLock;
import java.nio.file.Paths; import java.nio.file.Paths;
import java.util.Date; import java.util.Date;
import java.util.concurrent.*; import java.util.concurrent.*;
@ -71,7 +69,15 @@ public class VideoCacheManager {
public CompletableFuture<JsonResponse<String>> get(String deviceCode, Date startTime, Date endTime){ public CompletableFuture<JsonResponse<String>> get(String deviceCode, Date startTime, Date endTime){
String name = fileName(deviceCode, startTime, endTime); String name = fileName(deviceCode, startTime, endTime);
return tasks.get(name); CompletableFuture<JsonResponse<String>> future = tasks.get(name);
if(future == null){
File realFile = Paths.get(deviceProxyConfig.getPreDownloadForRecordInfo().getCachePath(),fileName(deviceCode, startTime, endTime) + ".mp4").toFile();
if(realFile.exists()){
log.info("文件 {} 已缓存, 直接返回", realFile.getAbsolutePath());
return CompletableFuture.completedFuture(JsonResponse.success(realFile.getAbsolutePath()));
}
}
return future;
} }
@SneakyThrows @SneakyThrows
@ -91,17 +97,14 @@ public class VideoCacheManager {
.addQuery("useDownload", true).build(); .addQuery("useDownload", true).build();
File file = Paths.get(deviceProxyConfig.getPreDownloadForRecordInfo().getCachePath(),fileName(deviceCode, startTime, endTime) + ".mp4.tmp").toFile(); File file = Paths.get(deviceProxyConfig.getPreDownloadForRecordInfo().getCachePath(),fileName(deviceCode, startTime, endTime) + ".mp4.tmp").toFile();
log.info("文件存储路径 => {}", file.getAbsolutePath()); log.info("文件存储路径 => {}", file.getAbsolutePath());
log.info("文件 {}, 是否存在: {}", file.getAbsolutePath(), file.exists()); log.info("临时文件 {}, 是否存在: {}", file.getAbsolutePath(), file.exists());
if(file.exists()){ if(file.exists()){
file.delete(); file.delete();
log.info("删除已存但未完成下载的文件 => {}", file.getAbsolutePath()); log.info("删除已存但未完成下载的临时文件 => {}", file.getAbsolutePath());
} }
try (FileOutputStream outputStream = new FileOutputStream(file)) { try (FileOutputStream outputStream = new FileOutputStream(file)) {
FileChannel channel = outputStream.getChannel();
FileLock lock = channel.lock();
HttpClient client = HttpClients.custom() HttpClient client = HttpClients.custom()
.setConnectionManager(manager) .setConnectionManager(manager)
.setConnectionManagerShared(true) .setConnectionManagerShared(true)
@ -114,10 +117,10 @@ public class VideoCacheManager {
return stream; return stream;
}); });
execute.close(); execute.close();
log.info("视频下载完成 => {}", file.getAbsolutePath()); log.info("临时文件下载完成 => {}", file.getAbsolutePath());
log.info("文件 {}, 是否存在: {}", file.getAbsolutePath(), file.exists()); log.info("临时文件 {}, 是否存在: {}", file.getAbsolutePath(), file.exists());
file.renameTo(realFile); file.renameTo(realFile);
lock.release(); log.info("保存视频文件 => {}", realFile.getAbsolutePath());
return JsonResponse.success(realFile.getAbsolutePath()); return JsonResponse.success(realFile.getAbsolutePath());
} catch (Exception e) { } catch (Exception e) {
log.error("视频下载失败 => {}", e.getMessage()); log.error("视频下载失败 => {}", e.getMessage());