支持vpt方式

This commit is contained in:
2025-02-07 23:00:23 +08:00
parent a016622cc9
commit ea3ce510d8
9 changed files with 191 additions and 15 deletions

View File

@ -7,6 +7,7 @@ import com.ycwl.basic.device.enums.DeviceStoreTypeEnum;
import com.ycwl.basic.device.operator.AliOssStorageOperator;
import com.ycwl.basic.device.operator.IDeviceStorageOperator;
import com.ycwl.basic.device.operator.LocalStorageOperator;
import com.ycwl.basic.device.operator.VptPassiveStorageOperator;
import com.ycwl.basic.device.operator.WvpActiveStorageOperator;
import com.ycwl.basic.device.operator.WvpPassiveStorageOperator;
import com.ycwl.basic.model.pc.device.entity.DeviceConfigEntity;
@ -24,6 +25,8 @@ public class DeviceFactory {
operator = new WvpActiveStorageOperator(config.getStoreConfigJson());
} else if (Integer.valueOf(DeviceStoreTypeEnum.WVP_PASSIVE.getType()).equals(config.getStoreType())) {
operator = new WvpPassiveStorageOperator(config.getStoreConfigJson());
} else if (Integer.valueOf(DeviceStoreTypeEnum.VPT_PASSIVE.getType()).equals(config.getStoreType())) {
operator = new VptPassiveStorageOperator(config.getStoreConfigJson());
} else if (Integer.valueOf(DeviceStoreTypeEnum.LOCAL.getType()).equals(config.getStoreType())) {
operator = new LocalStorageOperator(config.getStoreConfigJson());
}

View File

@ -9,6 +9,7 @@ public class FileObject {
private String path;
private String name;
private String url;
private boolean exact = false;
private boolean needDownload = false;
private Date createTime;
private Date endTime;

View File

@ -0,0 +1,8 @@
package com.ycwl.basic.device.entity.vpt_passive;
import lombok.Data;
@Data
public class VptPassiveStorageConfig {
private String deviceNo;
}

View File

@ -9,6 +9,7 @@ public enum DeviceStoreTypeEnum {
ALI_OSS(1, "阿里云OSS"),
WVP_ACTIVE(3, "WVP主动模式"),
WVP_PASSIVE(4, "WVP被动模式"),
VPT_PASSIVE(5, "VPT被动模式"),
LOCAL(2, "本地文件");
private final int type;

View File

@ -0,0 +1,117 @@
package com.ycwl.basic.device.operator;
import com.alibaba.fastjson.JSON;
import com.ycwl.basic.device.entity.common.FileObject;
import com.ycwl.basic.device.entity.vpt_passive.VptPassiveStorageConfig;
import com.ycwl.basic.device.entity.wvp_passive.WvpPassiveStorageConfig;
import com.ycwl.basic.model.wvp.WvpSyncReqVo;
import com.ycwl.basic.storage.utils.StorageUtil;
import com.ycwl.basic.utils.SnowFlakeUtil;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Date;
import java.util.List;
import java.util.concurrent.ConcurrentHashMap;
import java.util.stream.Collectors;
@Slf4j
public class VptPassiveStorageOperator extends ADeviceStorageOperator {
public static class Task {
public Long taskId;
public Long scenicId;
public Long deviceId;
public String deviceNo;
public Date startTime;
public Date endTime;
}
private static List<Task> taskList = Collections.synchronizedList(new ArrayList<>());
private static ConcurrentHashMap<Long, FileObject> fileListMap = new ConcurrentHashMap<>();
private VptPassiveStorageConfig config;
public VptPassiveStorageOperator(String configJson) {
loadConfig(configJson);
}
public static String getUrlForTask(Long taskId) {
return StorageUtil.joinPath("video-source", taskId.toString() + ".mp4");
}
public static void onReceiveResult(Long taskId, FileObject fileObject) {
if (fileObject == null) {
log.info("任务{}获取视频失败!", taskId);
fileListMap.put(taskId, new FileObject());
} else {
fileListMap.put(taskId, fileObject);
}
}
public static List<Task> getTaskListByScenicIdAndDeviceList(Long scenicId, List<WvpSyncReqVo.DeviceItem> deviceNos) {
List<Task> scenicTask = taskList.stream()
.filter(task -> task.scenicId.equals(scenicId))
.filter(task -> deviceNos.stream().anyMatch(deviceNo -> deviceNo.getDeviceNo().equals(task.deviceNo)))
.collect(Collectors.toList());
if (scenicTask.isEmpty()) {
return Collections.emptyList();
}
taskList.removeAll(scenicTask);
return scenicTask;
}
@Override
public void loadConfig(String configJson) {
this.config = JSON.parseObject(configJson, VptPassiveStorageConfig.class);
}
@Override
public List<FileObject> getFileListByDtRange(Date startDate, Date endDate) {
Task task = new Task();
task.taskId = SnowFlakeUtil.getLongId();
task.scenicId = device.getScenicId();
task.deviceId = device.getId();
if (StringUtils.isNotBlank(config.getDeviceNo())) {
task.deviceNo = config.getDeviceNo();
} else {
task.deviceNo = device.getNo2();
}
task.startTime = startDate;
task.endTime = endDate;
taskList.add(task);
Date taskStartTime = new Date();
while (true) {
if (new Date().getTime() - taskStartTime.getTime() > 80000L) {
log.info("任务{}获取视频超时!", task.taskId);
fileListMap.remove(task.taskId);
return Collections.emptyList();
}
if (fileListMap.containsKey(task.taskId)) {
FileObject fileObject = fileListMap.get(task.taskId);
if (StringUtils.isNotBlank(fileObject.getUrl())) {
log.info("任务{}获取视频成功!{}", task.taskId, fileObject);
fileListMap.remove(task.taskId);
fileObject.setExact(true);
return Collections.singletonList(fileObject);
} else {
log.info("任务{}获取视频失败!", task.taskId);
return Collections.emptyList();
}
} else {
try {
Thread.sleep(1000L);
} catch (InterruptedException e) {
return Collections.emptyList();
}
}
}
}
@Override
public boolean removeFilesBeforeDate(Date date) {
// 不做删除
return false;
}
}

View File

@ -80,6 +80,7 @@ public class WvpPassiveStorageOperator extends ADeviceStorageOperator {
while (true) {
if (new Date().getTime() - taskStartTime.getTime() > 80000L) {
log.info("任务{}获取视频超时!", task.taskId);
fileListMap.remove(task.taskId);
return Collections.emptyList();
}
if (fileListMap.containsKey(task.taskId)) {
@ -87,6 +88,7 @@ public class WvpPassiveStorageOperator extends ADeviceStorageOperator {
if (StringUtils.isNotBlank(fileObject.getUrl())) {
log.info("任务{}获取视频成功!{}", task.taskId, fileObject);
fileListMap.remove(task.taskId);
fileObject.setExact(true);
return Collections.singletonList(fileObject);
} else {
log.info("任务{}获取视频失败!", task.taskId);