| | |
| | | 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 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.LocalDateTime; |
| | | import java.util.*; |
| | | import java.util.stream.Collectors; |
| | | |
| | | /** |
| | | * @author xiaochen |
| | | * @date 2025/6/11 18:39 |
| | | */ |
| | | @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); |
| | | }); |
| | | |
| | | } 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()); |
| | | // 创建任务 |
| | | list.forEach(detail -> { |
| | | // 计算周期 |
| | | 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(); |
| | | |
| | | SysUser sysUser = sysUsers.get(0); |
| | | 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()) == 1 && 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(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); |
| | | } |
| | | } |
| | | } |
| | | |
| | | // 抽取重复点位 |
| | | Integer currentValue = detail.getCurrentValue(); |
| | | if(currentValue != cycle){ |
| | | // 周期天数加一 |
| | | detail.setCurrentValue(currentValue + 1); |
| | | // 获取重复点位 |
| | | Set<TLocation> repeatLocation = redisCache.getCacheSet(detail.getId() + 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() + REPEAT_LOCATION, locationSet); |
| | | |
| | | }else { |
| | | // 设置当前周期为0 |
| | | detail.setCurrentValue(0); |
| | | // 将重复点位置空 |
| | | redisCache.deleteObject(detail.getId() + REPEAT_LOCATION); |
| | | } |
| | | |
| | | // 创建任务 |
| | | 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.getId()); |
| | | templateCount.setTaskCount(tLocationList.size()* cycle); |
| | | templateCount.setUserId(sysUser.getUserId()); |
| | | templateCountService.save(templateCount); |
| | | |
| | | taskCleanService.saveBatch(tasks); |
| | | template.setTaskCount(tasks.size()); |
| | | 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()); |
| | | // 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; |
| | | // } |
| | | // |
| | | //} |