forked from uav/ard-uav-server
ard-work/src/main/java/com/ard/work/service/impl/ArdMediaServiceImpl.java
@@ -1,31 +1,297 @@ package com.ard.work.service.impl; import com.ard.work.domain.entity.ArdMedia; import com.ard.work.mapper.ArdMediaMapper; import com.ard.common.utils.DateUtils; import com.ard.common.utils.file.MimeTypeUtils; import com.ard.oss.service.impl.OssServiceContext; import com.ard.work.component.forest.TyClient; import com.ard.work.domain.dto.BoxPhotosDTO; import com.ard.work.domain.dto.BoxVideosDTO; import com.ard.work.domain.entity.ArdFlightTask; import com.ard.work.domain.entity.ArdMediaPic; import com.ard.work.domain.entity.ArdMediaVideo; import com.ard.work.domain.receiver.BoxPhotosReceiver; import com.ard.work.domain.receiver.BoxVideosReceiver; import com.ard.work.domain.receiver.TyHttpCommonReceiver; import com.ard.work.mapper.ArdFlightTaskMapper; import com.ard.work.mapper.ArdMediaPicMapper; import com.ard.work.mapper.ArdMediaVideoMapper; import com.ard.work.service.IArdMediaService; import com.ard.work.util.osSelect; import lombok.extern.slf4j.Slf4j; import org.springframework.scheduling.annotation.Scheduled; import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor; import org.springframework.stereotype.Service; import javax.annotation.Resource; import javax.net.ssl.HttpsURLConnection; import javax.net.ssl.SSLContext; import javax.net.ssl.TrustManager; import javax.net.ssl.X509TrustManager; import java.io.*; import java.net.HttpURLConnection; import java.net.URL; import java.net.URLConnection; import java.nio.file.Files; import java.nio.file.Path; import java.security.cert.X509Certificate; import java.util.ArrayList; import java.util.Date; import java.util.List; import java.util.Set; import java.util.concurrent.CompletableFuture; import java.util.concurrent.ConcurrentHashMap; import java.util.concurrent.TimeUnit; import java.util.stream.Collectors; /** * Description: * Author: lsy * Date: 2025年11月18日13:17 **/ @Slf4j @Service public class ArdMediaServiceImpl implements IArdMediaService { @Resource private ArdMediaMapper ardMediaMapper; /** * 查询媒体数据列表 * * @param ardMedia 媒体数据 * @return 媒体数据集合 */ @Override public List<ArdMedia> selectArdMediaList(ArdMedia ardMedia) { return ardMediaMapper.selectArdMedia(ardMedia); @Resource private TyClient tyClient; @Resource private ArdMediaPicMapper ardMediaPicMapper; @Resource private ArdMediaVideoMapper ardMediaVideoMapper; @Resource private ArdFlightTaskMapper ardFlightTaskMapper; // 调用接口获取架次图片 private List<BoxPhotosReceiver> getBoxPhotos(String serverId, BoxPhotosDTO boxPhotosDTO) { TyHttpCommonReceiver<List<BoxPhotosReceiver>> boxPhotos = tyClient.getBoxPhotos(serverId, boxPhotosDTO); return boxPhotos.getContent(); } // 调用接口获取架次视频 private List<BoxVideosReceiver> getBoxVideos(String serverId, BoxVideosDTO boxVideosDTO) { TyHttpCommonReceiver<List<BoxVideosReceiver>> boxVideos = tyClient.getBoxVideos(serverId, boxVideosDTO); return boxVideos.getContent(); } @Override public List<ArdMediaPic> selectArdMediaPicList(ArdMediaPic ardMediaPic) { return ardMediaPicMapper.selectPicList(ardMediaPic); } @Override public List<ArdMediaVideo> selectArdMediaVideoList(ArdMediaVideo ardMediaVideo) { return ardMediaVideoMapper.selectArdMediaVideoList(ardMediaVideo); } @Override public void insertPhotoToDatabase(ArdFlightTask flightTask, String serverId, String boxSn) { log.warn("开始获取图片信息"); // 查询图片 BoxPhotosDTO boxPhotosDTO = new BoxPhotosDTO(); boxPhotosDTO.setTaskId(flightTask.getTaskId()); boxPhotosDTO.setBoxSn(boxSn); boxPhotosDTO.setPageIndex(1); boxPhotosDTO.setPageSize(100); List<BoxPhotosReceiver> photoList = getBoxPhotos(serverId, boxPhotosDTO); if (photoList != null && !photoList.isEmpty()) { addPhotos(flightTask, photoList); } } @Override public void insertVideoToDatabase(ArdFlightTask flightTask, String serverId, String boxSn) { log.warn("开始获取视频信息"); BoxVideosDTO boxVideosDTO = new BoxVideosDTO(); boxVideosDTO.setTaskId(flightTask.getTaskId()); boxVideosDTO.setBoxSn(boxSn); boxVideosDTO.setPageIndex(1); boxVideosDTO.setPageSize(100); List<BoxVideosReceiver> videoList = getBoxVideos(serverId, boxVideosDTO); if (videoList != null && !videoList.isEmpty()) { addVideos(flightTask, videoList); } } @Override public void syncPicToDatabase(ArdFlightTask flightTask, String serverId, String boxSn) { // 获取本地数据库中已存在的图片记录 List<ArdMediaPic> existingPhotos = ardMediaPicMapper.selectPicByTaskId(flightTask.getTaskId()); int count = existingPhotos.size(); //获取本地库图片数量 // 查询图片 BoxPhotosDTO boxPhotosDTO = BoxPhotosDTO.builder() .taskId(flightTask.getTaskId()) .boxSn(boxSn) .pageIndex(1) .pageSize(500) .build(); TyHttpCommonReceiver<List<BoxPhotosReceiver>> boxPhotos = tyClient.getBoxPhotos(serverId, boxPhotosDTO); List<BoxPhotosReceiver> photoList = boxPhotos.getContent(); int size = photoList.size(); if (size > count) { log.info("任务【{}】图片数量不一致，需要更新数据库", flightTask.getTaskId()); // 需要插入新的视频 // 提取现有的图片路径（根据 imgUrl 来判断是否重复） List<String> existingPhotoUrls = existingPhotos.stream() .map(ArdMediaPic::getSourceUrl) // 提取 ArdMediaPic 中的图片 URL .collect(Collectors.toList()); // 筛选出 photoList 中没有出现在 existingPhotos 中的新图片 List<BoxPhotosReceiver> newPhotos = photoList.stream() .filter(photo -> !existingPhotoUrls.contains(photo.getImgUrl())) // 过滤出新图片 .collect(Collectors.toList()); if (!newPhotos.isEmpty()) { // 插入新的图片记录 addPhotos(flightTask, newPhotos); } } } @Override public void syncVideoToDatabase(ArdFlightTask flightTask, String serverId, String boxSn) { // 获取本地数据库中已存在的视频记录 List<ArdMediaVideo> existingVideos = ardMediaVideoMapper.selectArdMediaVideoByTaskId(flightTask.getTaskId()); int count = existingVideos.size(); //获取本地库视频数量 // 查询图片 BoxVideosDTO boxVideosDTO = BoxVideosDTO.builder() .taskId(flightTask.getTaskId()) .boxSn(boxSn) .pageIndex(1) .pageSize(500) .build(); TyHttpCommonReceiver<List<BoxVideosReceiver>> boxVideos = tyClient.getBoxVideos(serverId, boxVideosDTO); List<BoxVideosReceiver> videoList = boxVideos.getContent(); int size = videoList.size(); if (size > count) { log.info("任务【{}】视频数量不一致，需要更新数据库", flightTask.getTaskId()); // 需要插入新的视频 // 提取现有的图片路径（根据 imgUrl 来判断是否重复） List<String> existingVideoUrls = existingVideos.stream() .map(ArdMediaVideo::getSourceUrl) // 提取 ArdMedia 中的视频 URL .collect(Collectors.toList()); // 筛选出 photoList 中没有出现在 existingPhotos 中的新图片 List<BoxVideosReceiver> newVideos = videoList.stream() .filter(photo -> !existingVideoUrls.contains(photo.getImgUrl())) // 过滤出新视频 .collect(Collectors.toList()); if (!newVideos.isEmpty()) { // 插入新的视频记录 addVideos(flightTask, newVideos); // 更新任务的视频数量 flightTask.setVideoCount(flightTask.getVideoCount() + newVideos.size()); ardFlightTaskMapper.updateArdFlightTask(flightTask); } } } private void addPhotos(ArdFlightTask flightTask, List<BoxPhotosReceiver> photoList) { log.warn("开始添加图片"); for (BoxPhotosReceiver photo : photoList) { ArdMediaPic ardMedia = ArdMediaPic.builder() .id(photo.getPhotoId()) .taskId(flightTask.getTaskId()) .fileSize(photo.getFileSize()) .fileMd5(photo.getFileMd5()) .sourceUrl(photo.getImgUrl()) .longitude(photo.getLng()) .latitude(photo.getLat()) .height(photo.getHeight()) .captureTime(DateUtils.parseDate(photo.getTimeStamp())) .build(); ardMedia.setCreateTime(new Date()); ardMediaPicMapper.insertPic(ardMedia); } } private void addVideos(ArdFlightTask flightTask, List<BoxVideosReceiver> videoList) { for (BoxVideosReceiver video : videoList) { ArdMediaVideo ardMedia = ArdMediaVideo.builder() .id(video.getId()) .name(video.getName()) .taskId(flightTask.getTaskId()) // .fileSize(getFileSizeByUrl(video.getImgUrl())) .sourceUrl(video.getImgUrl()) .startTime(DateUtils.parseDate(video.getStartTime())) .endTime(DateUtils.parseDate(video.getEndTime())) .spanSeconds(Integer.parseInt(video.getSpanSeconds())) .build(); ardMedia.setCreateTime(new Date()); ardMediaVideoMapper.insertArdMediaVideo(ardMedia); } } // @Scheduled(fixedDelay = 10000) // 每10秒执行一次 // public void syncVideoMediaUrl() { // List<ArdMediaVideo> videos = ardMediaVideoMapper.selectVideosWithNullMedia(); // for (ArdMediaVideo video : videos) { // if (!uploadingVideos.add(video.getId())) { // continue; // } // CompletableFuture.runAsync(() -> { // try { // // 上传视频到 OSS // try (InputStream is = openUrlInputStream(video.getSourceUrl(), 5000, 10000)) { // if (is != null) { // String mediaUrl = ossServiceContext.upload( // "record", // video.getTaskId() + "/" + video.getId() + ".mp4", // is, // MimeTypeUtils.VIDEO_MP4 // ); // log.info("upload mediaUrl={}", mediaUrl); // video.setMediaUrl(mediaUrl); // } // } // // 生成缩略图 // processVideoThumbnail(video); // // // 更新数据库 // video.setUpdateTime(new Date()); // ardMediaVideoMapper.updateArdMediaVideo(video); // } catch (Exception e) { // log.error("生成 mediaUrl/缩略图失败 videoId={}", video.getId(), e); // } finally { // uploadingVideos.remove(video.getId()); // } // }, threadPoolTaskExecutor); // } // } // // @Scheduled(fixedDelay = 10000) // 每10秒执行一次 // public void syncPicMediaUrl() { // List<ArdMediaPic> pics = ardMediaPicMapper.selectPicsWithNullMedia(); // for (ArdMediaPic pic : pics) { // if (!uploadingPics.add(pic.getId())) { // continue; // } // CompletableFuture.runAsync(() -> { // try (InputStream inputStream = openUrlInputStream(pic.getSourceUrl(), 5000, 10000)) { // if (inputStream != null) { // String url = ossServiceContext.upload( // "pic", // pic.getTaskId() + "/" + pic.getId() + ".jpg", // inputStream, // MimeTypeUtils.IMAGE_JPG // ); // pic.setMediaUrl(url); // pic.setUpdateTime(new Date()); // ardMediaPicMapper.updatePic(pic); // } // } catch (Exception e) { // log.error("图片上传失败 taskId={}, photoId={}", pic.getTaskId(), pic.getId(), e); // } finally { // uploadingPics.remove(pic.getId()); // } // // }, threadPoolTaskExecutor); // 使用自定义线程池 // } // } }
