From 2fee40486f9473398be3c293e8311d45bd53adc9 Mon Sep 17 00:00:00 2001
From: 无关风月 <443237572@qq.com>
Date: 星期五, 11 七月 2025 20:47:13 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/master'
---
ruoyi-admin/src/main/java/com/ruoyi/web/controller/task/TaskUtil.java | 729 ++++++++++++++++++++++++++++---------------------------
1 files changed, 366 insertions(+), 363 deletions(-)
diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/task/TaskUtil.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/task/TaskUtil.java
index 2b3f0cd..d15f91c 100644
--- a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/task/TaskUtil.java
+++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/task/TaskUtil.java
@@ -1,363 +1,366 @@
-//package com.ruoyi.web.controller.task;
-//
-//
-//import com.alibaba.fastjson.JSONArray;
-//import com.alibaba.fastjson.JSONObject;
-//import com.baomidou.mybatisplus.core.toolkit.Wrappers;
-//import com.ruoyi.common.core.domain.entity.SysUser;
-//import com.ruoyi.common.core.redis.RedisCache;
-//import com.ruoyi.common.utils.CodeGenerateUtils;
-//import com.ruoyi.system.mapper.TCleanerMapper;
-//import com.ruoyi.system.model.*;
-//import com.ruoyi.system.service.*;
-//import javafx.concurrent.Task;
-//import lombok.extern.slf4j.Slf4j;
-//import org.apache.poi.ss.formula.functions.T;
-//import org.springframework.beans.factory.annotation.Autowired;
-//import org.springframework.scheduling.annotation.Scheduled;
-//import org.springframework.stereotype.Component;
-//import org.springframework.util.CollectionUtils;
-//
-//import java.math.BigDecimal;
-//import java.text.SimpleDateFormat;
-//import java.time.LocalDate;
-//import java.time.LocalDateTime;
-//import java.util.*;
-//import java.util.stream.Collectors;
-//
-///**
-// * @author xiaochen
-// * @date 2025/6/11 18:39
-// */
-//@Slf4j
-//@Component
-//public class TaskUtil {
-//
-// // 项目部权重标识
-// private static final String PROJECT_DEPT_WEIGHT = ":PROJECT_DEPT_WEIGHT";
-// // 保洁员权重标识
-// private static final String CLEANER_WEIGHT = ":CLEANER_WEIGHT";
-// // 重复点位标识
-// private static final String REPEAT_LOCATION = ":REPEAT_LOCATION";
-//
-// @Autowired
-// private RedisCache redisCache;
-// @Autowired
-// private TTemplateService templateService;
-// @Autowired
-// private TTemplateDetailService templateDetailService;
-// @Autowired
-// private TProjectDeptService projectDeptService;
-// @Autowired
-// private TCleanerService cleanerService;
-// @Autowired
-// private TLocationService locationService;
-// @Autowired
-// private TLocationTypeService locationTypeService;
-// @Autowired
-// private ISysUserService sysUserService;
-// @Autowired
-// private TTaskCleanService taskCleanService;
-// @Autowired
-// private TTemplateCountService templateCountService;
-//
-//
-// // 每一个小时执行一次
-//// @Scheduled(cron = "0 0 0 * * ?")
-//// @Scheduled(fixedRate = 1500000000)
-// public void dayOfCreateInspection() {
-// try {
-//
-// // 查询任务模板
-// List<TTemplate> list = templateService.list();
-// if (CollectionUtils.isEmpty(list)) {
-// return;
-// }
-// list.forEach(template -> {
-// // 创建任务
-// createInspection(template);
-// });
-//
-// // 查询前一天为执行的所有任务，改成已超时状态
-// List<TTask> taskCleanList = taskCleanService.list(Wrappers.lambdaQuery(TTask.class)
-// .like(TTask::getImplementTime, LocalDate.now().minusDays(1))
-// .eq(TTask::getStatus, 1));
-// if (!CollectionUtils.isEmpty(taskCleanList)) {
-// taskCleanList.forEach(task -> {
-// task.setStatus(2);
-// });
-// taskCleanService.updateBatchById(taskCleanList);
-// }
-//
-// } catch (Exception e) {
-// e.printStackTrace();
-// }
-// }
-//
-// public void createInspection(TTemplate template) {
-// // 查询所有的模板详情
-// List<TTemplateDetail> list = templateDetailService.list(Wrappers.lambdaQuery(TTemplateDetail.class)
-// .eq(TTemplateDetail::getTemplateId, template.getId()));
-// if (CollectionUtils.isEmpty(list)) {
-// return;
-// }
-// // 通过模板id查询员工巡检员
-// List<SysUser> sysUsers = sysUserService.selectUserByTempLateId(template.getId());
-// // 创建任务
-// for (TTemplateDetail detail : list) {
-// // 计算周期
-// int cycle = detail.getCycle();
-// switch (detail.getCycleType()){
-// case 2:
-// cycle = cycle * 7;
-// break;
-// case 3:
-// cycle = cycle * 30;
-// break;
-// case 4:
-// cycle = cycle * 90;
-// break;
-// case 5:
-// cycle = cycle * 365;
-// break;
-// }
-//
-// // 拿到保洁抽查次数
-// int num1 = detail.getNum1();
-// // 拿到项目部数
-// int num2 = detail.getNum2();
-// // 拿到每日重复点位
-// int num3 = detail.getNum3();
-// // 获取点位类型的占比
-// String num4 = detail.getNum4();
-//
-// // 未绑定员工
-// if(CollectionUtils.isEmpty(sysUsers)){
-// continue;
-// }
-//
-// int taskCount = 0;
-// for (SysUser sysUser : sysUsers) {
-// List<TProjectDept> projectDeptLists = new ArrayList<>();
-// if(sysUser.getDeptType() == 1){
-// TProjectDept projectDept = projectDeptService.getById(sysUser.getDeptId());
-// if("0".equals(projectDept.getParentId())){
-// projectDeptLists = projectDeptService.list(Wrappers.lambdaQuery(TProjectDept.class)
-// .eq(TProjectDept::getParentId, projectDept.getId()));
-// }else {
-// projectDeptLists.add(projectDept);
-// }
-// }else {
-// projectDeptLists = projectDeptService.list(Wrappers.lambdaQuery(TProjectDept.class)
-// .ne(TProjectDept::getParentId,0));
-// }
-// List<String> proDeptIds = projectDeptLists.stream().map(TProjectDept::getId).collect(Collectors.toList());
-//
-// // 获取项目部在该模板详情中的权重
-// List<String> projectDeptIds = redisCache.getCacheList(detail.getId() + PROJECT_DEPT_WEIGHT);
-// // 获取项目部列表
-// List<TProjectDept> projectDeptList;
-// if(CollectionUtils.isEmpty(projectDeptIds)){
-// projectDeptList = projectDeptLists;
-// }else {
-// projectDeptList = projectDeptService.list(Wrappers.lambdaQuery(TProjectDept.class)
-// .ne(TProjectDept::getParentId,0)
-// .in(TProjectDept::getId, proDeptIds)
-// .notIn(TProjectDept::getId, projectDeptIds));
-// // 所过所有的项目部都被抽取了，则重新抽取，并且清空项目部权重
-// if(CollectionUtils.isEmpty(projectDeptList)){
-// projectDeptList = projectDeptLists;
-// redisCache.deleteObject(detail.getId() + PROJECT_DEPT_WEIGHT);
-// }
-// }
-// // 如果可抽取的项目部数不足，先抽取余下项目部后，再清空权重，重新抽取
-// List<TProjectDept> projectDepts = randomSelection(projectDeptList, num2);
-// if(projectDepts.size() < num2){
-// List<String> proIds = projectDepts.stream().map(TProjectDept::getId).collect(Collectors.toList());
-// List<TProjectDept> projectDeptList1 = projectDeptService.list(Wrappers.lambdaQuery(TProjectDept.class)
-// .notIn(TProjectDept::getId, proIds)
-// .ne(TProjectDept::getParentId,0));
-// redisCache.deleteObject(detail.getId() + PROJECT_DEPT_WEIGHT);
-// List<TProjectDept> projectDepts1 = randomSelection(projectDeptList1, num2 - projectDepts.size());
-// List<String> proIds1 = projectDepts1.stream().map(TProjectDept::getId).collect(Collectors.toList());
-// // 将已抽取的项目部id保存到redis中
-// redisCache.setCacheList(detail.getId() + PROJECT_DEPT_WEIGHT, proIds1);
-// projectDepts.addAll(projectDepts1);
-// }else {
-// List<String> proIds = projectDepts.stream().map(TProjectDept::getId).collect(Collectors.toList());
-// // 将已抽取的项目部id保存到redis中
-// redisCache.setCacheList(detail.getId() + PROJECT_DEPT_WEIGHT, proIds);
-// }
-// List<String> proIds = projectDepts.stream().map(TProjectDept::getId).collect(Collectors.toList());
-//
-// // 拿到抽取的项目部下的所有保洁员
-//// List<TProjectDept> tProjectDeptList = projectDeptService.list(Wrappers.lambdaQuery(TProjectDept.class)
-//// .in(TProjectDept::getId, proIds));
-// // 获取片区id
-// List<String> areaIds = projectDepts.stream().map(TProjectDept::getId).collect(Collectors.toList());
-// if(CollectionUtils.isEmpty(areaIds)){
-// return;
-// }
-//
-// // 计算每天需要抽取多少个保洁员
-// long count = cleanerService.count(Wrappers.lambdaQuery(TCleaner.class)
-// .in(TCleaner::getProjectId, areaIds));
-// int cleanerSums = num1 * Integer.parseInt(count + "");
-// // 获取每天需要抽取的保洁员数,向上取整
-// if (cleanerSums < cycle){
-// cleanerSums = cycle;
-// }
-// int dayCleanerCount = cleanerSums / cycle;
-//
-// if((cycle == detail.getCurrentValue()) && cleanerSums % cycle != 0){
-// dayCleanerCount++;
-// }
-//
-// // 获取保洁员权重
-// List<String> cleanerIds = redisCache.getCacheList(detail.getId() + CLEANER_WEIGHT);
-// // 获取保洁员列表
-// List<TCleaner> cleaners;
-// if(CollectionUtils.isEmpty(cleanerIds)){
-// cleaners = cleanerService.list(Wrappers.lambdaQuery(TCleaner.class)
-// .in(TCleaner::getProjectId, areaIds));
-// }else {
-// cleaners = cleanerService.list(Wrappers.lambdaQuery(TCleaner.class)
-// .in(TCleaner::getProjectId, areaIds)
-// .notIn(TCleaner::getId, cleanerIds));
-// if(CollectionUtils.isEmpty(cleaners)){
-// cleaners = cleanerService.list(Wrappers.lambdaQuery(TCleaner.class)
-// .in(TCleaner::getProjectId, areaIds));
-// redisCache.deleteObject(detail.getId() + CLEANER_WEIGHT);
-// }
-// }
-//
-// // 抽取保洁员
-// List<TCleaner> tCleaners = randomSelection(cleaners, dayCleanerCount);
-// if(CollectionUtils.isEmpty(tCleaners)){
-// log.error("没有可抽取的保洁员,模板id为：{}",detail.getId());
-// continue;
-// }
-// if(tCleaners.size() < dayCleanerCount){
-// List<String> cleanIds = tCleaners.stream().map(TCleaner::getId).collect(Collectors.toList());
-// List<TCleaner> cleaners1 = cleanerService.list(Wrappers.lambdaQuery(TCleaner.class)
-// .in(TCleaner::getProjectId, areaIds)
-// .notIn(TCleaner::getId, cleanIds));
-// redisCache.deleteObject(detail.getId() + CLEANER_WEIGHT);
-// List<TCleaner> tCleaners1 = randomSelection(cleaners1, dayCleanerCount - tCleaners.size());
-// List<String> cleanIds1 = tCleaners1.stream().map(TCleaner::getId).collect(Collectors.toList());
-// // 将已抽取的保洁员id保存到redis中
-// cleanIds1.addAll(cleanIds);
-// redisCache.setCacheList(detail.getId() + CLEANER_WEIGHT, cleanIds1);
-// tCleaners.addAll(tCleaners1);
-// }else {
-// List<String> cleanIds = tCleaners.stream().map(TCleaner::getId).collect(Collectors.toList());
-// redisCache.setCacheList(detail.getId() + CLEANER_WEIGHT, cleanIds);
-// }
-//
-// // 通过保洁员id查询点位
-// List<String> cleanersIds = tCleaners.stream().map(TCleaner::getId).collect(Collectors.toList());
-// List<TLocation> locationList = locationService.list(Wrappers.lambdaQuery(TLocation.class)
-// .in(TLocation::getLocationCleaner, cleanersIds));
-// // 查询点位类型
-// JSONArray jsonArray = JSONObject.parseArray(num4);
-// List<TLocation> tLocationList = new ArrayList<>();
-// for (Object o : jsonArray) {
-// JSONObject jsonObject = JSONObject.parseObject(o.toString());
-// String id = jsonObject.getString("id");
-// BigDecimal value = jsonObject.getBigDecimal("value");
-// List<TLocation> locations = locationList.stream().filter(tLocation -> tLocation.getLocationType().equals(id)).collect(Collectors.toList());
-// if(!CollectionUtils.isEmpty(locations)){
-// BigDecimal bigDecimal = new BigDecimal(locations.size()).multiply(value.divide(new BigDecimal(100), 2, BigDecimal.ROUND_DOWN)).setScale(0, BigDecimal.ROUND_UP);
-// int locationCount = bigDecimal.intValue();
-// if(locationCount > 0){
-// List<TLocation> tLocations = randomSelection(locations, locationCount);
-// tLocationList.addAll(tLocations);
-// }
-// }
-// }
-// tLocationList = tLocationList.stream().distinct().collect(Collectors.toList());
-// // 抽取重复点位
-// Integer currentValue = detail.getCurrentValue();
-// if(currentValue != cycle){
-// // 周期天数加一
-// detail.setCurrentValue(currentValue + 1);
-// // 获取重复点位
-// Set<TLocation> repeatLocation = redisCache.getCacheSet(detail.getId() + ":" + sysUser.getUserId() + REPEAT_LOCATION);
-// if(!CollectionUtils.isEmpty(repeatLocation)){
-// List<TLocation> locations = randomSelection(new ArrayList<>(tLocationList), num3);
-// tLocationList.addAll(locations);
-// }
-// Set<TLocation> locationSet = new HashSet<>(tLocationList);
-// redisCache.setCacheSet(detail.getId() + ":" + sysUser.getUserId() + REPEAT_LOCATION, locationSet);
-//
-// }else {
-// // 设置当前周期为0
-// detail.setCurrentValue(0);
-// // 将重复点位置空
-// redisCache.deleteObject(detail.getId() + ":" + sysUser.getUserId() + REPEAT_LOCATION);
-// // 将项目部权重置空
-// redisCache.deleteObject(detail.getId() + PROJECT_DEPT_WEIGHT);
-// // 将保洁员权重置空
-// redisCache.deleteObject(detail.getId() + CLEANER_WEIGHT);
-// }
-//
-// // 创建任务
-// List<TTask> tasks = new ArrayList<>();
-// for (TLocation tLocation : tLocationList) {
-// TTask task = new TTask();
-// // 获取保洁员
-// tCleaners.stream().filter(tCleaner -> tCleaner.getId().equals(tLocation.getLocationCleaner())).findFirst().ifPresent(tCleaner -> {
-// task.setProjectId(tCleaner.getProjectId());
-// task.setCleanerId(tCleaner.getId());
-// });
-// // 获取巡检员
-// task.setPatrolInspector(sysUser.getUserId().toString());
-// task.setPatrolInspectorDept(sysUser.getDeptId());
-// task.setUserId(sysUser.getUserId());
-// task.setStatus(1);
-// task.setLocationId(tLocation.getId());
-// task.setImplementTime(LocalDateTime.now().plusDays(1));
-// task.setTaskType(1);
-// task.setTemplateId(detail.getTemplateId());
-// String nameAndCode = CodeGenerateUtils.generateVolumeSn();
-// task.setTaskName(nameAndCode);
-// task.setTaskCode(nameAndCode);
-// tasks.add(task);
-// }
-//
-// // 添加应生成任务数量
-// TTemplateCount templateCount = new TTemplateCount();
-// templateCount.setTemplateId(detail.getTemplateId());
-// templateCount.setTaskCount(tLocationList.size()* cycle);
-// templateCount.setUserId(sysUser.getUserId());
-// templateCountService.save(templateCount);
-//
-// taskCleanService.saveBatch(tasks);
-// taskCount = taskCount + tasks.size();
-// }
-// template.setTaskCount(taskCount);
-// templateService.updateById(template);
-// templateDetailService.updateById(detail);
-// }
-// }
-//
-// /**
-// * 从集合中随机抽取数量
-// * @param list
-// * @param size
-// * @return
-// * @param <T>
-// */
-// private static <T> List<T> randomSelection(List<T> list, int size) {
-// List<T> selected = new ArrayList<>();
-// int length = list.size();
-// for (int i = 0; i < size; i++) {
-// Collections.shuffle(list);
-// // 如果需求的大小超过了集合的长度，则只能取到集合的全部元素
-// int subListSize = Math.min(size - selected.size(), length);
-// selected.addAll(list.subList(0, subListSize));
-// }
-// return selected;
-// }
-//
-//}
+package com.ruoyi.web.controller.task;
+
+
+import com.alibaba.fastjson.JSONArray;
+import com.alibaba.fastjson.JSONObject;
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
+import com.ruoyi.common.core.domain.entity.SysUser;
+import com.ruoyi.common.core.redis.RedisCache;
+import com.ruoyi.common.utils.CodeGenerateUtils;
+import com.ruoyi.system.mapper.TCleanerMapper;
+import com.ruoyi.system.model.*;
+import com.ruoyi.system.service.*;
+import javafx.concurrent.Task;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.poi.ss.formula.functions.T;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.scheduling.annotation.Scheduled;
+import org.springframework.stereotype.Component;
+import org.springframework.util.CollectionUtils;
+
+import java.math.BigDecimal;
+import java.text.SimpleDateFormat;
+import java.time.LocalDate;
+import java.time.LocalDateTime;
+import java.util.*;
+import java.util.stream.Collectors;
+
+/**
+ * @author xiaochen
+ * @date 2025/6/11 18:39
+ */
+@Slf4j
+@Component
+public class TaskUtil {
+
+ // 项目部权重标识
+ private static final String PROJECT_DEPT_WEIGHT = ":PROJECT_DEPT_WEIGHT";
+ // 保洁员权重标识
+ private static final String CLEANER_WEIGHT = ":CLEANER_WEIGHT";
+ // 重复点位标识
+ private static final String REPEAT_LOCATION = ":REPEAT_LOCATION";
+
+ @Autowired
+ private RedisCache redisCache;
+ @Autowired
+ private TTemplateService templateService;
+ @Autowired
+ private TTemplateDetailService templateDetailService;
+ @Autowired
+ private TProjectDeptService projectDeptService;
+ @Autowired
+ private TCleanerService cleanerService;
+ @Autowired
+ private TLocationService locationService;
+ @Autowired
+ private TLocationTypeService locationTypeService;
+ @Autowired
+ private ISysUserService sysUserService;
+ @Autowired
+ private TTaskCleanService taskCleanService;
+ @Autowired
+ private TTemplateCountService templateCountService;
+
+
+ // 每一个小时执行一次
+// @Scheduled(cron = "0 0 0 * * ?")
+// @Scheduled(fixedRate = 1500000000)
+ public void dayOfCreateInspection() {
+ try {
+
+ // 查询任务模板
+ List<TTemplate> list = templateService.list();
+ if (CollectionUtils.isEmpty(list)) {
+ return;
+ }
+ list.forEach(template -> {
+ // 创建任务
+ createInspection(template);
+ });
+
+ // 查询前一天为执行的所有任务，改成已超时状态
+ List<TTask> taskCleanList = taskCleanService.list(Wrappers.lambdaQuery(TTask.class)
+ .like(TTask::getImplementTime, LocalDate.now().minusDays(1))
+ .eq(TTask::getStatus, 1));
+ if (!CollectionUtils.isEmpty(taskCleanList)) {
+ taskCleanList.forEach(task -> {
+ task.setStatus(2);
+ });
+ taskCleanService.updateBatchById(taskCleanList);
+ }
+
+ } catch (Exception e) {
+ e.printStackTrace();
+ }
+ }
+
+ public void createInspection(TTemplate template) {
+ // 查询所有的模板详情
+ List<TTemplateDetail> list = templateDetailService.list(Wrappers.lambdaQuery(TTemplateDetail.class)
+ .eq(TTemplateDetail::getTemplateId, template.getId()));
+ if (CollectionUtils.isEmpty(list)) {
+ return;
+ }
+ // 通过模板id查询员工巡检员
+ List<SysUser> sysUsers = sysUserService.selectUserByTempLateId(template.getId());
+ // 创建任务
+ for (TTemplateDetail detail : list) {
+ // 计算周期
+ int cycle = detail.getCycle();
+ switch (detail.getCycleType()){
+ case 2:
+ cycle = cycle * 7;
+ break;
+ case 3:
+ cycle = cycle * 30;
+ break;
+ case 4:
+ cycle = cycle * 90;
+ break;
+ case 5:
+ cycle = cycle * 365;
+ break;
+ }
+
+ // 拿到保洁抽查次数
+ int num1 = detail.getNum1();
+ // 拿到项目部数
+ int num2 = detail.getNum2();
+ // 拿到每日重复点位
+ int num3 = detail.getNum3();
+ // 获取点位类型的占比
+ String num4 = detail.getNum4();
+
+ // 未绑定员工
+ if(CollectionUtils.isEmpty(sysUsers)){
+ continue;
+ }
+
+ int taskCount = 0;
+ for (SysUser sysUser : sysUsers) {
+ List<TProjectDept> projectDeptLists = new ArrayList<>();
+ if(sysUser.getDeptType() == 1){
+ TProjectDept projectDept = projectDeptService.getById(sysUser.getDeptId());
+ if("0".equals(projectDept.getParentId())){
+ projectDeptLists = projectDeptService.list(Wrappers.lambdaQuery(TProjectDept.class)
+ .eq(TProjectDept::getParentId, projectDept.getId()));
+ }else {
+ projectDeptLists.add(projectDept);
+ }
+ }else {
+ projectDeptLists = projectDeptService.list(Wrappers.lambdaQuery(TProjectDept.class)
+ .ne(TProjectDept::getParentId,0));
+ }
+ List<String> proDeptIds = projectDeptLists.stream().map(TProjectDept::getId).collect(Collectors.toList());
+
+ // 获取项目部在该模板详情中的权重
+ List<String> projectDeptIds = redisCache.getCacheList(detail.getId() + PROJECT_DEPT_WEIGHT);
+ // 获取项目部列表
+ List<TProjectDept> projectDeptList;
+ if(CollectionUtils.isEmpty(projectDeptIds)){
+ projectDeptList = projectDeptLists;
+ }else {
+ projectDeptList = projectDeptService.list(Wrappers.lambdaQuery(TProjectDept.class)
+ .ne(TProjectDept::getParentId,0)
+ .in(TProjectDept::getId, proDeptIds)
+ .notIn(TProjectDept::getId, projectDeptIds));
+ // 所过所有的项目部都被抽取了，则重新抽取，并且清空项目部权重
+ if(CollectionUtils.isEmpty(projectDeptList)){
+ projectDeptList = projectDeptLists;
+ redisCache.deleteObject(detail.getId() + PROJECT_DEPT_WEIGHT);
+ }
+ }
+ // 如果可抽取的项目部数不足，先抽取余下项目部后，再清空权重，重新抽取
+ List<TProjectDept> projectDepts = randomSelection(projectDeptList, num2);
+ if(projectDepts.size() < num2){
+ List<String> proIds = projectDepts.stream().map(TProjectDept::getId).collect(Collectors.toList());
+ List<TProjectDept> projectDeptList1 = projectDeptService.list(Wrappers.lambdaQuery(TProjectDept.class)
+ .notIn(TProjectDept::getId, proIds)
+ .ne(TProjectDept::getParentId,0));
+ redisCache.deleteObject(detail.getId() + PROJECT_DEPT_WEIGHT);
+ List<TProjectDept> projectDepts1 = randomSelection(projectDeptList1, num2 - projectDepts.size());
+ List<String> proIds1 = projectDepts1.stream().map(TProjectDept::getId).collect(Collectors.toList());
+ // 将已抽取的项目部id保存到redis中
+ redisCache.setCacheList(detail.getId() + PROJECT_DEPT_WEIGHT, proIds1);
+ projectDepts.addAll(projectDepts1);
+ }else {
+ List<String> proIds = projectDepts.stream().map(TProjectDept::getId).collect(Collectors.toList());
+ // 将已抽取的项目部id保存到redis中
+ redisCache.setCacheList(detail.getId() + PROJECT_DEPT_WEIGHT, proIds);
+ }
+ List<String> proIds = projectDepts.stream().map(TProjectDept::getId).collect(Collectors.toList());
+
+ // 拿到抽取的项目部下的所有保洁员
+// List<TProjectDept> tProjectDeptList = projectDeptService.list(Wrappers.lambdaQuery(TProjectDept.class)
+// .in(TProjectDept::getId, proIds));
+ // 获取片区id
+ List<String> areaIds = projectDepts.stream().map(TProjectDept::getId).collect(Collectors.toList());
+ if(CollectionUtils.isEmpty(areaIds)){
+ return;
+ }
+
+ // 计算每天需要抽取多少个保洁员
+ long count = cleanerService.count(Wrappers.lambdaQuery(TCleaner.class)
+ .in(TCleaner::getProjectId, areaIds));
+ int cleanerSums = num1 * Integer.parseInt(count + "");
+ // 获取每天需要抽取的保洁员数,向上取整
+ if (cleanerSums < cycle){
+ cleanerSums = cycle;
+ }
+ int dayCleanerCount = cleanerSums / cycle;
+
+ if((cycle == detail.getCurrentValue()) && cleanerSums % cycle != 0){
+ dayCleanerCount++;
+ }
+
+ // 获取保洁员权重
+ List<String> cleanerIds = redisCache.getCacheList(detail.getId() + CLEANER_WEIGHT);
+ // 获取保洁员列表
+ List<TCleaner> cleaners;
+ if(CollectionUtils.isEmpty(cleanerIds)){
+ cleaners = cleanerService.list(Wrappers.lambdaQuery(TCleaner.class)
+ .in(TCleaner::getProjectId, areaIds));
+ }else {
+ cleaners = cleanerService.list(Wrappers.lambdaQuery(TCleaner.class)
+ .in(TCleaner::getProjectId, areaIds)
+ .notIn(TCleaner::getId, cleanerIds));
+ if(CollectionUtils.isEmpty(cleaners)){
+ cleaners = cleanerService.list(Wrappers.lambdaQuery(TCleaner.class)
+ .in(TCleaner::getProjectId, areaIds));
+ redisCache.deleteObject(detail.getId() + CLEANER_WEIGHT);
+ }
+ }
+
+ // 抽取保洁员
+ List<TCleaner> tCleaners = randomSelection(cleaners, dayCleanerCount);
+ if(CollectionUtils.isEmpty(tCleaners)){
+ log.error("没有可抽取的保洁员,模板id为：{}",detail.getId());
+ continue;
+ }
+ if(tCleaners.size() < dayCleanerCount){
+ List<String> cleanIds = tCleaners.stream().map(TCleaner::getId).collect(Collectors.toList());
+ List<TCleaner> cleaners1 = cleanerService.list(Wrappers.lambdaQuery(TCleaner.class)
+ .in(TCleaner::getProjectId, areaIds)
+ .notIn(TCleaner::getId, cleanIds));
+ redisCache.deleteObject(detail.getId() + CLEANER_WEIGHT);
+ List<TCleaner> tCleaners1 = randomSelection(cleaners1, dayCleanerCount - tCleaners.size());
+ List<String> cleanIds1 = tCleaners1.stream().map(TCleaner::getId).collect(Collectors.toList());
+ // 将已抽取的保洁员id保存到redis中
+ cleanIds1.addAll(cleanIds);
+ redisCache.setCacheList(detail.getId() + CLEANER_WEIGHT, cleanIds1);
+ tCleaners.addAll(tCleaners1);
+ }else {
+ List<String> cleanIds = tCleaners.stream().map(TCleaner::getId).collect(Collectors.toList());
+ redisCache.setCacheList(detail.getId() + CLEANER_WEIGHT, cleanIds);
+ }
+
+ // 通过保洁员id查询点位
+ List<String> cleanersIds = tCleaners.stream().map(TCleaner::getId).collect(Collectors.toList());
+ List<TLocation> locationList = locationService.list(Wrappers.lambdaQuery(TLocation.class)
+ .in(TLocation::getLocationCleaner, cleanersIds));
+ // 查询点位类型
+ JSONArray jsonArray = JSONObject.parseArray(num4);
+ List<TLocation> tLocationList = new ArrayList<>();
+ for (Object o : jsonArray) {
+ JSONObject jsonObject = JSONObject.parseObject(o.toString());
+ String id = jsonObject.getString("id");
+ BigDecimal value = jsonObject.getBigDecimal("value");
+ List<TLocation> locations = locationList.stream().filter(tLocation -> tLocation.getLocationType().equals(id)).collect(Collectors.toList());
+ if(!CollectionUtils.isEmpty(locations)){
+ BigDecimal bigDecimal = new BigDecimal(locations.size()).multiply(value.divide(new BigDecimal(100), 2, BigDecimal.ROUND_DOWN)).setScale(0, BigDecimal.ROUND_UP);
+ int locationCount = bigDecimal.intValue();
+ if(locationCount > 0){
+ List<TLocation> tLocations = randomSelection(locations, locationCount);
+ tLocationList.addAll(tLocations);
+ }
+ }
+ }
+ tLocationList = tLocationList.stream().distinct().collect(Collectors.toList());
+ // 抽取重复点位
+ Integer currentValue = detail.getCurrentValue();
+ if(currentValue != cycle){
+ // 周期天数加一
+ detail.setCurrentValue(currentValue + 1);
+ // 获取重复点位
+ Set<TLocation> repeatLocation = redisCache.getCacheSet(detail.getId() + ":" + sysUser.getUserId() + REPEAT_LOCATION);
+ if(!CollectionUtils.isEmpty(repeatLocation)){
+ List<TLocation> locations = randomSelection(new ArrayList<>(tLocationList), num3);
+ tLocationList.addAll(locations);
+ }
+ Set<TLocation> locationSet = new HashSet<>(tLocationList);
+ redisCache.setCacheSet(detail.getId() + ":" + sysUser.getUserId() + REPEAT_LOCATION, locationSet);
+
+ }else {
+ // 设置当前周期为0
+ detail.setCurrentValue(0);
+ // 将重复点位置空
+ redisCache.deleteObject(detail.getId() + ":" + sysUser.getUserId() + REPEAT_LOCATION);
+ // 将项目部权重置空
+ redisCache.deleteObject(detail.getId() + PROJECT_DEPT_WEIGHT);
+ // 将保洁员权重置空
+ redisCache.deleteObject(detail.getId() + CLEANER_WEIGHT);
+ }
+
+ // 创建任务
+ List<TTask> tasks = new ArrayList<>();
+ for (TLocation tLocation : tLocationList) {
+ TTask task = new TTask();
+ // 获取保洁员
+ tCleaners.stream().filter(tCleaner -> tCleaner.getId().equals(tLocation.getLocationCleaner())).findFirst().ifPresent(tCleaner -> {
+ task.setProjectId(tCleaner.getProjectId());
+ task.setCleanerId(tCleaner.getId());
+ });
+ // 获取巡检员
+ task.setPatrolInspector(sysUser.getUserId().toString());
+ task.setPatrolInspectorDept(sysUser.getDeptId());
+ task.setUserId(sysUser.getUserId());
+ task.setStatus(1);
+ task.setLocationId(tLocation.getId());
+ task.setImplementTime(LocalDateTime.now().plusDays(1));
+ task.setTaskType(1);
+ task.setTemplateId(detail.getTemplateId());
+ String nameAndCode = CodeGenerateUtils.generateVolumeSn();
+ task.setTaskName(nameAndCode);
+ task.setTaskCode(nameAndCode);
+ tasks.add(task);
+ }
+
+ // 添加应生成任务数量
+ TTemplateCount templateCount = new TTemplateCount();
+ templateCount.setTemplateId(detail.getTemplateId());
+ // 查询所有的保洁员下面的点位
+ List<String> cleanIds = cleaners.stream().map(TCleaner::getId).collect(Collectors.toList());
+ List<TLocation> tLocations = locationService.list(Wrappers.lambdaQuery(TLocation.class)
+ .in(TLocation::getLocationCleaner, cleanIds));
+ templateCount.setTaskCount(tLocations.size());
+ templateCount.setUserId(sysUser.getUserId());
+ templateCountService.save(templateCount);
+ taskCleanService.saveBatch(tasks);
+ taskCount = taskCount + tasks.size();
+ }
+ template.setTaskCount(taskCount);
+ templateService.updateById(template);
+ templateDetailService.updateById(detail);
+ }
+ }
+
+ /**
+ * 从集合中随机抽取数量
+ * @param list
+ * @param size
+ * @return
+ * @param <T>
+ */
+ private static <T> List<T> randomSelection(List<T> list, int size) {
+ List<T> selected = new ArrayList<>();
+ int length = list.size();
+ for (int i = 0; i < size; i++) {
+ Collections.shuffle(list);
+ // 如果需求的大小超过了集合的长度，则只能取到集合的全部元素
+ int subListSize = Math.min(size - selected.size(), length);
+ selected.addAll(list.subList(0, subListSize));
+ }
+ return selected;
+ }
+
+}
--
Gitblit v1.7.1