413 lines
18 KiB
Java
413 lines
18 KiB
Java
package com.ruoyi.device.controller;
|
||
|
||
import com.ruoyi.common.core.domain.R;
|
||
import com.ruoyi.common.core.web.controller.BaseController;
|
||
import com.ruoyi.device.api.domain.StatisticsVO;
|
||
import com.ruoyi.device.api.enums.AircraftStatusEnum;
|
||
import com.ruoyi.device.api.enums.DockStatusEnum;
|
||
import com.ruoyi.device.api.enums.PayloadStatusEnum;
|
||
import com.ruoyi.device.service.api.IAircraftService;
|
||
import com.ruoyi.device.service.impl.DaJiangBufferDeviceImpl;
|
||
import com.ruoyi.device.service.impl.TuohengBufferDeviceImpl;
|
||
import com.ruoyi.device.service.api.IDockService;
|
||
import com.ruoyi.device.domain.api.IDeviceDomain;
|
||
import com.ruoyi.device.domain.model.Device;
|
||
import com.ruoyi.device.service.api.IPayloadService;
|
||
import com.ruoyi.device.service.dto.AircraftDTO;
|
||
import com.ruoyi.device.service.dto.AircraftDetailDTO;
|
||
import com.ruoyi.device.service.dto.DockDTO;
|
||
import com.ruoyi.device.service.dto.DockDetailDTO;
|
||
import com.ruoyi.device.service.dto.PayloadDTO;
|
||
import org.slf4j.Logger;
|
||
import org.slf4j.LoggerFactory;
|
||
import org.springframework.beans.factory.annotation.Autowired;
|
||
import org.springframework.web.bind.annotation.GetMapping;
|
||
import org.springframework.web.bind.annotation.RequestMapping;
|
||
import org.springframework.web.bind.annotation.RestController;
|
||
|
||
import java.util.List;
|
||
import java.util.Map;
|
||
import java.util.stream.Collectors;
|
||
|
||
/**
|
||
* 统计Controller
|
||
*
|
||
* @author ruoyi
|
||
* @date 2026-01-23
|
||
*/
|
||
@RestController
|
||
@RequestMapping("/statistics")
|
||
public class StaticsController extends BaseController
|
||
{
|
||
private static final Logger log = LoggerFactory.getLogger(StaticsController.class);
|
||
|
||
@Autowired
|
||
private IDockService dockService;
|
||
|
||
@Autowired
|
||
private IAircraftService aircraftService;
|
||
|
||
@Autowired
|
||
private IPayloadService payloadService;
|
||
|
||
@Autowired
|
||
private DaJiangBufferDeviceImpl daJiangBufferDeviceService;
|
||
|
||
@Autowired
|
||
private TuohengBufferDeviceImpl tuohengBufferDeviceService;
|
||
|
||
@Autowired
|
||
private IDeviceDomain deviceDomain;
|
||
|
||
/**
|
||
* 获取系统统计信息（合并大疆和拓恒）
|
||
*
|
||
* @return 统计信息
|
||
*/
|
||
@GetMapping
|
||
public R<StatisticsVO> getStatistics()
|
||
{
|
||
log.info("========== 开始统计所有设备信息（大疆+拓恒） ==========");
|
||
|
||
// 获取大疆统计
|
||
StatisticsVO djiStats = buildDjiStatisticsVO();
|
||
|
||
// 获取拓恒统计
|
||
StatisticsVO thStats = buildThStatisticsVO();
|
||
|
||
// 合并统计结果
|
||
StatisticsVO totalStats = new StatisticsVO();
|
||
|
||
// 机场统计
|
||
totalStats.setDockCount(djiStats.getDockCount() + thStats.getDockCount());
|
||
totalStats.setIdleDockCount(djiStats.getIdleDockCount() + thStats.getIdleDockCount());
|
||
totalStats.setWorkingDockCount(djiStats.getWorkingDockCount() + thStats.getWorkingDockCount());
|
||
totalStats.setDebuggingDockCount(djiStats.getDebuggingDockCount() + thStats.getDebuggingDockCount());
|
||
totalStats.setOfflineDockCount(djiStats.getOfflineDockCount() + thStats.getOfflineDockCount());
|
||
|
||
// 无人机统计
|
||
totalStats.setAircraftCount(djiStats.getAircraftCount() + thStats.getAircraftCount());
|
||
totalStats.setPowerOnInCabinCount(djiStats.getPowerOnInCabinCount() + thStats.getPowerOnInCabinCount());
|
||
totalStats.setPowerOffInCabinCount(djiStats.getPowerOffInCabinCount() + thStats.getPowerOffInCabinCount());
|
||
totalStats.setInMissionCount(djiStats.getInMissionCount() + thStats.getInMissionCount());
|
||
totalStats.setDebuggingAircraftCount(djiStats.getDebuggingAircraftCount() + thStats.getDebuggingAircraftCount());
|
||
totalStats.setOfflineAircraftCount(djiStats.getOfflineAircraftCount() + thStats.getOfflineAircraftCount());
|
||
|
||
// 挂载统计
|
||
totalStats.setPayloadCount(djiStats.getPayloadCount() + thStats.getPayloadCount());
|
||
totalStats.setOfflinePayloadCount(djiStats.getOfflinePayloadCount() + thStats.getOfflinePayloadCount());
|
||
|
||
log.info("========== 所有设备统计完成 ==========");
|
||
log.info("总计: 机场={}, 任务中机场={}, 无人机={}, 任务中无人机={}",
|
||
totalStats.getDockCount(), totalStats.getWorkingDockCount(),
|
||
totalStats.getAircraftCount(), totalStats.getInMissionCount());
|
||
|
||
return R.ok(totalStats);
|
||
}
|
||
|
||
|
||
|
||
|
||
|
||
@GetMapping("/dji")
|
||
public R<StatisticsVO> getDjiStatistics()
|
||
{
|
||
return R.ok(buildDjiStatisticsVO());
|
||
}
|
||
|
||
|
||
@GetMapping("/th")
|
||
public R<StatisticsVO> getThStatistics()
|
||
{
|
||
return R.ok(buildThStatisticsVO());
|
||
}
|
||
|
||
private StatisticsVO buildDjiStatisticsVO (){
|
||
log.info("========== 开始统计DJI设备信息 ==========");
|
||
StatisticsVO vo = new StatisticsVO();
|
||
|
||
// 获取所有机场
|
||
List<DockDTO> allDocks = dockService.selectDockList(new DockDTO());
|
||
|
||
// 过滤出大疆机场
|
||
List<DockDTO> docks = filterDocksByManufacturer(allDocks, "dajiang");
|
||
vo.setDockCount(docks.size());
|
||
log.info("大疆机场总数: {}", vo.getDockCount());
|
||
|
||
// 批量获取机场详情 - 优化：从N次查询减少到1次批量查询
|
||
Map<Long, DockDetailDTO> dockDetailsMap = null;
|
||
if (!docks.isEmpty()) {
|
||
List<Long> dockIds = docks.stream()
|
||
.map(DockDTO::getDockId)
|
||
.collect(Collectors.toList());
|
||
dockDetailsMap = daJiangBufferDeviceService.getDockDetailsByIds(dockIds);
|
||
}
|
||
|
||
// 统计各状态机场数量
|
||
int idleCount = 0;
|
||
int workingCount = 0;
|
||
int debuggingCount = 0;
|
||
int offlineCount = 0;
|
||
|
||
if (docks != null && dockDetailsMap != null) {
|
||
log.info("---------- 开始统计机场状态 ----------");
|
||
for (DockDTO dock : docks) {
|
||
DockDetailDTO dockDetail = dockDetailsMap.get(dock.getDockId());
|
||
if (dockDetail != null && dockDetail.getDockStatus() != null) {
|
||
String status = dockDetail.getDockStatus();
|
||
log.info("机场[ID:{}, Name:{}] 状态: {}", dock.getDockId(), dock.getDockName(), status);
|
||
if (DockStatusEnum.IDLE.getCode().equalsIgnoreCase(status)) {
|
||
idleCount++;
|
||
log.debug(" -> 匹配到IDLE状态");
|
||
} else if (DockStatusEnum.WORKING.getCode().equalsIgnoreCase(status)) {
|
||
workingCount++;
|
||
log.info(" -> 匹配到WORKING状态 (任务中)");
|
||
} else if (DockStatusEnum.Debugging.getCode().equalsIgnoreCase(status)) {
|
||
debuggingCount++;
|
||
log.debug(" -> 匹配到Debugging状态");
|
||
} else {
|
||
offlineCount++;
|
||
log.debug(" -> 其他状态，归类为离线");
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
vo.setIdleDockCount(idleCount);
|
||
vo.setWorkingDockCount(workingCount);
|
||
vo.setDebuggingDockCount(debuggingCount);
|
||
vo.setOfflineDockCount(offlineCount);
|
||
|
||
log.info("机场状态统计结果 -> 空闲:{}, 任务中:{}, 调试:{}, 离线:{}", idleCount, workingCount, debuggingCount, offlineCount);
|
||
|
||
// 获取所有无人机
|
||
List<AircraftDTO> allAircrafts = aircraftService.selectAircraftList(new AircraftDTO());
|
||
|
||
// 过滤出大疆无人机
|
||
List<AircraftDTO> aircrafts = filterAircraftsByManufacturer(allAircrafts, "dajiang");
|
||
vo.setAircraftCount(aircrafts.size());
|
||
log.info("大疆无人机总数: {}", vo.getAircraftCount());
|
||
|
||
// 批量获取无人机详情 - 优化：从N次查询减少到1次批量查询
|
||
Map<Long, AircraftDetailDTO> aircraftDetailsMap = null;
|
||
if (!aircrafts.isEmpty()) {
|
||
List<Long> aircraftIds = aircrafts.stream()
|
||
.map(AircraftDTO::getAircraftId)
|
||
.collect(Collectors.toList());
|
||
aircraftDetailsMap = daJiangBufferDeviceService.getAircraftDetailsByIds(aircraftIds);
|
||
}
|
||
|
||
// 统计各状态无人机数量
|
||
int powerOnInCabinCount = 0;
|
||
int powerOffInCabinCount = 0;
|
||
int inMissionCount = 0;
|
||
int debuggingAircraftCount = 0;
|
||
int offlineAircraftCount = 0;
|
||
|
||
if (aircrafts != null && aircraftDetailsMap != null) {
|
||
log.info("---------- 开始统计无人机状态 ----------");
|
||
for (AircraftDTO aircraft : aircrafts) {
|
||
AircraftDetailDTO aircraftDetail = aircraftDetailsMap.get(aircraft.getAircraftId());
|
||
if (aircraftDetail != null && aircraftDetail.getAircraftStatus() != null) {
|
||
String status = aircraftDetail.getAircraftStatus();
|
||
log.info("无人机[ID:{}, Name:{}] 状态: {}", aircraft.getAircraftId(), aircraft.getAircraftName(), status);
|
||
if (AircraftStatusEnum.POWER_ON_IN_CABIN.getCode().equalsIgnoreCase(status)) {
|
||
powerOnInCabinCount++;
|
||
log.debug(" -> 匹配到舱内开机状态");
|
||
} else if (AircraftStatusEnum.POWER_OFF_IN_CABIN.getCode().equalsIgnoreCase(status)) {
|
||
powerOffInCabinCount++;
|
||
log.debug(" -> 匹配到舱内关机状态");
|
||
} else if (AircraftStatusEnum.IN_MISSION.getCode().equalsIgnoreCase(status)) {
|
||
inMissionCount++;
|
||
log.info(" -> 匹配到IN_MISSION状态 (任务中)");
|
||
} else if (AircraftStatusEnum.DEBUGGING.getCode().equalsIgnoreCase(status)) {
|
||
debuggingAircraftCount++;
|
||
log.debug(" -> 匹配到调试状态");
|
||
} else if (AircraftStatusEnum.OFFLINE.getCode().equalsIgnoreCase(status)) {
|
||
offlineAircraftCount++;
|
||
log.debug(" -> 匹配到离线状态");
|
||
} else {
|
||
offlineAircraftCount++;
|
||
log.debug(" -> 其他状态，归类为离线");
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
vo.setPowerOnInCabinCount(powerOnInCabinCount);
|
||
vo.setPowerOffInCabinCount(powerOffInCabinCount);
|
||
vo.setInMissionCount(inMissionCount);
|
||
vo.setDebuggingAircraftCount(debuggingAircraftCount);
|
||
vo.setOfflineAircraftCount(offlineAircraftCount);
|
||
|
||
log.info("无人机状态统计结果 -> 舱内开机:{}, 舱内关机:{}, 任务中:{}, 调试:{}, 离线:{}",
|
||
powerOnInCabinCount, powerOffInCabinCount, inMissionCount, debuggingAircraftCount, offlineAircraftCount);
|
||
|
||
// 获取所有挂载
|
||
List<PayloadDTO> payloads = payloadService.selectPayloadList(new PayloadDTO());
|
||
vo.setPayloadCount(payloads != null ? payloads.size() : 0);
|
||
|
||
// 统计离线挂载数量（暂时设置为0，因为挂载状态需要从实时数据获取）
|
||
vo.setOfflinePayloadCount(0);
|
||
|
||
log.info("========== DJI设备统计完成 ==========");
|
||
log.info("最终统计结果: 机场总数={}, 任务中机场={}, 无人机总数={}, 任务中无人机={}",
|
||
vo.getDockCount(), vo.getWorkingDockCount(), vo.getAircraftCount(), vo.getInMissionCount());
|
||
|
||
return vo;
|
||
}
|
||
|
||
/**
|
||
* 构建拓恒设备统计信息
|
||
*/
|
||
private StatisticsVO buildThStatisticsVO() {
|
||
log.info("========== 开始统计拓恒设备信息 ==========");
|
||
StatisticsVO vo = new StatisticsVO();
|
||
|
||
// 获取所有机场
|
||
List<DockDTO> allDocks = dockService.selectDockList(new DockDTO());
|
||
|
||
// 过滤出拓恒机场
|
||
List<DockDTO> thDocks = filterDocksByManufacturer(allDocks, "tuoheng");
|
||
vo.setDockCount(thDocks.size());
|
||
log.info("拓恒机场总数: {}", vo.getDockCount());
|
||
|
||
// 批量获取拓恒机场详情
|
||
Map<Long, DockDetailDTO> dockDetailsMap = null;
|
||
if (!thDocks.isEmpty()) {
|
||
List<Long> dockIds = thDocks.stream()
|
||
.map(DockDTO::getDockId)
|
||
.collect(Collectors.toList());
|
||
dockDetailsMap = tuohengBufferDeviceService.getDockDetailsByIds(dockIds);
|
||
}
|
||
|
||
// 统计各状态机场数量
|
||
int idleCount = 0;
|
||
int workingCount = 0;
|
||
int debuggingCount = 0;
|
||
int offlineCount = 0;
|
||
|
||
if (dockDetailsMap != null) {
|
||
log.info("---------- 开始统计拓恒机场状态 ----------");
|
||
for (DockDTO dock : thDocks) {
|
||
DockDetailDTO dockDetail = dockDetailsMap.get(dock.getDockId());
|
||
if (dockDetail != null && dockDetail.getDockStatus() != null) {
|
||
String status = dockDetail.getDockStatus();
|
||
log.info("拓恒机场[ID:{}, Name:{}] 状态: {}", dock.getDockId(), dock.getDockName(), status);
|
||
if (DockStatusEnum.IDLE.getCode().equalsIgnoreCase(status)) {
|
||
idleCount++;
|
||
} else if (DockStatusEnum.WORKING.getCode().equalsIgnoreCase(status)) {
|
||
workingCount++;
|
||
} else if (DockStatusEnum.Debugging.getCode().equalsIgnoreCase(status)) {
|
||
debuggingCount++;
|
||
} else {
|
||
offlineCount++;
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
vo.setIdleDockCount(idleCount);
|
||
vo.setWorkingDockCount(workingCount);
|
||
vo.setDebuggingDockCount(debuggingCount);
|
||
vo.setOfflineDockCount(offlineCount);
|
||
|
||
log.info("拓恒机场状态统计 -> 空闲:{}, 任务中:{}, 调试:{}, 离线:{}",
|
||
idleCount, workingCount, debuggingCount, offlineCount);
|
||
|
||
// 获取所有无人机
|
||
List<AircraftDTO> allAircrafts = aircraftService.selectAircraftList(new AircraftDTO());
|
||
|
||
// 过滤出拓恒无人机
|
||
List<AircraftDTO> thAircrafts = filterAircraftsByManufacturer(allAircrafts, "tuoheng");
|
||
vo.setAircraftCount(thAircrafts.size());
|
||
log.info("拓恒无人机总数: {}", vo.getAircraftCount());
|
||
|
||
// 批量获取拓恒无人机详情
|
||
Map<Long, AircraftDetailDTO> aircraftDetailsMap = null;
|
||
if (!thAircrafts.isEmpty()) {
|
||
List<Long> aircraftIds = thAircrafts.stream()
|
||
.map(AircraftDTO::getAircraftId)
|
||
.collect(Collectors.toList());
|
||
aircraftDetailsMap = tuohengBufferDeviceService.getAircraftDetailsByIds(aircraftIds);
|
||
}
|
||
|
||
// 统计各状态无人机数量
|
||
int powerOnInCabinCount = 0;
|
||
int powerOffInCabinCount = 0;
|
||
int inMissionCount = 0;
|
||
int debuggingAircraftCount = 0;
|
||
int offlineAircraftCount = 0;
|
||
|
||
if (aircraftDetailsMap != null) {
|
||
log.info("---------- 开始统计拓恒无人机状态 ----------");
|
||
for (AircraftDTO aircraft : thAircrafts) {
|
||
AircraftDetailDTO aircraftDetail = aircraftDetailsMap.get(aircraft.getAircraftId());
|
||
if (aircraftDetail != null && aircraftDetail.getAircraftStatus() != null) {
|
||
String status = aircraftDetail.getAircraftStatus();
|
||
log.info("拓恒无人机[ID:{}, Name:{}] 状态: {}",
|
||
aircraft.getAircraftId(), aircraft.getAircraftName(), status);
|
||
if (AircraftStatusEnum.POWER_ON_IN_CABIN.getCode().equalsIgnoreCase(status)) {
|
||
powerOnInCabinCount++;
|
||
} else if (AircraftStatusEnum.POWER_OFF_IN_CABIN.getCode().equalsIgnoreCase(status)) {
|
||
powerOffInCabinCount++;
|
||
} else if (AircraftStatusEnum.IN_MISSION.getCode().equalsIgnoreCase(status)) {
|
||
inMissionCount++;
|
||
} else if (AircraftStatusEnum.DEBUGGING.getCode().equalsIgnoreCase(status)) {
|
||
debuggingAircraftCount++;
|
||
} else {
|
||
offlineAircraftCount++;
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
vo.setPowerOnInCabinCount(powerOnInCabinCount);
|
||
vo.setPowerOffInCabinCount(powerOffInCabinCount);
|
||
vo.setInMissionCount(inMissionCount);
|
||
vo.setDebuggingAircraftCount(debuggingAircraftCount);
|
||
vo.setOfflineAircraftCount(offlineAircraftCount);
|
||
|
||
log.info("拓恒无人机状态统计 -> 舱内开机:{}, 舱内关机:{}, 任务中:{}, 调试:{}, 离线:{}",
|
||
powerOnInCabinCount, powerOffInCabinCount, inMissionCount,
|
||
debuggingAircraftCount, offlineAircraftCount);
|
||
|
||
// 挂载统计（拓恒设备暂时设置为0）
|
||
vo.setPayloadCount(0);
|
||
vo.setOfflinePayloadCount(0);
|
||
|
||
log.info("========== 拓恒设备统计完成 ==========");
|
||
return vo;
|
||
}
|
||
|
||
/**
|
||
* 根据厂商过滤机场列表
|
||
*/
|
||
private List<DockDTO> filterDocksByManufacturer(List<DockDTO> docks, String manufacturer) {
|
||
if (docks == null || docks.isEmpty()) {
|
||
return List.of();
|
||
}
|
||
return docks.stream()
|
||
.filter(dock -> {
|
||
Device device = deviceDomain.selectDeviceByDeviceId(dock.getDeviceId());
|
||
return device != null && manufacturer.equals(device.getDeviceManufacturer());
|
||
})
|
||
.collect(Collectors.toList());
|
||
}
|
||
|
||
/**
|
||
* 根据厂商过滤无人机列表
|
||
*/
|
||
private List<AircraftDTO> filterAircraftsByManufacturer(List<AircraftDTO> aircrafts, String manufacturer) {
|
||
if (aircrafts == null || aircrafts.isEmpty()) {
|
||
return List.of();
|
||
}
|
||
return aircrafts.stream()
|
||
.filter(aircraft -> {
|
||
Device device = deviceDomain.selectDeviceByDeviceId(aircraft.getDeviceId());
|
||
return device != null && manufacturer.equals(device.getDeviceManufacturer());
|
||
})
|
||
.collect(Collectors.toList());
|
||
}
|
||
}
|