雷达视频下载

zc-business/src/main/java/com/zc/business/controller/RadarController.java

@@ -22,6 +22,9 @@ import java.time.Duration;
 import java.time.LocalDateTime;
 import java.time.format.DateTimeFormatter;
 import java.util.List;
+import java.util.concurrent.ExecutorService;
+import java.util.concurrent.Executors;
+import java.util.concurrent.TimeUnit;
 
 /**
  *
@@ -42,57 +45,80 @@ public class RadarController {
         LocalDateTime fifteenMinutesAgo = now.minus(Duration.ofMinutes(5));
         String FifteenMinutesAgo = fifteenMinutesAgo.format(formatter);
 
-        // 获取当前时间前20分钟的时间点
+        // 获取当前时间前25分钟的时间点
-        LocalDateTime twentyMinutesAgo = now.minus(Duration.ofMinutes(20));
+        LocalDateTime twentyMinutesAgo = now.minus(Duration.ofMinutes(25));
         String TwentyMinutesAgo = twentyMinutesAgo.format(formatter);
         DcWarning dcWarning = new DcWarning();
 
         dcWarning.setStartDate(TwentyMinutesAgo);
         dcWarning.setEndDate(FifteenMinutesAgo);
         List<DcWarning> radarList = dcWarningService.radarList(dcWarning);
-        radarList.forEach(radar -> {
+        // 创建一个固定大小的线程池
-            if (radar.getId() != null) {
+        ExecutorService executor = Executors.newFixedThreadPool(5); // 线程池大小
-                try {
-                    //获取雷达事件视频
-                    String baseUrl = dcWarningService.getRadarIncidentVideo(radar.getId());
-                    //  String baseUrl = "http://10.0.11.252:9021/profile/mp4/G35 长清大学城站 内广场_20241108_090235_60.mp4";
-                    String ftpServer= configService.selectConfigByKey("FTP-IP");//密钥
-                    // int port = 21;
-                    int port= Integer.parseInt(configService.selectConfigByKey("FTP-PORT"));
-                    //ftpuser
-                    //String user = "1911390090@qq.com";
-                    String username= configService.selectConfigByKey("FTP-USER");//密钥
-                    //Dxc123!@#
-                    String password= configService.selectConfigByKey("FPT-PASSWORD");//密钥
-     /*               String ftpServer = "10.168.71.135";
-                    int port = 21;
-                    String username = "1911390090@qq.com";
-                    String password = "989878wxl";*/
-                    String[] split = StringUtils.split(baseUrl, "_");
-                    // 拼接文件名
-                    String fileName = radar.getStakeMark() + "_" + split[1] + "_" + split[2] + "_" + split[3];
-
-                    DateTimeFormatter dirFormatter = DateTimeFormatter.ofPattern("yyyyMMdd");
-                    DateTimeFormatter hourFormatter = DateTimeFormatter.ofPattern("HH");
-                    String datePart = now.format(dirFormatter);
-                    String hourPart = now.format(hourFormatter);
-                    //文件路径
-//        String remoteFilePath = "/path/to/remote/directory";
-                    String remoteFilePath = "/radar/" + datePart + "/" + hourPart;
-                    String download = directDownloadToFtp(baseUrl, ftpServer, port, username, password, remoteFilePath, fileName);
-                    DcWarning dcWarning1 = new DcWarning();
-                    dcWarning1.setId(radar.getId());
-                    dcWarning1.setRadarUrl(download);
-                    int i = dcWarningService.updateDcWarning(dcWarning1);
-                } catch (Exception e) {
-                    throw new RuntimeException(e);
-                }
 
+        try {
+            radarList.forEach(radar -> {
+                if (radar != null && radar.getId() != null) {
+                    executor.submit(() -> processRadarEvent(radar, now));
+                }
+            });
+
+            // 关闭线程池，并等待所有任务完成
+            executor.shutdown();
+            executor.awaitTermination(1, TimeUnit.HOURS); // 设置超时时间
+        } catch (InterruptedException e) {
+            Thread.currentThread().interrupt(); // 恢复中断状态
+            log.error("线程池关闭过程中发生中断", e);
+        } finally {
+            if (!executor.isTerminated()) {
+                executor.shutdownNow(); // 强制关闭未完成的任务
             }
-        });
+        }
     }
 
