package com.dji.sample.droneairport.service.impl;
|
|
|
import cn.hutool.json.JSONArray;
|
import cn.hutool.json.JSONObject;
|
//import com.alibaba.fastjson.JSON;
|
import cn.hutool.json.JSONUtil;
|
import com.alibaba.fastjson.JSON;
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import com.dji.sample.common.model.ResponseResult;
|
import com.dji.sample.component.redis.RedisOpsUtils;
|
import com.dji.sample.droneairport.dao.DbUploadMapper;
|
import com.dji.sample.droneairport.dao.DeviceExpanSionMapper;
|
import com.dji.sample.droneairport.model.Entity.DbUploadEntity;
|
import com.dji.sample.droneairport.model.Entity.DeviceExpanSionEntity;
|
import com.dji.sample.droneairport.model.dto.*;
|
import com.dji.sample.droneairport.model.param.ReturnTaskParam;
|
import com.dji.sample.droneairport.service.RegistService;
|
import com.dji.sample.droneairport.utils.AuthUtil;
|
import com.dji.sample.droneairport.utils.SM2.Utils;
|
import com.dji.sample.droneairport.utils.SM4Util;
|
import com.dji.sample.manage.dao.IDeviceMapper;
|
import com.dji.sample.manage.model.entity.DeviceEntity;
|
import com.dji.sample.manage.model.enums.DeviceModeCodeEnum;
|
import com.dji.sample.manage.model.enums.DockModeCodeEnum;
|
import com.dji.sample.manage.model.receiver.OsdDockReceiver;
|
import com.dji.sample.manage.model.receiver.OsdSubDeviceReceiver;
|
import com.dji.sample.manage.service.IDeviceRedisService;
|
import com.dji.sample.patches.dao.GetPatchesMapper;
|
import com.dji.sample.patches.dao.TaskInfoMapper;
|
import com.dji.sample.patches.model.entity.LotInfo;
|
import com.dji.sample.patches.model.entity.TaskInfo;
|
import com.dji.sample.patches.utils.TimerUtil;
|
import com.dji.sample.wayline.dao.IWaylineFileMapper;
|
import com.dji.sample.wayline.dao.IWaylineJobMapper;
|
import com.dji.sample.wayline.model.entity.WaylineFileEntity;
|
import com.dji.sample.wayline.model.entity.WaylineJobEntity;
|
import com.dji.sample.wayline.model.enums.WaylineJobStatusEnum;
|
import com.fasterxml.jackson.core.JsonProcessingException;
|
import com.fasterxml.jackson.databind.ObjectMapper;
|
import org.bouncycastle.crypto.InvalidCipherTextException;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.http.*;
|
import org.springframework.stereotype.Service;
|
import org.springframework.web.client.RestTemplate;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import java.io.File;
|
import java.io.IOException;
|
import java.security.NoSuchAlgorithmException;
|
import java.time.*;
|
import java.util.*;
|
import java.util.stream.Collectors;
|
|
import static com.dji.sample.droneairport.utils.TaskAllocator.assignTasks;
|
import static com.dji.sample.patches.utils.TimerUtil.getCurrentTimestampPlus8Hours;
|
import static com.dji.sample.patches.utils.TimerUtil.getNextDayEightAMTimestamp;
|
|
@Service
|
public class RegistServiceImpl implements RegistService {
|
private final RestTemplate restTemplate = new RestTemplate();
|
private final ObjectMapper objectMapper = new ObjectMapper();
|
@Autowired
|
private IDeviceRedisService redisService;
|
@Autowired
|
private DeviceExpanSionMapper expanSionMapper;
|
@Autowired
|
private TaskInfoMapper taskInfoMapper;
|
@Autowired
|
private GetPatchesMapper patchesMapper;
|
@Autowired
|
private TimerUtil timerUtil;
|
@Autowired
|
private IDeviceMapper deviceMapper;
|
@Autowired
|
private IWaylineJobMapper waylineJobMapper;
|
@Autowired
|
private IWaylineFileMapper fileMapper;
|
@Autowired
|
private DbUploadMapper dbUploadMapper;
|
|
public String registPort(DeviceExpanSionEntity airport) {
|
try {
|
RegistDto registDto = airportEntityToDto(airport);
|
String jsonBody = buildRequestBody(registDto);
|
String base64 = SM4Util.encrypt("jsimjrby3wqb7dbq", jsonBody);
|
// 设置请求头
|
HttpHeaders headers = new HttpHeaders();
|
headers.setContentType(MediaType.APPLICATION_JSON);
|
headers.setAccept(List.of(MediaType.APPLICATION_JSON, MediaType.TEXT_PLAIN, MediaType.ALL));
|
headers.set("Accept-Language", "zh-CN,zh;q=0.9");
|
headers.setConnection("keep-alive");
|
headers.set("x-auth-token", "eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJ3b3Jrc3BhY2VfaWQiOiJlM2RlYTBmNS0zN2YyLTRkNzktYWU1OC00OTBhZjMyMjgwNjkiLCJzdWIiOiJDbG91ZEFwaVNhbXBsZSIsInVzZXJfdHlwZSI6IjEiLCJuYmYiOjE3MjA2NTg3NDMsImxvZyI6IkxvZ2dlcltjb20uZGppLnNhbXBsZS5jb21tb24ubW9kZWwuQ3VzdG9tQ2xhaW1dIiwiaXNzIjoiREpJIiwiaWQiOiI5Y2MwMGY3Zi0yODMwLTRmM2UtYjdmNC1lN2E1ZjIzYjNjNGYiLCJleHAiOjE4MDcwNTg3NDMsImlhdCI6MTcyMDY1ODc0MywidXNlcm5hbWUiOiJhZG1pbkRQIn0.CrRFYUTK357HleqrrfL7nLdwFJw5Bcv3ms33n_Or8eM");
|
headers.set("x-lc-token", "");
|
// 构建请求实体
|
HttpEntity<String> requestEntity = new HttpEntity<>(base64, headers);
|
// 发送请求
|
ResponseEntity<String> response = restTemplate.exchange(
|
"http://localhost:6789/droneAirport/test",
|
HttpMethod.POST,
|
requestEntity,
|
String.class);
|
return response.getBody();
|
|
} catch (Exception e) {
|
// 异常处理
|
throw new IllegalArgumentException("无人机机场注册失败: " + e.getMessage());
|
}
|
}
|
|
// private String buildRequestBody(RegistDto dto) {
|
// try {
|
// return objectMapper.writeValueAsString(dto);
|
// } catch (JsonProcessingException e) {
|
// throw new RuntimeException("数据有误", e);
|
// }
|
// }
|
|
public <T> String buildRequestBody(T obj) {
|
try {
|
return objectMapper.writeValueAsString(obj);
|
} catch (JsonProcessingException e) {
|
throw new RuntimeException("对象转换为JSON字符串时发生错误", e);
|
}
|
}
|
|
@Override
|
public String addDrone(String workspaceId) {
|
try {
|
DeviceExpanSionEntity airport = expanSionMapper.selectOne(new LambdaQueryWrapper<DeviceExpanSionEntity>()
|
.eq(DeviceExpanSionEntity::getWorkspaceId, workspaceId)
|
.eq(DeviceExpanSionEntity::getType, 0));
|
DeviceExpanSionEntity drone = expanSionMapper.selectOne(new LambdaQueryWrapper<DeviceExpanSionEntity>()
|
.eq(DeviceExpanSionEntity::getWorkspaceId, workspaceId)
|
.eq(DeviceExpanSionEntity::getType, 1));
|
AddDeviceDto dto = deviceEntityToDto(drone);
|
// 构建请求体
|
String jsonBody = buildRequestBody(dto);
|
String base64 = SM4Util.encrypt("jsimjrby3wqb7dbq", jsonBody);
|
// 设置请求头
|
|
HttpHeaders headers = new HttpHeaders();
|
headers.setContentType(MediaType.APPLICATION_JSON);
|
headers.setAccept(List.of(MediaType.APPLICATION_JSON, MediaType.TEXT_PLAIN, MediaType.ALL));
|
headers.set("Accept-Language", "zh-CN,zh;q=0.9");
|
headers.setConnection("keep-alive");
|
headers.set("x-auth-token", "eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJ3b3Jrc3BhY2VfaWQiOiJlM2RlYTBmNS0zN2YyLTRkNzktYWU1OC00OTBhZjMyMjgwNjkiLCJzdWIiOiJDbG91ZEFwaVNhbXBsZSIsInVzZXJfdHlwZSI6IjEiLCJuYmYiOjE3MjA2NTg3NDMsImxvZyI6IkxvZ2dlcltjb20uZGppLnNhbXBsZS5jb21tb24ubW9kZWwuQ3VzdG9tQ2xhaW1dIiwiaXNzIjoiREpJIiwiaWQiOiI5Y2MwMGY3Zi0yODMwLTRmM2UtYjdmNC1lN2E1ZjIzYjNjNGYiLCJleHAiOjE4MDcwNTg3NDMsImlhdCI6MTcyMDY1ODc0MywidXNlcm5hbWUiOiJhZG1pbkRQIn0.CrRFYUTK357HleqrrfL7nLdwFJw5Bcv3ms33n_Or8eM");
|
headers.set("x-lc-token", "");
|
// 构建请求实体
|
HttpEntity<String> requestEntity = new HttpEntity<>(base64, headers);
|
// 发送请求
|
ResponseEntity<String> response = restTemplate.exchange(
|
"http://localhost:6789/droneAirport/test",
|
HttpMethod.POST,
|
requestEntity,
|
String.class);
|
registPort(airport);
|
return response.getBody();
|
|
} catch (Exception e) {
|
// 异常处理
|
throw new IllegalArgumentException("无人机机场设备注册失败: " + e.getMessage());
|
}
|
}
|
|
@Override
|
public String getDroneState(String deviceId) throws Exception {
|
DroneStateDto data = getAirportFromCache(deviceId);
|
String json = JSON.toJSONString(data);
|
String SM4Screct = SM4Util.encrypt("jsimjrby3wqb7dbq", json);
|
return SM4Screct;
|
}
|
|
// @Override
|
// public ResponseResult authorization(String base64, String Secret) throws Exception {
|
// String sm4Secrect = deSM2(Secret, "23E57DA1E4AB865CCBC325B668762207DEF74345B782237808AE0BABDF26734D");
|
// String decryptedJson = SM4Util.decrypt(sm4Secrect, base64);
|
// JSONObject jsonObject = new JSONObject(decryptedJson);
|
// String deviceId = jsonObject.getStr("deviceid");
|
// boolean online = redisService.checkDeviceOnline(deviceId);
|
// if (online) {
|
// return ResponseResult.success(200, "无人机机场授权成功！", "", "");
|
// }
|
// return ResponseResult.error(500, "无人机机场授权失败！");
|
// }
|
@Override
|
public void noAuthorization(String sn) {
|
AuthorDto dto = new AuthorDto();
|
dto.setSn(sn);
|
dto.setStatus(0);
|
String jsonBody = buildRequestBody(dto);
|
String base64 = SM4Util.encrypt("jsimjrby3wqb7dbq", jsonBody);
|
// 设置请求头
|
HttpHeaders headers = new HttpHeaders();
|
headers.setContentType(MediaType.APPLICATION_JSON);
|
headers.setAccept(List.of(MediaType.APPLICATION_JSON, MediaType.TEXT_PLAIN, MediaType.ALL));
|
headers.set("Accept-Language", "zh-CN,zh;q=0.9");
|
headers.setConnection("keep-alive");
|
headers.set("x-auth-token", "eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJ3b3Jrc3BhY2VfaWQiOiJlM2RlYTBmNS0zN2YyLTRkNzktYWU1OC00OTBhZjMyMjgwNjkiLCJzdWIiOiJDbG91ZEFwaVNhbXBsZSIsInVzZXJfdHlwZSI6IjEiLCJuYmYiOjE3MjA2NTg3NDMsImxvZyI6IkxvZ2dlcltjb20uZGppLnNhbXBsZS5jb21tb24ubW9kZWwuQ3VzdG9tQ2xhaW1dIiwiaXNzIjoiREpJIiwiaWQiOiI5Y2MwMGY3Zi0yODMwLTRmM2UtYjdmNC1lN2E1ZjIzYjNjNGYiLCJleHAiOjE4MDcwNTg3NDMsImlhdCI6MTcyMDY1ODc0MywidXNlcm5hbWUiOiJhZG1pbkRQIn0.CrRFYUTK357HleqrrfL7nLdwFJw5Bcv3ms33n_Or8eM");
|
headers.set("x-lc-token", "");
|
// 构建请求实体
|
HttpEntity<String> requestEntity = new HttpEntity<>(base64, headers);
|
// 发送请求
|
ResponseEntity<String> response = restTemplate.exchange(
|
"http://localhost:6789/droneAirport/test",
|
HttpMethod.POST,
|
requestEntity,
|
String.class);
|
}
|
|
@Override
|
public void Authorization(String sn) {
|
AuthorDto dto = new AuthorDto();
|
dto.setSn(sn);
|
dto.setStatus(1);
|
String jsonBody = buildRequestBody(dto);
|
String base64 = SM4Util.encrypt("jsimjrby3wqb7dbq", jsonBody);
|
// 设置请求头
|
HttpHeaders headers = new HttpHeaders();
|
headers.setContentType(MediaType.APPLICATION_JSON);
|
headers.setAccept(List.of(MediaType.APPLICATION_JSON, MediaType.TEXT_PLAIN, MediaType.ALL));
|
headers.set("Accept-Language", "zh-CN,zh;q=0.9");
|
headers.setConnection("keep-alive");
|
headers.set("x-auth-token", "eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJ3b3Jrc3BhY2VfaWQiOiJlM2RlYTBmNS0zN2YyLTRkNzktYWU1OC00OTBhZjMyMjgwNjkiLCJzdWIiOiJDbG91ZEFwaVNhbXBsZSIsInVzZXJfdHlwZSI6IjEiLCJuYmYiOjE3MjA2NTg3NDMsImxvZyI6IkxvZ2dlcltjb20uZGppLnNhbXBsZS5jb21tb24ubW9kZWwuQ3VzdG9tQ2xhaW1dIiwiaXNzIjoiREpJIiwiaWQiOiI5Y2MwMGY3Zi0yODMwLTRmM2UtYjdmNC1lN2E1ZjIzYjNjNGYiLCJleHAiOjE4MDcwNTg3NDMsImlhdCI6MTcyMDY1ODc0MywidXNlcm5hbWUiOiJhZG1pbkRQIn0.CrRFYUTK357HleqrrfL7nLdwFJw5Bcv3ms33n_Or8eM");
|
headers.set("x-lc-token", "");
|
// 构建请求实体
|
HttpEntity<String> requestEntity = new HttpEntity<>(base64, headers);
|
// 发送请求
|
ResponseEntity<String> response = restTemplate.exchange(
|
"http://localhost:6789/droneAirport/test",
|
HttpMethod.POST,
|
requestEntity,
|
String.class);
|
}
|
|
@Override
|
public ReturnTaskParam AddTask(String base64, String Secret, String username) throws Exception {
|
ReturnTaskParam param = new ReturnTaskParam();
|
String taskId = UUID.randomUUID().toString();
|
String traceid = UUID.randomUUID().toString();
|
String sm4Secrect = deSM2(Secret, "23E57DA1E4AB865CCBC325B668762207DEF74345B782237808AE0BABDF26734D");
|
String decryptedJson = SM4Util.decrypt(sm4Secrect, base64);
|
JSONObject jsonObject = new JSONObject(decryptedJson);
|
AddTaskDto addTaskDto = jsonToDto(jsonObject);
|
TaskInfo taskInfo = new TaskInfo();
|
List<String> listx = convertStringToList(addTaskDto.getDeviceid());
|
taskInfo.setId(traceid);
|
taskInfo.setTaskId(taskId);
|
taskInfo.setName(addTaskDto.getBizidname());
|
taskInfo.setDeviceIds(addTaskDto.getDeviceid());
|
taskInfo.setCreateUser(username);
|
param.setTraceId(traceid);
|
taskInfoMapper.insert(taskInfo);
|
|
// 转换 TaskListDto 为 LotInfo
|
List<LotInfo> lotInfos = new ArrayList<>();
|
List<TaskListDto> tasklist = addTaskDto.getTasklist();
|
for (TaskListDto taskListDto : tasklist) {
|
LotInfo lotInfo = convertToLotInfo(taskListDto);
|
lotInfo.setTaskId(taskId);
|
lotInfo.setType(1);
|
lotInfo.setTaskName(addTaskDto.getBizidname());
|
lotInfo.setUserName(username);
|
lotInfos.add(lotInfo);
|
patchesMapper.insert(lotInfo);
|
}
|
|
// 获取机场信息
|
List<Airport> airports = getAirportsBySn(listx);
|
List<AirportTasks> airportTasks = assignTasks(airports, lotInfos);
|
|
// 处理并插入图斑到数据库
|
processAndInsertTasks(airportTasks);
|
|
// 根据机场位置生成航线任务
|
for (AirportTasks airportTask : airportTasks) {
|
Airport airport = airportTask.getAirport();
|
List<LotInfo> lotInfosForAirport = airportTask.getTasks();
|
|
if (!lotInfosForAirport.isEmpty()) {
|
String waylineName = TimerUtil.getTimeName();
|
MultipartFile multipartFile = timerUtil.getFile(waylineName, lotInfosForAirport, airport.getLat(), airport.getLon());
|
WaylineFileEntity waylineFile = timerUtil.backWayline(multipartFile, waylineName, airport.getWorkspaceId(), username);
|
Long time = getNextDayEightAMTimestamp(getDistinctTaskIdCount(airport.getWorkspaceId()));
|
List<List<Long>> listOfLists = new ArrayList<>();
|
List<Long> sublist = new ArrayList<>();
|
sublist.add(time);
|
listOfLists.add(sublist);
|
List<Long> lists1 = new ArrayList<>();
|
lists1.add(time);
|
String times = TimerUtil.convertTimestampToFormattedString(time);
|
param.setJobId(SM4Util.encrypt("jsimjrby3wqb7dbq", taskId));
|
timerUtil.sendPostWithParameters(waylineName, waylineFile.getWaylineId(), times, lists1, listOfLists, taskId + "~" + airport.getWorkspaceId(), airport.getDockSn());
|
timerUtil.updatePatchesStatu(lotInfosForAirport);
|
}
|
}
|
|
return param;
|
}
|
|
|
@Override
|
public ReturnTaskParam updateTask(String base64, String Secret, String username) throws Exception {
|
ReturnTaskParam param = new ReturnTaskParam();
|
String traceid = UUID.randomUUID().toString();
|
String sm4Secrect = deSM2(Secret, "23E57DA1E4AB865CCBC325B668762207DEF74345B782237808AE0BABDF26734D");
|
String decryptedJson = SM4Util.decrypt(sm4Secrect, base64);
|
JSONObject jsonObject = new JSONObject(decryptedJson);
|
UpdataTaskDto addTaskDto = UpjsonToDto(jsonObject);
|
String taskId = addTaskDto.getTaskid();
|
String Bizidname = getBizidnameByTask(taskId);
|
delPatchesByTaskId(taskId);
|
delTaskInfo(taskId);
|
delWaylineJob(taskId);
|
TaskInfo taskInfo = new TaskInfo();
|
List<String> listx = convertStringToList(addTaskDto.getDeviceid());
|
taskInfo.setId(traceid);
|
taskInfo.setTaskId(taskId);
|
taskInfo.setName(Bizidname);
|
taskInfo.setDeviceIds(addTaskDto.getDeviceid());
|
taskInfo.setCreateUser(username);
|
param.setTraceId(traceid);
|
taskInfoMapper.insert(taskInfo);
|
|
// 转换 TaskListDto 为 LotInfo
|
List<LotInfo> lotInfos = new ArrayList<>();
|
List<TaskListDto> tasklist = addTaskDto.getTasklist();
|
for (TaskListDto taskListDto : tasklist) {
|
LotInfo lotInfo = convertToLotInfo(taskListDto);
|
lotInfo.setTaskId(taskId);
|
lotInfo.setType(1);
|
lotInfo.setTaskName(Bizidname);
|
lotInfo.setUserName(username);
|
lotInfos.add(lotInfo);
|
patchesMapper.insert(lotInfo);
|
}
|
|
// 获取机场信息
|
List<Airport> airports = getAirportsBySn(listx);
|
List<AirportTasks> airportTasks = assignTasks(airports, lotInfos);
|
|
// 处理并插入图斑到数据库
|
processAndInsertTasks(airportTasks);
|
|
// 根据机场位置生成航线任务
|
for (AirportTasks airportTask : airportTasks) {
|
Airport airport = airportTask.getAirport();
|
List<LotInfo> lotInfosForAirport = airportTask.getTasks();
|
|
if (!lotInfosForAirport.isEmpty()) {
|
String waylineName = TimerUtil.getTimeName();
|
MultipartFile multipartFile = timerUtil.getFile(waylineName, lotInfosForAirport, airport.getLat(), airport.getLon());
|
WaylineFileEntity waylineFile = timerUtil.backWayline(multipartFile, waylineName, airport.getWorkspaceId(), username);
|
|
// 将为规划的图斑状态更新为已规划
|
Long time = getNextDayEightAMTimestamp(getDistinctTaskIdCount(airport.getWorkspaceId()));
|
List<List<Long>> listOfLists = new ArrayList<>();
|
List<Long> sublist = new ArrayList<>();
|
sublist.add(time);
|
listOfLists.add(sublist);
|
List<Long> lists1 = new ArrayList<>();
|
lists1.add(time);
|
String times = TimerUtil.convertTimestampToFormattedString(time);
|
param.setJobId(taskId);
|
timerUtil.sendPostWithParameters(waylineName, waylineFile.getWaylineId(), times, lists1, listOfLists, airport.getWorkspaceId() + "~" + taskId, airport.getDockSn());
|
timerUtil.updatePatchesStatu(lotInfosForAirport);
|
}
|
}
|
|
return param;
|
}
|
|
@Override
|
public ResponseResult cancelTask(String taskId) {
|
LambdaUpdateWrapper<WaylineJobEntity> updateWrapper = new LambdaUpdateWrapper<>();
|
updateWrapper.like(WaylineJobEntity::getJobId, taskId)
|
.set(WaylineJobEntity::getStatus, 4);
|
waylineJobMapper.update(null, updateWrapper);
|
return null;
|
}
|
|
public void processAndInsertTasks(List<AirportTasks> airportTasks) {
|
for (AirportTasks airportTasksEntry : airportTasks) {
|
Airport airport = airportTasksEntry.getAirport();
|
List<LotInfo> tasks = airportTasksEntry.getTasks();
|
|
// 处理每个任务
|
for (LotInfo lotInfo : tasks) {
|
lotInfo.setWorkspaceId(airport.getWorkspaceId()); // 设置 workspaceId
|
patchesMapper.updateById(lotInfo);
|
}
|
}
|
}
|
|
@Override
|
public ResponseResult queryTaskStatus(String taskid) {
|
List<WaylineJobEntity> jobEntity = waylineJobMapper.selectList(new LambdaQueryWrapper<WaylineJobEntity>().like(WaylineJobEntity::getJobId, taskid));
|
|
if (jobEntity == null || jobEntity.isEmpty()) {
|
return ResponseResult.error(201, "查询无人机机场任务状态失败！");
|
}
|
|
// 检查是否所有 status 都等于 4
|
boolean allStatusAreFour = jobEntity.stream()
|
.allMatch(job -> job.getStatus() == 4);
|
|
// 检查是否所有 status 都等于 3
|
boolean allStatusAreThree = jobEntity.stream()
|
.allMatch(job -> job.getStatus() == 3);
|
|
if (allStatusAreFour) {
|
return ResponseResult.error(204, "查询无人机机场任务状态成功！");
|
} else if (allStatusAreThree) {
|
return ResponseResult.error(200, "查询无人机机场任务状态成功！");
|
} else {
|
return ResponseResult.error(202, "查询无人机机场任务状态成功！");
|
}
|
}
|
|
/**
|
* 根据任务 ID 查找对应的 DB 文件路径
|
*
|
* @param taskId 任务 ID
|
* @return 匹配的 DB 文件路径列表
|
*/
|
public String findDbFilesByTaskId(String taskId) {
|
DbUploadEntity entity = dbUploadMapper.selectOne(new LambdaQueryWrapper<DbUploadEntity>()
|
.eq(DbUploadEntity::getTaskId, taskId));
|
if (entity != null) {
|
DbUploadDto dto = DbEntityToDto(entity);
|
return SM4Util.encrypt("jsimjrby3wqb7dbq", buildRequestBody(dto));
|
}
|
return null;
|
|
// File baseDir = new File(BASE_DIR);
|
//
|
// // 检查基础目录是否存在并且是目录
|
// if (!baseDir.exists() || !baseDir.isDirectory()) {
|
// throw new IllegalStateException("Base directory does not exist or is not a directory.");
|
// }
|
//
|
// // 构建文件路径
|
// File targetDir = new File(baseDir, taskId);
|
// File dbFile = new File(targetDir, ".db");
|
//
|
// // 如果文件存在，则返回文件路径
|
// if (dbFile.exists()) {
|
// return dbFile.getAbsolutePath();
|
// }
|
//
|
// return null; // 或者抛出异常，视需求而定
|
}
|
|
|
public void delPatchesByTaskId(String taskId) {
|
patchesMapper.delete(new LambdaUpdateWrapper<LotInfo>().eq(LotInfo::getTaskId, taskId));
|
}
|
|
public DbUploadDto DbEntityToDto(DbUploadEntity entity) {
|
DbUploadDto dto = new DbUploadDto();
|
dto.setAttachmentCount(entity.getAttachmentCount());
|
dto.setCount(entity.getCount());
|
dto.setSign(entity.getDataSign());
|
dto.setExtension(entity.getExtension());
|
dto.setHash(entity.getHash());
|
dto.setRegioncode(entity.getRegioncode());
|
dto.setTaskId(entity.getTaskId());
|
dto.setKey(entity.getDecrypt());
|
dto.setTimestamp(entity.getTimestamp());
|
dto.setSize(entity.getSize());
|
dto.setUrl(entity.getUrl());
|
return dto;
|
}
|
|
public void delTaskInfo(String taskId) {
|
taskInfoMapper.delete(new LambdaUpdateWrapper<TaskInfo>().eq(TaskInfo::getTaskId, taskId));
|
}
|
|
// private String buildRequestBody(AddDeviceDto dto) {
|
// try {
|
// return objectMapper.writeValueAsString(dto);
|
// } catch (JsonProcessingException e) {
|
// throw new RuntimeException("数据有误", e);
|
// }
|
// }
|
|
public String getWorkspaceId(String devicesn) {
|
DeviceEntity entity = deviceMapper.selectOne(new LambdaQueryWrapper<DeviceEntity>()
|
.eq(DeviceEntity::getDeviceSn, devicesn));
|
return entity.getWorkspaceId();
|
}
|
|
public String getBizidnameByTask(String taskId) {
|
TaskInfo taskInfo = taskInfoMapper.selectOne(new LambdaQueryWrapper<TaskInfo>().eq(TaskInfo::getTaskId, taskId));
|
return taskInfo.getName();
|
}
|
|
public static String enSM2(String data, String pubKey) throws InvalidCipherTextException, IOException {
|
String encrypt = Utils.encrypt(data, pubKey);
|
encrypt = Utils.hexToBase64(encrypt);
|
return encrypt;
|
}
|
|
public static String deSM2(String Base64, String priKey) throws Exception {
|
String hex = Utils.convertBase64ToHex(Base64);
|
return Utils.decrypt(hex, priKey);
|
}
|
|
|
public static Map<String, String> createKey() throws NoSuchAlgorithmException {
|
Map<String, String> keyPair = Utils.createKeyPair();
|
Map<String, String> newKey = new HashMap<>();
|
newKey.put("publicKey", keyPair.get("pbcky"));
|
newKey.put("privateKey", keyPair.get("pveky"));
|
return newKey;
|
}
|
|
public AddDeviceDto deviceEntityToDto(DeviceExpanSionEntity entity) {
|
AddDeviceDto.AddDeviceDtoBuilder builder = AddDeviceDto.builder();
|
if (entity != null) {
|
builder.brand(entity.getBrand())
|
.height(entity.getHeight())
|
.longitude(entity.getLongitude())
|
.latitude(entity.getLatitude())
|
.radius(entity.getRadius())
|
.regioncode(entity.getRegioncode())
|
.deviceid(entity.getDeviceSn())
|
.model(entity.getModel())
|
.build();
|
}
|
return builder.build();
|
}
|
|
public RegistDto airportEntityToDto(DeviceExpanSionEntity entity) {
|
RegistDto.RegistDtoBuilder builder = RegistDto.builder();
|
if (entity != null) {
|
builder.serviceUrl(entity.getServiceUrl())
|
.contacts(entity.getContracts())
|
.phone(entity.getPhone())
|
.liveStreamPluginUrl(entity.getLiveStreamPluginUrl())
|
.build();
|
}
|
return builder.build();
|
}
|
|
public OsdSubDeviceReceiver getDroneFromCache(String sn) {
|
String key = "osd:" + sn;
|
return (OsdSubDeviceReceiver) RedisOpsUtils.get(key);
|
}
|
|
public void delWaylineJob(String taskId) {
|
waylineJobMapper.delete(new LambdaQueryWrapper<WaylineJobEntity>().like(WaylineJobEntity::getJobId, taskId));
|
}
|
|
public DroneStateDto getAirportFromCache(String deviceId) {
|
Optional<OsdSubDeviceReceiver> deviceOsdOpt = redisService.getDeviceOsd(deviceId, OsdSubDeviceReceiver.class);
|
boolean result = deviceOsdOpt.isPresent();
|
if (!result) {
|
String sn = deviceMapper.selectOne(new LambdaQueryWrapper<DeviceEntity>().eq(DeviceEntity::getChildSn, deviceId)).getDeviceSn();
|
Optional<OsdDockReceiver> dockOsdOpt = redisService.getDeviceOsd(sn, OsdDockReceiver.class);
|
boolean results = dockOsdOpt.isPresent();
|
if (!results) {
|
DroneStateDto.DroneStateDtoBuilder builder = DroneStateDto.builder();
|
builder.gimbalYaw(0.0)
|
.gimbalPitch(0.0)
|
.batteryCapacityPercent(0)
|
.height(0)
|
.latitude(0)
|
.longitude(0)
|
.modeCode(1)
|
.elevation(0.0)
|
.gpsState(0)
|
.build();
|
return builder.build();
|
}
|
OsdDockReceiver osdDockReceiver = objectMapper.convertValue(dockOsdOpt.get(), OsdDockReceiver.class);
|
int mode;
|
int sne = osdDockReceiver.getModeCode().getVal();
|
if (sne == 0) {
|
mode = 0;
|
} else {
|
mode = 2;
|
}
|
DroneStateDto.DroneStateDtoBuilder builder = DroneStateDto.builder();
|
builder.gimbalYaw(0.0)
|
.gimbalPitch(0.0)
|
.batteryCapacityPercent(osdDockReceiver.getDroneChargeState().getCapacityPercent())
|
.height(osdDockReceiver.getHeight())
|
.latitude(osdDockReceiver.getLatitude())
|
.longitude(osdDockReceiver.getLongitude())
|
.modeCode(mode)
|
.elevation(0.0)
|
.gpsState(2)
|
.build();
|
return builder.build();
|
}
|
OsdSubDeviceReceiver receiver = objectMapper.convertValue(deviceOsdOpt.get(), OsdSubDeviceReceiver.class);
|
DroneStateDto.DroneStateDtoBuilder builder = DroneStateDto.builder();
|
builder.gimbalYaw(receiver.getPayloads().get(0).getGimbalYaw())
|
.gimbalPitch(receiver.getPayloads().get(0).getGimbalPitch())
|
.batteryCapacityPercent(receiver.getBattery().getCapacityPercent())
|
.height(receiver.getHeight())
|
.latitude(receiver.getLatitude())
|
.longitude(receiver.getLongitude())
|
.modeCode(2)
|
.elevation(receiver.getElevation())
|
.gpsState(2)
|
.build();
|
return builder.build();
|
}
|
|
public static AddTaskDto jsonToDto(JSONObject jsonObject) {
|
AddTaskDto dto = new AddTaskDto();
|
String deviceid = jsonObject.getStr("deviceid");
|
dto.setDeviceid(deviceid);
|
dto.setBizidname(jsonObject.getStr("bizidname"));
|
JSONArray taskListArray = jsonObject.getJSONArray("tasklist");
|
if (taskListArray != null) {
|
List<TaskListDto> taskList = new ArrayList<>();
|
for (int i = 0; i < taskListArray.size(); i++) {
|
JSONObject taskObject = taskListArray.getJSONObject(i);
|
if (taskObject != null) {
|
TaskListDto task = new TaskListDto();
|
task.bsm = taskObject.getStr("bsm");
|
task.xzqdm = taskObject.getStr("xzqdm");
|
task.dkbh = taskObject.getStr("dkbh");
|
task.dklx = taskObject.getStr("dklx");
|
task.zdkbh = taskObject.getStr("zdkbh");
|
task.dkmc = taskObject.getStr("dkmc");
|
task.dkmj = taskObject.getDouble("dkmj");
|
task.dkfw = taskObject.getStr("dkfw");
|
task.bz = taskObject.getStr("bz");
|
taskList.add(task);
|
}
|
}
|
dto.tasklist = taskList;
|
}
|
|
return dto;
|
}
|
|
public static UpdataTaskDto UpjsonToDto(JSONObject jsonObject) {
|
UpdataTaskDto dto = new UpdataTaskDto();
|
String deviceid = jsonObject.getStr("deviceid");
|
String taskid = jsonObject.getStr("taskid");
|
dto.setDeviceid(deviceid);
|
dto.setTaskid(taskid);
|
JSONArray taskListArray = jsonObject.getJSONArray("tasklist");
|
if (taskListArray != null) {
|
List<TaskListDto> taskList = new ArrayList<>();
|
for (int i = 0; i < taskListArray.size(); i++) {
|
JSONObject taskObject = taskListArray.getJSONObject(i);
|
if (taskObject != null) {
|
TaskListDto task = new TaskListDto();
|
task.bsm = taskObject.getStr("bsm");
|
task.xzqdm = taskObject.getStr("xzqdm");
|
task.dkbh = taskObject.getStr("dkbh");
|
task.dklx = taskObject.getStr("dklx");
|
task.zdkbh = taskObject.getStr("zdkbh");
|
task.dkmc = taskObject.getStr("dkmc");
|
task.dkmj = taskObject.getDouble("dkmj");
|
task.dkfw = taskObject.getStr("dkfw");
|
task.bz = taskObject.getStr("bz");
|
taskList.add(task);
|
}
|
}
|
dto.tasklist = taskList;
|
}
|
|
return dto;
|
}
|
|
public static LotInfo convertToLotInfo(TaskListDto taskListDto) {
|
if (taskListDto == null) {
|
return null;
|
}
|
return LotInfo.builder()
|
.bsm(taskListDto.bsm)
|
.dklx(taskListDto.dklx)
|
.xzqdm(taskListDto.xzqdm)
|
.dkbh(taskListDto.dkbh)
|
.dkmc(taskListDto.dkmc)
|
.dkmj(taskListDto.dkmj)
|
.dkfw(taskListDto.dkfw)
|
.bz(taskListDto.bz)
|
.build();
|
}
|
|
|
public List<Airport> getAirportsBySn(List<String> snList) {
|
List<Airport> airports = new ArrayList<>();
|
// 查询所有设备
|
List<DeviceExpanSionEntity> entities = expanSionMapper.selectList(
|
new LambdaQueryWrapper<DeviceExpanSionEntity>().in(DeviceExpanSionEntity::getDeviceSn, snList)
|
);
|
|
// 使用一个Map来存储SN和经纬度的对应关系
|
Map<String, DeviceExpanSionEntity> entityMap = new HashMap<>();
|
for (DeviceExpanSionEntity entity : entities) {
|
entityMap.put(entity.getDeviceSn(), entity);
|
}
|
|
// 遍历SN列表，创建Airport对象
|
for (String sn : snList) {
|
DeviceExpanSionEntity entity = entityMap.get(sn);
|
if (entity != null) {
|
DeviceEntity deviceEntity = deviceMapper.selectOne(new LambdaQueryWrapper<DeviceEntity>().eq(DeviceEntity::getChildSn, sn));
|
Airport airport = new Airport(entity.getWorkspaceId(), deviceEntity.getDeviceSn(), entity.getLatitude(), entity.getLongitude());
|
airports.add(airport);
|
}
|
}
|
|
return airports;
|
}
|
|
public static List<String> convertStringToList(String input) {
|
// 使用 Hutool 解析字符串
|
JSONArray jsonArray = JSONUtil.parseArray(input);
|
|
// 将 JSONArray 转换为 List<String>
|
return jsonArray.stream()
|
.map(Object::toString)
|
.collect(Collectors.toList());
|
}
|
|
public void updateTaskInfo(String traceid, String taskId, String deviceIds, String username) {
|
TaskInfo taskInfo = new TaskInfo();
|
taskInfo.setId(traceid);
|
taskInfo.setTaskId(taskId);
|
taskInfo.setDeviceIds(deviceIds);
|
taskInfo.setCreateUser(username);
|
|
LambdaUpdateWrapper<TaskInfo> updateWrapper = new LambdaUpdateWrapper<>();
|
updateWrapper.eq(TaskInfo::getId, traceid);
|
updateWrapper.eq(TaskInfo::getTaskId, taskId);
|
|
// 执行更新操作
|
taskInfoMapper.update(taskInfo, updateWrapper);
|
}
|
|
public void updateJob(String jobid, String waylineId) {
|
LambdaUpdateWrapper<WaylineJobEntity> updateWrapper = new LambdaUpdateWrapper<>();
|
updateWrapper.eq(WaylineJobEntity::getJobId, jobid)
|
.set(WaylineJobEntity::getFileId, waylineId);
|
|
// 调用 update 方法进行更新操作
|
waylineJobMapper.update(null, updateWrapper);
|
}
|
|
/**
|
* 统计给定工作空间下，创建时间为当天的不同 taskId 的数量。
|
*
|
* @param workspaceId 工作空间 ID
|
* @return 不同 taskId 的数量
|
*/
|
public Integer getDistinctTaskIdCount(String workspaceId) {
|
// 获取当前日期的开始和结束时间
|
LocalDate today = LocalDate.now();
|
LocalDateTime startTime = today.atStartOfDay();
|
LocalDateTime endTime = today.plusDays(1).atStartOfDay();
|
|
// 转换为毫秒时间戳
|
long startTimestamp = startTime.atZone(ZoneId.systemDefault()).toInstant().toEpochMilli();
|
long endTimestamp = endTime.atZone(ZoneId.systemDefault()).toInstant().toEpochMilli();
|
|
// 使用 LambdaQueryWrapper 进行查询
|
LambdaQueryWrapper<LotInfo> queryWrapper = Wrappers.lambdaQuery();
|
queryWrapper.eq(LotInfo::getWorkspaceId, workspaceId)
|
.ge(LotInfo::getCreateTime, startTimestamp)
|
.lt(LotInfo::getCreateTime, endTimestamp);
|
|
// 获取符合条件的所有 taskId
|
Set<String> taskIdSet = patchesMapper.selectList(queryWrapper).stream()
|
.map(LotInfo::getTaskId)
|
.collect(Collectors.toSet());
|
|
// 返回不同 taskId 的数量
|
return taskIdSet.size();
|
}
|
|
|
}
|
