forked from uav/ard-uav-server
blame | history | patch | commit | commitdiff | ignore whitespace
liusuyi
2025-12-29 e7af0cdd75430765fcb880fa32391847eb6e615d 
 ard-work/src/main/java/com/ard/work/service/impl/ArdMediaServiceImpl.java


@@ -1,31 +1,227 @@


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 媒体数据集合


     */


    @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<ArdMedia> selectArdMediaList(ArdMedia ardMedia) {


        return ardMediaMapper.selectArdMedia(ardMedia);


    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);


        }


    }


}