+    private void processRadarEvent(DcWarning radar, LocalDateTime now) {
+        try {
+            // 获取雷达事件视频
+            String baseUrl = dcWarningService.getRadarIncidentVideo(radar.getId());
+            System.out.println("雷达视频地址：" + baseUrl);
 
+            if (baseUrl == null || baseUrl.isEmpty()) {
+                log.warn("雷达事件ID {} 没有对应的视频URL", radar.getId());
+                return;
+            }
+
+            String ftpServer= configService.selectConfigByKey("FTP-IP");//密钥
+            int port= Integer.parseInt(configService.selectConfigByKey("FTP-PORT"));
+            String username= configService.selectConfigByKey("FTP-USER");//密钥
+            String password= configService.selectConfigByKey("FPT-PASSWORD");//密钥
+//            String ftpServer = "192.168.3.1";
+//            int port = 21;
+//            String username = "1911390090@qq.com";
+//            String password = "989878wxl";
+
+            String[] split = StringUtils.split(baseUrl, "/");
+            // 拼接文件名
+            String fileName = split[split.length - 1];
+
+            DateTimeFormatter dirFormatter = DateTimeFormatter.ofPattern("yyyyMMdd");
+            DateTimeFormatter hourFormatter = DateTimeFormatter.ofPattern("HH");
+            String datePart = now.format(dirFormatter);
+            String hourPart = now.format(hourFormatter);
+            // 文件路径
+            String remoteFilePath = "/radar/" + datePart + "/" + hourPart;
+            String download = directDownloadToFtp(baseUrl, ftpServer, port, username, password, remoteFilePath, fileName);
+
+            DcWarning dcWarning1 = new DcWarning();
+            dcWarning1.setId(radar.getId());
+            dcWarning1.setRadarUrl(download);
+            int i = dcWarningService.updateDcWarning(dcWarning1);
+            if (i <= 0) {
+                log.error("更新数据库失败，ID: {}", radar.getId());
+            }
+        } catch (Exception e) {
+            log.error("处理雷达事件ID {} 时发生异常", radar.getId(), e);
+        }
+    }
     public static String directDownloadToFtp(String baseUrl, String ftpServer, int port, String user, String pass, String remotePath, String fileName) throws IOException, URISyntaxException, InterruptedException {
         FTPClient ftpClient = new FTPClient();
         HttpURLConnection httpConn = null;
@@ -104,8 +130,7 @@ public class RadarController {
             System.out.println("encodedUrl: " + encodedUrl);
             URI uri = new URI(encodedUrl);
             URL url = uri.toURL();
-            // 睡眠3秒钟
+
-            Thread.sleep(3000);  //
             httpConn = (HttpURLConnection) url.openConnection();
             int responseCode = httpConn.getResponseCode();
 
@@ -115,7 +140,7 @@ public class RadarController {
                 ftpClient.connect(ftpServer, port);
                 boolean loginResult = ftpClient.login(user, pass);
                 if (!loginResult) {
-                    throw new IOException("FTP登录失败");
+                    log.info("FTP登录失败");
                 }
                 log.info("成功登录到FTP服务器");
                 // 设置为二进制传输模式和被动模式
@@ -145,7 +170,7 @@ public class RadarController {
                     log.info("最终工作目录: " + ftpClient.printWorkingDirectory());
                 }
                 // 上传文件到FTP服务器
-                boolean done = ftpClient.storeFile(fileName, inputStream);
+                boolean done =  ftpClient.storeFile(new String(fileName.getBytes("GBK"),"ISO-8859-1"), inputStream);
                 if (done) {
                     log.info("文件上传成功");
                     return URL+ftpClient.printWorkingDirectory()+"/" + fileName;