通威测试视觉相关逻辑
LAPTOP-S9HJSOEB\昊天
1 year ago
parent
818bf0a376
commit
413669db08
@ -0,0 +1,125 @@
|
|||||||
|
package com.zhehekeji.web.service.Queue;
|
||||||
|
|
||||||
|
import com.zhehekeji.common.util.SpringContextUtil;
|
||||||
|
import com.zhehekeji.web.lib.CameraControlModule;
|
||||||
|
import com.zhehekeji.web.lib.CameraDelayTask;
|
||||||
|
import com.zhehekeji.web.lib.TaskDelayExecutor;
|
||||||
|
|
||||||
|
import java.time.LocalDateTime;
|
||||||
|
import java.util.concurrent.*;
|
||||||
|
|
||||||
|
public class SingleThreadQueueWithTimeout {
|
||||||
|
|
||||||
|
private static ExecutorService exec = Executors.newFixedThreadPool(1);
|
||||||
|
class Task implements Delayed{
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public long getDelay(TimeUnit unit) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public int compareTo(Delayed o) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private static DelayQueue<CameraDelayTask> queue = new DelayQueue<>();
|
||||||
|
|
||||||
|
public static void addMp4DelayTask(Integer cameraId, String path, LocalDateTime startTime, LocalDateTime endTime, Long delayTime) {
|
||||||
|
CameraDelayTask cameraDelayTask = new CameraDelayTask(cameraId, startTime, endTime,path, 0,delayTime);
|
||||||
|
queue.add(cameraDelayTask);
|
||||||
|
}
|
||||||
|
|
||||||
|
public static void addPicDelayTask(Integer cameraId, String path, Long delayTime) {
|
||||||
|
CameraDelayTask cameraDelayTask = new CameraDelayTask(cameraId, null, null,path, 1,delayTime);
|
||||||
|
queue.add(cameraDelayTask);
|
||||||
|
}
|
||||||
|
|
||||||
|
public static void addGyrateCameraTask(Integer cameraId, Long delayTime,Integer ptzId){
|
||||||
|
CameraDelayTask cameraDelayTask = new CameraDelayTask(cameraId, null, null,null, 2,delayTime);
|
||||||
|
cameraDelayTask.setPtzId(ptzId);
|
||||||
|
queue.add(cameraDelayTask);
|
||||||
|
}
|
||||||
|
|
||||||
|
public static void runMp4DownloadExecutor(){
|
||||||
|
exec.execute(new SingleThreadQueueWithTimeout.Consumer());
|
||||||
|
}
|
||||||
|
|
||||||
|
private static class Consumer implements Runnable {
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void run() {
|
||||||
|
while (true) {
|
||||||
|
try {
|
||||||
|
CameraControlModule cameraControlModule = SpringContextUtil.getBean(CameraControlModule.class);
|
||||||
|
CameraDelayTask cameraDelayTask = queue.take();
|
||||||
|
if(cameraDelayTask != null){
|
||||||
|
|
||||||
|
if(cameraDelayTask.getType() == 0){
|
||||||
|
|
||||||
|
cameraControlModule.downloadMp4(cameraDelayTask.getCameraId(), cameraDelayTask.getPath(), cameraDelayTask.getStartTime(), cameraDelayTask.getEndTime());
|
||||||
|
}else if(cameraDelayTask.getType() == 1){
|
||||||
|
cameraControlModule.pic(cameraDelayTask.getCameraId(),0, cameraDelayTask.getPath());
|
||||||
|
}else if(cameraDelayTask.getType() == 2){
|
||||||
|
cameraControlModule.toPtz(cameraDelayTask.getPtzId(),cameraDelayTask.getCameraId());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
} catch (InterruptedException e) {
|
||||||
|
e.printStackTrace();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
public static void main(String[] args) {
|
||||||
|
// 创建一个单线程的 ExecutorService
|
||||||
|
ExecutorService executorService = Executors.newSingleThreadExecutor();
|
||||||
|
|
||||||
|
// 创建一个 LinkedBlockingQueue 来存储任务
|
||||||
|
BlockingQueue<Runnable> queue = new LinkedBlockingQueue<>();
|
||||||
|
|
||||||
|
// 添加任务到队列
|
||||||
|
queue.add(() -> {
|
||||||
|
System.out.println("任务 1 开始执行");
|
||||||
|
try {
|
||||||
|
Thread.sleep(2000); // 模拟任务执行时间
|
||||||
|
} catch (InterruptedException e) {
|
||||||
|
Thread.currentThread().interrupt();
|
||||||
|
System.out.println("任务 1 被中断");
|
||||||
|
}
|
||||||
|
System.out.println("任务 1 执行结束");
|
||||||
|
});
|
||||||
|
|
||||||
|
queue.add(() -> {
|
||||||
|
System.out.println("任务 2 开始执行");
|
||||||
|
try {
|
||||||
|
Thread.sleep(1000); // 模拟任务执行时间
|
||||||
|
} catch (InterruptedException e) {
|
||||||
|
Thread.currentThread().interrupt();
|
||||||
|
System.out.println("任务 2 被中断");
|
||||||
|
}
|
||||||
|
System.out.println("任务 2 执行结束");
|
||||||
|
});
|
||||||
|
|
||||||
|
// 提交任务并设置超时处理
|
||||||
|
Future<?> future1 = executorService.submit(queue.poll());
|
||||||
|
Future<?> future2 = executorService.submit(queue.poll());
|
||||||
|
|
||||||
|
try {
|
||||||
|
// 设置超时时间
|
||||||
|
future1.get(3000, TimeUnit.MILLISECONDS);
|
||||||
|
future2.get(2000, TimeUnit.MILLISECONDS);
|
||||||
|
} catch (TimeoutException e) {
|
||||||
|
System.out.println("任务超时");
|
||||||
|
future1.cancel(true); // 取消任务
|
||||||
|
future2.cancel(true); // 取消任务
|
||||||
|
} catch (InterruptedException | ExecutionException e) {
|
||||||
|
e.printStackTrace();
|
||||||
|
} finally {
|
||||||
|
executorService.shutdown(); // 关闭 ExecutorService
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue