package com.ruoyi.web.controller; import java.util.List; import javax.servlet.http.HttpServletResponse; import org.springframework.security.access.prepost.PreAuthorize; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.GetMapping; import org.springframework.web.bind.annotation.PostMapping; import org.springframework.web.bind.annotation.PutMapping; import org.springframework.web.bind.annotation.DeleteMapping; import org.springframework.web.bind.annotation.PathVariable; import org.springframework.web.bind.annotation.RequestBody; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.RestController; import com.ruoyi.common.annotation.Log; import com.ruoyi.common.core.controller.BaseController; import com.ruoyi.common.core.domain.AjaxResult; import com.ruoyi.common.enums.BusinessType; import com.ruoyi.system.domain.Routes; import com.ruoyi.system.service.IRoutesService; import com.ruoyi.common.utils.poi.ExcelUtil; import com.ruoyi.common.core.page.TableDataInfo; import com.ruoyi.system.domain.dto.PlatformStyleDTO; import com.alibaba.fastjson2.JSON; import com.alibaba.fastjson2.JSONArray; import com.alibaba.fastjson2.JSONObject; import org.springframework.data.redis.core.RedisTemplate; import org.springframework.beans.factory.annotation.Qualifier; import java.util.ArrayList; /** * 实体部署与航线Controller * * @author ruoyi * @date 2026-01-14 */ @RestController @RequestMapping("/system/routes") public class RoutesController extends BaseController { @Autowired private IRoutesService routesService; @Autowired private RedisTemplate redisTemplate; @Autowired @Qualifier("fourTRedisTemplate") private RedisTemplate fourTRedisTemplate; /** * 保存平台样式到 Redis */ @PreAuthorize("@ss.hasPermi('system:routes:edit')") @PostMapping("/savePlatformStyle") public AjaxResult savePlatformStyle(@RequestBody PlatformStyleDTO dto) { if (dto.getRoomId() == null || dto.getRouteId() == null || dto.getPlatformId() == null) { return AjaxResult.error("参数不完整"); } String key = "room:" + dto.getRoomId() + ":route:" + dto.getRouteId() + ":platforms"; redisTemplate.opsForHash().put(key, String.valueOf(dto.getPlatformId()), JSON.toJSONString(dto)); return success(); } /** * 从 Redis 获取平台样式 */ @PreAuthorize("@ss.hasPermi('system:routes:query')") @GetMapping("/getPlatformStyle") public AjaxResult getPlatformStyle(PlatformStyleDTO dto) { if (dto.getRoomId() == null || dto.getRouteId() == null || dto.getPlatformId() == null) { return AjaxResult.error("参数不完整"); } String key = "room:" + dto.getRoomId() + ":route:" + dto.getRouteId() + ":platforms"; Object val = redisTemplate.opsForHash().get(key, String.valueOf(dto.getPlatformId())); if (val != null) { return success(JSON.parseObject(val.toString(), PlatformStyleDTO.class)); } return success(); } /** * 保存4T数据到 Redis(按房间存储) */ @PreAuthorize("@ss.hasPermi('system:routes:edit')") @PostMapping("/save4TData") public AjaxResult save4TData(@RequestBody java.util.Map params) { Object roomId = params.get("roomId"); Object data = params.get("data"); if (roomId == null || data == null) { return AjaxResult.error("参数不完整"); } String key = "room:" + String.valueOf(roomId) + ":4t"; fourTRedisTemplate.opsForValue().set(key, data.toString()); return success(); } /** * 从 Redis 获取4T数据 */ @PreAuthorize("@ss.hasPermi('system:routes:query')") @GetMapping("/get4TData") public AjaxResult get4TData(Long roomId) { if (roomId == null) { return AjaxResult.error("房间ID不能为空"); } String key = "room:" + String.valueOf(roomId) + ":4t"; String val = fourTRedisTemplate.opsForValue().get(key); if (val != null && !val.isEmpty()) { try { return success(JSON.parseObject(val)); } catch (Exception e) { return success(val); } } return success(); } /** * 获取导弹发射参数列表(Redis,房间+航线+平台为 key,值为数组,每项含 angle/distance/launchTimeMinutesFromK/startLng/startLat/platformHeadingDeg) */ @PreAuthorize("@ss.hasPermi('system:routes:query')") @GetMapping("/missile-params") public AjaxResult getMissileParams(Long roomId, Long routeId, Long platformId) { if (roomId == null || routeId == null || platformId == null) { return AjaxResult.error("参数不完整"); } String key = "missile:params:" + roomId + ":" + routeId + ":" + platformId; Object val = redisTemplate.opsForValue().get(key); if (val == null) { return success(new ArrayList<>()); } String raw = val.toString(); if (raw.startsWith("[")) { return success(JSON.parseArray(raw)); } JSONObject single = JSON.parseObject(raw); List list = new ArrayList<>(); list.add(single); return success(list); } /** * 保存导弹发射参数到 Redis(追加一条记录,支持多次发射;每条含 angle/distance/launchTimeMinutesFromK/startLng/startLat/platformHeadingDeg) */ @PreAuthorize("@ss.hasPermi('system:routes:edit')") @PostMapping("/missile-params") public AjaxResult saveMissileParams(@RequestBody java.util.Map body) { Object roomId = body.get("roomId"); Object routeId = body.get("routeId"); Object platformId = body.get("platformId"); if (roomId == null || routeId == null || platformId == null) { return AjaxResult.error("参数不完整"); } String key = "missile:params:" + roomId + ":" + routeId + ":" + platformId; Object val = redisTemplate.opsForValue().get(key); JSONArray list = new JSONArray(); if (val != null) { String raw = val.toString(); if (raw.startsWith("[")) { list = JSON.parseArray(raw); } else { list.add(JSON.parseObject(raw)); } } list.add(body); redisTemplate.opsForValue().set(key, list.toJSONString()); return success(); } /** * 查询实体部署与航线列表 */ @PreAuthorize("@ss.hasPermi('system:routes:list')") @GetMapping("/list") public TableDataInfo list(Routes routes) { startPage(); List list = routesService.selectRoutesList(routes); return getDataTable(list); } /** * 导出实体部署与航线列表 */ @PreAuthorize("@ss.hasPermi('system:routes:export')") @Log(title = "实体部署与航线", businessType = BusinessType.EXPORT) @PostMapping("/export") public void export(HttpServletResponse response, Routes routes) { List list = routesService.selectRoutesList(routes); ExcelUtil util = new ExcelUtil(Routes.class); util.exportExcel(response, list, "实体部署与航线数据"); } /** * 获取实体部署与航线详细信息 */ @PreAuthorize("@ss.hasPermi('system:routes:query')") @GetMapping(value = "/{id}") public AjaxResult getInfo(@PathVariable("id") Long id) { return success(routesService.selectRoutesById(id)); } /** * 新增实体部署与航线 */ @PreAuthorize("@ss.hasPermi('system:routes:add')") @Log(title = "实体部署与航线", businessType = BusinessType.INSERT) @PostMapping public AjaxResult add(@RequestBody Routes routes) { // 1. 执行插入,MyBatis 会通过 useGeneratedKeys="true" 自动将新 ID 注入 routes 对象 int rows = routesService.insertRoutes(routes); // 2. 不要用 toAjax,直接返回 success 并带上 routes 对象 // 这样前端 response.data 就会包含这个带有 ID 的完整对象 return rows > 0 ? AjaxResult.success(routes) : AjaxResult.error("新增航线失败"); } /** * 修改实体部署与航线 */ @PreAuthorize("@ss.hasPermi('system:routes:edit')") @Log(title = "实体部署与航线", businessType = BusinessType.UPDATE) @PutMapping public AjaxResult edit(@RequestBody Routes routes) { return toAjax(routesService.updateRoutes(routes)); } /** * 删除实体部署与航线 */ @PreAuthorize("@ss.hasPermi('system:routes:remove')") @Log(title = "实体部署与航线", businessType = BusinessType.DELETE) @DeleteMapping("/{ids}") public AjaxResult remove(@PathVariable Long[] ids) { return toAjax(routesService.deleteRoutesByIds(ids)); } }