~nishiwei/ard-uav-server.git

			@@ -18,7 +18,7 @@
			import com.ard.work.service.IArdMediaService;
			import com.ard.work.util.osSelect;
			import lombok.extern.slf4j.Slf4j;
			import org.springframework.scheduling.annotation.Async;
			import org.springframework.scheduling.annotation.Scheduled;
			import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
			import org.springframework.stereotype.Service;

			@@ -27,18 +27,20 @@
			import javax.net.ssl.SSLContext;
			import javax.net.ssl.TrustManager;
			import javax.net.ssl.X509TrustManager;
			import java.io.ByteArrayInputStream;
			import java.io.ByteArrayOutputStream;
			import java.io.IOException;
			import java.io.InputStream;
			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;

			/**
			@@ -62,6 +64,9 @@
			private ArdFlightTaskMapper ardFlightTaskMapper;
			@Resource
			private ThreadPoolTaskExecutor threadPoolTaskExecutor;

			private final Set<Long> uploadingVideos = ConcurrentHashMap.newKeySet();
			private final Set<Long> uploadingPics = ConcurrentHashMap.newKeySet();

			// 调用接口获取架次图片
			private List<BoxPhotosReceiver> getBoxPhotos(String serverId, BoxPhotosDTO boxPhotosDTO) {
			@@ -114,9 +119,8 @@
			}
			}


			@Override
			public void updatePhotoToDatabase(ArdFlightTask flightTask, String serverId, String boxSn) {
			public void syncPicToDatabase(ArdFlightTask flightTask, String serverId, String boxSn) {

			// 获取本地数据库中已存在的图片记录
			List<ArdMediaPic> existingPhotos = ardMediaPicMapper.selectPicByTaskId(flightTask.getTaskId());
			@@ -151,9 +155,8 @@
			}
			}


			@Override
			public void updateVideoToDatabase(ArdFlightTask flightTask, String serverId, String boxSn) {
			public void syncVideoToDatabase(ArdFlightTask flightTask, String serverId, String boxSn) {

			// 获取本地数据库中已存在的视频记录
			List<ArdMediaVideo> existingVideos = ardMediaVideoMapper.selectArdMediaVideoByTaskId(flightTask.getTaskId());
			@@ -193,9 +196,7 @@

			private void addPhotos(ArdFlightTask flightTask, List<BoxPhotosReceiver> photoList) {
			log.warn("开始添加图片");
			String boxSn = flightTask.getBoxSn();
			for (BoxPhotosReceiver photo : photoList) {
			CompletableFuture.runAsync(() -> {
			ArdMediaPic ardMedia = ArdMediaPic.builder()
			.id(photo.getPhotoId())
			.taskId(flightTask.getTaskId())
			@@ -207,159 +208,135 @@
			.height(photo.getHeight())
			.captureTime(DateUtils.parseDate(photo.getTimeStamp()))
			.build();

			String url = "";
			try (InputStream inputStream = openUrlInputStream(photo.getImgUrl(), 5000, 10000)) {
			if (inputStream != null) {
			url = ossServiceContext.upload(
			"pic",
			boxSn + "/" + ardMedia.getId() + ".jpg",
			inputStream,
			MimeTypeUtils.IMAGE_JPG
			);
			}
			} catch (Exception e) {
			log.error("图片上传失败 taskId={}, photoId={}", flightTask.getTaskId(), photo.getPhotoId(), e);
			}

			ardMedia.setMediaUrl(url);
			ardMedia.setCreateTime(new Date());
			ardMediaPicMapper.insertPic(ardMedia);

			}, threadPoolTaskExecutor); // 使用自定义线程池
			}
			}

			private void addVideos(ArdFlightTask flightTask, List<BoxVideosReceiver> videoList) {
			String boxSn = flightTask.getBoxSn();

			for (BoxVideosReceiver video : videoList) {
			CompletableFuture.runAsync(() -> {
			ArdMediaVideo ardMedia = ArdMediaVideo.builder()
			.id(video.getId())
			.name(video.getName())
			.taskId(flightTask.getTaskId())
			.fileSize(getFileSizeByUrl(video.getImgUrl()))
			// .fileSize(getFileSizeByUrl(video.getImgUrl()))
			.sourceUrl(video.getImgUrl())
			.startTime(DateUtils.parseDate(video.getStartTime()))
			.endTime(DateUtils.parseDate(video.getEndTime()))
			.spanSeconds(Integer.parseInt(video.getSpanSeconds()))
			.build();

			// 视频上传
			String videoUrl = null;
			try (InputStream videoStream = openUrlInputStream(ardMedia.getSourceUrl(), 5000, 10000)) {
			if (videoStream != null) {
			videoUrl = ossServiceContext.upload("record",
			boxSn + "/" + ardMedia.getId() + ".mp4",
			videoStream,
			MimeTypeUtils.VIDEO_MP4);
			}
			} catch (Exception e) {
			log.error("视频上传失败 taskId={}, videoId={}", flightTask.getTaskId(), video.getId(), e);
			}

			// 缩略图上传（失败不影响视频入库）
			String thumbUrl = null;
			try (ByteArrayInputStream thumbStream = generateThumbnail(ardMedia.getSourceUrl(), 1, 200)) {
			if (thumbStream != null) {
			thumbUrl = ossServiceContext.upload("record",
			boxSn + "/" + ardMedia.getId() + ".jpg",
			thumbStream,
			MimeTypeUtils.IMAGE_JPEG);
			}
			} catch (Exception e) {
			log.warn("缩略图生成或上传失败 taskId={}, videoId={}", flightTask.getTaskId(), video.getId(), e);
			}

			// 入库（即使缩略图为空也入库）
			ardMedia.setMediaUrl(videoUrl);
			ardMedia.setThumbnailUrl(thumbUrl);
			ardMedia.setCreateTime(new Date());
			ardMediaVideoMapper.insertArdMediaVideo(ardMedia);

			}, threadPoolTaskExecutor);
			}
			}

			/**
			* 处理视频缩略图
			*/
			private void processVideoThumbnail(ArdMediaVideo video) {
			// 生成缩略图
			try (ByteArrayInputStream in = generateThumbnail(video.getSourceUrl())) {
			String thumbUrl = ossServiceContext.upload(
			"record",
			video.getTaskId() + "/" + video.getId() + ".jpg",
			in,
			MimeTypeUtils.IMAGE_JPEG
			);
			video.setThumbnailUrl(thumbUrl);
			} catch (Exception e) {
			log.warn("视频缩略图生成或上传失败 taskId={}, videoId={}", video.getTaskId(), video.getId(), e);
			}
			// 获取文件大小
			private long getFileSizeByUrl(String fileUrl) {
			try {
			long videoSize = getFileSizeFromUrl(video.getSourceUrl());
			video.setFileSize(videoSize);
			} catch (Exception e) {
			log.warn("获取视频大小失败 url={}", video.getSourceUrl(), e);
			}
			// 更新数据库
			video.setUpdateTime(new Date());
			ardMediaVideoMapper.updateArdMediaVideo(video);
			}

			public long getFileSizeFromUrl(String fileUrl) throws IOException {
			URL url = new URL(fileUrl);
			HttpURLConnection conn = (HttpURLConnection) url.openConnection();
			conn.setRequestMethod("HEAD");
			conn.setConnectTimeout(5000);
			conn.setReadTimeout(5000);
			conn.connect();
			return conn.getContentLengthLong(); // 返回字节大小
			} catch (Exception e) {
			return -1;
			}
			long size = conn.getContentLengthLong(); // 返回字节数
			conn.disconnect();
			return size;
			}

			// 生成视频缩略图
			public ByteArrayInputStream generateThumbnail(String videoFile, int skipSeconds, int targetWidth) {
			String ffmpegPath = "";
			if (osSelect.isWindows()) {
			ffmpegPath = "ffmpeg";
			} else if (osSelect.isLinux()) {
			ffmpegPath = "/usr/local/ffmpeg/bin/ffmpeg";
			}

			private ByteArrayInputStream generateThumbnail(String videoFile) {

			String ffmpeg = osSelect.isWindows()
			? "ffmpeg"
			: "/usr/local/ffmpeg/bin/ffmpeg";

			Path tempImg = null;
			try {
			List<String> command = new ArrayList<>();
			command.add(ffmpegPath);
			tempImg = Files.createTempFile("thumb_", ".jpg");

			command.add("-timeout");
			command.add("3000000");
			command.add("-rw_timeout");
			command.add("3000000");
			List<String> cmd = new ArrayList<>();
			cmd.add(ffmpeg);
			cmd.add("-y");
			cmd.add("-loglevel");
			cmd.add("error");
			cmd.add("-i");
			cmd.add(videoFile);
			cmd.add("-ss");
			cmd.add(String.valueOf(1));
			cmd.add("-vframes");
			cmd.add("1");
			cmd.add("-vf");
			cmd.add("scale=" + 200 + ":-1");
			cmd.add(tempImg.toAbsolutePath().toString());

			command.add("-ss");
			command.add(String.format("00:00:%02d", skipSeconds));
			command.add("-i");
			command.add(videoFile);
			command.add("-vframes");
			command.add("1");
			log.info("ffmpeg cmd: {}", String.join(" ", cmd));

			// 等比缩放
			command.add("-vf");
			command.add("scale=" + targetWidth + ":-1");
			Process process = new ProcessBuilder(cmd).start();

			// 确保输出 JPEG
			command.add("-f");
			command.add("mjpeg"); // 关键点：输出格式固定为 JPEG
			command.add("-vcodec");
			command.add("mjpeg");
			drain(process.getErrorStream());

			// 输出到管道
			command.add("pipe:1");
			boolean finished = process.waitFor(15, TimeUnit.SECONDS); // 超时保护
			if (!finished) {
			process.destroyForcibly();
			throw new RuntimeException("ffmpeg timeout");
			}

			// ⭐ 打印最终命令（可复制）
			log.info("ffmpeg command: {}", String.join(" ", command));
			if (process.exitValue() != 0) {
			throw new RuntimeException("ffmpeg exit=" + process.exitValue());
			}

			ProcessBuilder pb = new ProcessBuilder(command);
			pb.redirectErrorStream(false);
			Process process = pb.start();
			byte[] data = Files.readAllBytes(tempImg);
			return new ByteArrayInputStream(data);

			// 读取 FFmpeg 输出（JPEG 数据）到内存
			ByteArrayOutputStream baos = new ByteArrayOutputStream();
			try (InputStream is = process.getInputStream()) {
			byte[] buffer = new byte[8192];
			int len;
			while ((len = is.read(buffer)) != -1) {
			baos.write(buffer, 0, len);
			} catch (Exception e) {
			throw new RuntimeException("生成缩略图失败", e);
			} finally {
			if (tempImg != null) {
			try {
			Files.deleteIfExists(tempImg);
			} catch (IOException ignore) {
			}
			}
			}
			}

			int exitCode = process.waitFor();
			if (exitCode != 0) {
			throw new RuntimeException("FFmpeg 执行失败，退出码: " + exitCode);
			private static void drain(InputStream is) {
			new Thread(() -> {
			try (BufferedReader br = new BufferedReader(new InputStreamReader(is))) {
			while (br.readLine() != null) {
			}
			} catch (IOException ignore) {
			}
			}, "ffmpeg-stderr").start();
			}

			return new ByteArrayInputStream(baos.toByteArray());

			} catch (IOException \| InterruptedException e) {
			throw new RuntimeException("FFmpeg 生成缩略图失败", e);
			}
			}

			/**
			* 将 URL 转成 InputStream
			@@ -408,4 +385,72 @@
			return null;
			}
			}

			@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); // 使用自定义线程池
			}
			}

			}