IOT/MiddlePlatform
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

#init

Browse Source
This commit is contained in:
rqian
2026-04-17 10:42:16 +08:00
parent 17c52c978c
commit d26dde07d7
64 changed files with 4737 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

347
maibu-netty-server/src/main/java/com/maibu/service/DeviceTaskService.java Normal file
View File

@@ -0,0 +1,347 @@
package com.fastbee.service;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.fastbee.common.core.domain.model.LoginUser;
import com.fastbee.dto.DevicePlanStatisticsDTO;
import com.fastbee.dto.DevicePlanTaskStatisticsDTO;
import com.fastbee.dto.DeviceTaskQueryDTO;
import com.fastbee.dto.DeviceWorkStatisticsDTO;
import com.fastbee.iot.domain.Device;
import com.fastbee.iot.domain.DevicePlan;
import com.fastbee.iot.domain.DevicePlanTask;
import com.fastbee.iot.domain.DeviceTaskPlanRule;
import com.fastbee.iot.enums.DeviceTaskStaus;
import com.fastbee.iot.mapper.DeviceMapper;
import com.fastbee.iot.mapper.DevicePlanMapper;
import com.fastbee.iot.mapper.DevicePlanTaskMapper;
import com.fastbee.iot.mapper.WorkRecordMapper;
import com.fastbee.iot.model.WorkRecord;
import com.fastbee.memory.GlobalMemory;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;
import org.springframework.web.bind.annotation.RequestParam;
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter;
import java.util.ArrayList;
import java.util.List;
import java.util.Random;
import java.util.stream.Collectors;
@Service
@Slf4j
public class DeviceTaskService {
@Autowired
private DevicePlanMapper devicePlanMapper;
@Autowired
private DevicePlanTaskMapper devicePlanTaskMapper;
@Autowired
private DeviceMapper deviceMapper;
@Autowired
private WorkRecordMapper workRecordMapper;
public int insertOrUpdate(DevicePlan devicePlan, LoginUser loginUser) {
Long id = devicePlan.getId();
if (id != null) {
devicePlan.setUpdateBy(loginUser.getUsername());
devicePlan.setUpdateTime(LocalDateTime.now());
transferDays(devicePlan);
int i = devicePlanMapper.updateById(devicePlan);
GlobalMemory.addDevicePlan(devicePlan);
return i;
} else {
devicePlan.setTaskStaus(DeviceTaskStaus.NEW);
devicePlan.setCreateTime(LocalDateTime.now());
devicePlan.setCreateBy(loginUser.getUsername());
transferDays(devicePlan);
int i = devicePlanMapper.insert(devicePlan);
GlobalMemory.addDevicePlan(devicePlan);
return i;
}
}
public void transferDays(DevicePlan devicePlan) {
DeviceTaskPlanRule deviceTaskPlanRules = devicePlan.getDeviceTaskPlanRules();
if (deviceTaskPlanRules != null && !CollectionUtils.isEmpty(deviceTaskPlanRules.getDays())) {
List<String> days = new ArrayList<>();
deviceTaskPlanRules.getDays().forEach(x -> {
String day = null;
switch (x) {
case MONDAY:
day = "周一";
break;
case TUESDAY:
day = "周二";
break;
case WEDNESDAY:
day = "周三";
break;
case THURSDAY:
day = "周四";
break;
case FRIDAY:
day = "周五";
break;
case SATURDAY:
day = "周六";
break;
case SUNDAY:
day = "周日";
break;
}
days.add(day);
});
devicePlan.getDeviceTaskPlanRules().setDaysTranslate(days);
}
}
public List<DevicePlan> getDeviceTaskByUser(Long userId) {
if (userId != null) {
LambdaQueryWrapper<DevicePlan> queryWrapper = new LambdaQueryWrapper<>();
queryWrapper.eq(DevicePlan::getUserId, userId);
return transferDevicePlanData(devicePlanMapper.selectList(queryWrapper));
} else {
return new ArrayList<>();
}
}
public List<DevicePlan> getDeviceTask(DeviceTaskQueryDTO dto) {
if (dto != null) {
LambdaQueryWrapper<DevicePlan> queryWrapper = new LambdaQueryWrapper<>();
if (dto.getUserId() != null) {
queryWrapper.eq(DevicePlan::getUserId, dto.getUserId());
}
if (dto.getDeviceId() != null) {
queryWrapper.eq(DevicePlan::getDeviceId, dto.getDeviceId());
}
if (dto.getTaskStaus() != null) {
queryWrapper.eq(DevicePlan::getTaskStaus, dto.getTaskStaus());
}
if (dto.getPlanName() != null) {
queryWrapper.like(DevicePlan::getPlanName, dto.getPlanName());
}
queryWrapper.eq(DevicePlan::getDelFlag, 0);
List<DevicePlan> list = devicePlanMapper.selectList(queryWrapper);
if (!CollectionUtils.isEmpty(list)) {
list.forEach(x -> {
Device device = deviceMapper.selectDeviceBySerialNumber(x.getDeviceId());
if (device != null) {
x.setDeviceAlias(device.getDeviceAlias());
}
});
}
return transferDevicePlanData(list);
} else {
return new ArrayList<>();
}
}
public int delete(Long id) {
DevicePlan planTask = devicePlanMapper.selectById(id);
if (planTask != null) {
planTask.setDelFlag(1);
GlobalMemory.removeDevicePlan(planTask);
return devicePlanMapper.updateById(planTask);
}
return 0;
}
public void start(Long id, LoginUser loginUser) {
if (id != null) {
DevicePlan planTask = devicePlanMapper.selectById(id);
if (planTask != null) {
planTask.setTaskStaus(DeviceTaskStaus.EXECUTING);
planTask.setUpdateTime(LocalDateTime.now());
planTask.setUpdateBy(loginUser.getUsername());
devicePlanMapper.updateById(planTask);
}
}
}
public List<DevicePlanStatisticsDTO> getDevicePlanStatistics(Long userId) {
List<DevicePlanStatisticsDTO> list = new ArrayList<>();
LocalDate today = LocalDate.now();
LocalDateTime monthStart = LocalDateTime.of(
today.getYear(), // 当前年
today.getMonth(), // 当前月
1, // 当月第一天
0, 0, 0, 0 // 时:分:秒:纳秒（00:00:00.000）
);
LocalDate lastDayOfMonth = today.withDayOfMonth(today.lengthOfMonth()); // 获取当月最后一天
LocalDateTime monthEnd = LocalDateTime.of(
lastDayOfMonth.getYear(),
lastDayOfMonth.getMonth(),
lastDayOfMonth.getDayOfMonth(),
23, 59, 59, 999_999_999 // 时:分:秒:纳秒（23:59:59.999）
);
LambdaQueryWrapper<DevicePlan> queryWrapper = new LambdaQueryWrapper<>();
queryWrapper.ge(DevicePlan::getCreateTime, monthStart)
.le(DevicePlan::getCreateTime, monthEnd)
.eq(DevicePlan::getUserId,userId);
List<DevicePlan> devicePlans = transferDevicePlanData(devicePlanMapper.selectList(queryWrapper));
if (!CollectionUtils.isEmpty(devicePlans)) {
List<LocalDate> dates = getMonthDays();
for (LocalDate date : dates) {
List<DevicePlan> plans = devicePlans.stream().filter(x -> x.getCreateTime().toLocalDate().equals(date)).collect(Collectors.toList());
if (!CollectionUtils.isEmpty(plans)) {
DevicePlanStatisticsDTO dto = new DevicePlanStatisticsDTO();
DateTimeFormatter formatter = DateTimeFormatter.ofPattern("M.d");
dto.setName(formatter.format(date));
long count = plans.stream().filter(x -> DeviceTaskStaus.FINISH.equals(x.getTaskStaus())).count();
dto.setCompleted((int) count);
dto.setPlanned(plans.size());
list.add(dto);
}
}
}
return list;
}
public List<LocalDate> getMonthDays() {
LocalDate today = LocalDate.now();
int totalDays = today.lengthOfMonth();
LocalDate firstDayOfMonth = LocalDate.of(today.getYear(), today.getMonth(), 1);
List<LocalDate> allDaysOfCurrentMonth = new ArrayList<>();
for (int i = 0; i < totalDays; i++) {
// 每次累加1天，获取当月每一天
LocalDate currentDay = firstDayOfMonth.plusDays(i);
allDaysOfCurrentMonth.add(currentDay);
}
return allDaysOfCurrentMonth;
}
public List<DevicePlanTaskStatisticsDTO> getDevicePlanTaskStatistics(Long userId) {
List<DevicePlanTaskStatisticsDTO> list = new ArrayList<>();
// 1. 获取当前日期
LocalDate today = LocalDate.now();
// 2. 构造当前月份的起始时间：当月第一天 00:00:00
LocalDateTime monthStart = LocalDateTime.of(
today.getYear(), // 当前年
today.getMonth(), // 当前月
1, // 当月第一天
0, 0, 0, 0 // 时:分:秒:纳秒（00:00:00.000）
);
// 3. 构造当前月份的结束时间：当月最后一天 23:59:59.999
LocalDate lastDayOfMonth = today.withDayOfMonth(today.lengthOfMonth()); // 获取当月最后一天
LocalDateTime monthEnd = LocalDateTime.of(
lastDayOfMonth.getYear(),
lastDayOfMonth.getMonth(),
lastDayOfMonth.getDayOfMonth(),
23, 59, 59, 999_999_999 // 时:分:秒:纳秒（23:59:59.999）
);
// 4. 构建 LambdaQueryWrapper 进行范围查询
LambdaQueryWrapper<DevicePlan> query = new LambdaQueryWrapper<>();
query.eq(DevicePlan::getUserId,userId);
List<DevicePlan> devicePlans = transferDevicePlanData(devicePlanMapper.selectList(query));
if(CollectionUtils.isEmpty(devicePlans)) return new ArrayList<>();
List<Long> planIds = devicePlans.stream().map(DevicePlan::getId).collect(Collectors.toList());
if(CollectionUtils.isEmpty(planIds)) return new ArrayList<>();
LambdaQueryWrapper<DevicePlanTask> queryWrapper = new LambdaQueryWrapper<>();
// 核心：筛选 createTime 大于等于起始时间，且小于等于结束时间
queryWrapper.ge(DevicePlanTask::getCreateTime, monthStart)
.le(DevicePlanTask::getCreateTime, monthEnd)
.in(DevicePlanTask::getPlanId,planIds);
List<DevicePlanTask> devicePlanTasks = transferDeviceTaskData(devicePlanTaskMapper.selectList(queryWrapper));
if (!CollectionUtils.isEmpty(devicePlanTasks)) {
List<LocalDate> dates = getMonthDays();
for (LocalDate date : dates) {
List<DevicePlanTask> planTasks = devicePlanTasks.stream().filter(x -> x.getCreateTime().toLocalDate().equals(date)).collect(Collectors.toList());
if (!CollectionUtils.isEmpty(planTasks)) {
DevicePlanTaskStatisticsDTO dto = new DevicePlanTaskStatisticsDTO();
DateTimeFormatter formatter = DateTimeFormatter.ofPattern("M.d");
dto.setDate(formatter.format(date));
long finishCount = planTasks.stream().filter(x -> DeviceTaskStaus.FINISH.equals(x.getTaskStaus())).count();
dto.setSuccess((int) finishCount);
long failedCount = planTasks.stream().filter(x -> DeviceTaskStaus.FAILED.equals(x.getTaskStaus())).count();
dto.setFailed((int) failedCount);
long pauseCount = planTasks.stream().filter(x -> DeviceTaskStaus.PAUSE.equals(x.getTaskStaus())).count();
dto.setPaused((int) pauseCount);
list.add(dto);
}
}
}
return list;
}
public WorkRecord concurrentDevicePlanTask(String deviceId) {
LambdaQueryWrapper<DevicePlanTask> queryWrapper = new LambdaQueryWrapper<>();
queryWrapper.eq(DevicePlanTask::getDeviceId, deviceId)
.eq(DevicePlanTask::getTaskStaus, DeviceTaskStaus.EXECUTING);
List<DevicePlanTask> devicePlanTasks = devicePlanTaskMapper.selectList(queryWrapper);
Long routId = CollectionUtils.isEmpty(devicePlanTasks) ? null : devicePlanTasks.get(0).getRouteId();
return routId == null ? null : workRecordMapper.selectWorkRecordById(routId);
}
public List<DeviceWorkStatisticsDTO> workStatistics(Long userId) {
//todo update
List<DeviceWorkStatisticsDTO> result = new ArrayList<>();
Random random = new Random();
int times = random.nextInt(100);
int s = random.nextInt(1000);
DeviceWorkStatisticsDTO dto = new DeviceWorkStatisticsDTO();
dto.setLabel("割草次数");
dto.setValue(times);
DeviceWorkStatisticsDTO dto2 = new DeviceWorkStatisticsDTO();
dto2.setLabel("割草时长(s)");
dto2.setValue(s);
DeviceWorkStatisticsDTO dto3 = new DeviceWorkStatisticsDTO();
dto3.setLabel("割草面积(m2)");
dto3.setValue(s * 5);
result.add(dto);
result.add(dto2);
result.add(dto3);
return result;
}
public List<DevicePlanTask> deviceTaskPool(DeviceTaskQueryDTO dto) {
List<DevicePlanTask> list = new ArrayList<>();
GlobalMemory.devicePlanTaskPrepareMap.values().forEach(list::addAll);
list.addAll(GlobalMemory.devicePlanTaskExecuteMap.values());
if (dto != null) {
if (dto.getTaskStaus() != null) {
list = list.stream().filter(x -> x.getTaskStaus().equals(dto.getTaskStaus())).collect(Collectors.toList());
}
if (!StringUtils.isEmpty(dto.getDeviceId())) {
list = list.stream().filter(x -> x.getDeviceId().equals(dto.getDeviceId())).collect(Collectors.toList());
}
if (dto.getPlanId() != null) {
list = list.stream().filter(x -> x.getPlanId().equals(dto.getPlanId())).collect(Collectors.toList());
}
}
return transferDeviceTaskData(list);
}
public List<DevicePlan> transferDevicePlanData(List<DevicePlan> list) {
if (CollectionUtils.isEmpty(list)) return list;
list.forEach(x -> {
DeviceTaskStaus taskStaus = x.getTaskStaus();
x.setTaskStausTranslate(taskStaus.getDescription());
});
return list;
}
public List<DevicePlanTask> transferDeviceTaskData(List<DevicePlanTask> list) {
if (CollectionUtils.isEmpty(list)) return list;
list.forEach(x -> {
DeviceTaskStaus taskStaus = x.getTaskStaus();
x.setTaskStausTranslate(taskStaus.getDescription());
});
return list;
}
}