From c00093644d9e0bd126f23cd826982f09e55c2051 Mon Sep 17 00:00:00 2001
From: liusuyi <1951119284@qq.com>
Date: Fri, 19 Dec 2025 16:56:41 +0800
Subject: [PATCH] 优化飞行状态

---
 ard-work/src/main/java/com/ard/work/service/impl/ArdFlightTaskPartServiceImpl.java |   35 ++++++++++++++++++++++-------------
 1 files changed, 22 insertions(+), 13 deletions(-)

diff --git a/ard-work/src/main/java/com/ard/work/service/impl/ArdFlightTaskPartServiceImpl.java b/ard-work/src/main/java/com/ard/work/service/impl/ArdFlightTaskPartServiceImpl.java
index d6fb8c5..9985e7e 100644
--- a/ard-work/src/main/java/com/ard/work/service/impl/ArdFlightTaskPartServiceImpl.java
+++ b/ard-work/src/main/java/com/ard/work/service/impl/ArdFlightTaskPartServiceImpl.java
@@ -1,7 +1,10 @@
 package com.ard.work.service.impl;
 
 import cn.hutool.core.util.SerializeUtil;
+import com.ard.work.domain.cache.UavStateCache;
 import com.ard.work.domain.cache.UavTelemetryDataCache;
+import com.ard.work.domain.dto.UavFlightStateDTO;
+import com.ard.work.domain.dto.state.UavState;
 import com.ard.work.domain.dto.telemetry.TelemetryData;
 import com.ard.work.domain.entity.ArdFlightTaskPart;
 import com.ard.work.mapper.ArdFlightTaskPartMapper;
@@ -11,8 +14,6 @@
 import org.springframework.scheduling.annotation.Async;
 import org.springframework.scheduling.annotation.Scheduled;
 import org.springframework.stereotype.Service;
-import java.time.LocalDateTime;
-import java.time.temporal.ChronoUnit;
 import java.util.ArrayList;
 import java.util.Date;
 import java.util.List;
@@ -30,7 +31,7 @@
     /**
      * 缓存:boxSn -> 当前分片数据
      */
-    private final ConcurrentHashMap<String, List<TelemetryData>> cacheMap = new ConcurrentHashMap<>();
+    private final ConcurrentHashMap<String, List<UavFlightStateDTO>> cacheMap = new ConcurrentHashMap<>();
 
     /**
      * 分片保存阈值条数
@@ -44,11 +45,19 @@
     @Scheduled(fixedRate = 1000)
     public void readTelemetryTask() {
         UavTelemetryDataCache.forEach((boxSn, telemetryData) -> {
-            // 只记录相对高度大于1米的数据
-            float relativeHeight = telemetryData.getUltrasonic(); // 或者 telemetryData.getRelativeHeight(),根据你的字段
-            if (relativeHeight > 1) {
-                addTelemetryData(telemetryData);
+
+            // 从状态缓存里取当前状态
+            UavState uavState = UavStateCache.get(boxSn);
+            if (uavState == null) {
+                return; // 状态不存在,不记录
             }
+
+            UavFlightStateDTO dto = new UavFlightStateDTO(
+                    uavState,
+                    telemetryData
+            );
+
+            addTelemetryData(dto);
         });
     }
 
@@ -56,10 +65,10 @@
      * 添加一条遥测数据到缓存
      */
     @Override
-    public void addTelemetryData(TelemetryData telemetryData) {
-        String boxSn = telemetryData.getBoxSn();
-        List<TelemetryData> list = cacheMap.computeIfAbsent(boxSn, k -> new ArrayList<>());
-        list.add(telemetryData);
+    public void addTelemetryData(UavFlightStateDTO uavFlightStateDTO) {
+        String boxSn = uavFlightStateDTO.getTelemetryData().getBoxSn();
+        List<UavFlightStateDTO> list = cacheMap.computeIfAbsent(boxSn, k -> new ArrayList<>());
+        list.add(uavFlightStateDTO);
 
         // 当缓存条数达到阈值 或 时间间隔超过 CHUNK_INTERVAL 时,分片保存
         if (list.size() >= CHUNK_SIZE) {
@@ -72,7 +81,7 @@
      */
     @Async
     public void saveChunk(String boxSn) {
-        List<TelemetryData> dataList = cacheMap.getOrDefault(boxSn, new ArrayList<>());
+        List<UavFlightStateDTO> dataList = cacheMap.getOrDefault(boxSn, new ArrayList<>());
         if (dataList.isEmpty()) {
             return;
         }
@@ -104,7 +113,7 @@
      */
     @Override
     public void flushRemaining(String boxSn) {
-        List<TelemetryData> dataList = cacheMap.getOrDefault(boxSn, new ArrayList<>());
+        List<UavFlightStateDTO> dataList = cacheMap.getOrDefault(boxSn, new ArrayList<>());
         if (!dataList.isEmpty()) {
             saveChunk(boxSn);
         }

--
Gitblit v1.9.3