blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
龚焕茏
2024-08-02 5e859a25ddf7d61753e5849b3c6ef36cb0cbf68c 
 ycl-server/src/main/java/com/ycl/platform/service/impl/WorkOrderServiceImpl.java


@@ -1,35 +1,53 @@


package com.ycl.platform.service.impl;




import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;


import com.baomidou.mybatisplus.core.metadata.IPage;


import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;


import com.baomidou.mybatisplus.extension.conditions.update.LambdaUpdateChainWrapper;


import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;


import com.ycl.exception.ServiceException;


import com.ycl.platform.domain.entity.*;


import com.ycl.platform.domain.form.*;


import com.ycl.platform.domain.query.DistributeWorkOrderQuery;


import com.ycl.platform.domain.vo.DistributeWorkOrderVO;


import com.ycl.platform.mapper.*;


import com.ycl.platform.service.WorkOrderService;


import com.ycl.system.Result;


import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;


import com.ycl.platform.domain.vo.WorkOrderVO;


import com.ycl.platform.domain.query.HomeQuery;


import com.ycl.platform.domain.query.ScreenQuery;


import com.ycl.platform.domain.query.WorkOrderQuery;


import java.util.List;




import com.baomidou.mybatisplus.core.metadata.IPage;


import com.ycl.platform.domain.vo.DistributeWorkOrderVO;


import com.ycl.platform.domain.vo.WorkOrderVO;


import com.ycl.platform.domain.vo.WorkOrderYwConditionRecordVO;


import com.ycl.platform.domain.vo.screen.ScreenWorkOrderVO;


import com.ycl.platform.mapper.WorkOrderAuditingRecordMapper;


import com.ycl.platform.mapper.WorkOrderDistributeRecordMapper;


import com.ycl.platform.mapper.WorkOrderMapper;


import com.ycl.platform.mapper.WorkOrderYwConditionRecordMapper;


import com.ycl.platform.service.NotifyService;


import com.ycl.platform.service.WorkOrderAuditingRecordService;


import com.ycl.platform.service.WorkOrderService;


import com.ycl.platform.service.YwPointService;


import com.ycl.system.Result;


import com.ycl.system.model.LoginUser;


import com.ycl.system.page.PageUtil;


import com.ycl.utils.DateUtils;


import enumeration.general.FastDistributeTypeEnum;


import com.ycl.utils.SecurityUtils;


import com.ycl.utils.redis.RedisCache;


import com.ycl.utils.uuid.IdUtils;


import enumeration.general.NotifyTypeEnum;


import enumeration.general.UrgentLevelEnum;


import enumeration.general.WorkOrderDistributeWayEnum;


import enumeration.general.WorkOrderStatusEnum;


import org.springframework.stereotype.Service;


import lombok.RequiredArgsConstructor;


import org.apache.commons.lang3.ObjectUtils;


import org.springframework.beans.BeanUtils;


import org.springframework.beans.factory.annotation.Autowired;


import org.springframework.stereotype.Service;


import org.springframework.transaction.annotation.Transactional;


import org.springframework.util.Assert;


import java.util.ArrayList;


import java.util.Objects;


import java.util.stream.Collectors;




import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;


import lombok.RequiredArgsConstructor;


import org.springframework.util.StringUtils;




import java.time.LocalDateTime;


import java.time.ZoneId;


import java.util.*;


import java.util.stream.Collectors;




/**


 * 工单 服务实现类


@@ -42,11 +60,18 @@


public class WorkOrderServiceImpl extends ServiceImpl<WorkOrderMapper, WorkOrder> implements WorkOrderService {




    private final WorkOrderMapper workOrderMapper;


    private final YwUnitMapper ywUnitMapper;


    private final YwPeopleMapper ywPeopleMapper;


    private final YwPointService ywPointService;


    private final WorkOrderAuditingRecordMapper workOrderAuditingRecordMapper;


    private final WorkOrderAuditingRecordService workOrderAuditingRecordService;


    private final WorkOrderYwConditionRecordMapper workOrderYwConditionRecordMapper;


    private final NotifyService notifyService;


    private final WorkOrderDistributeRecordMapper workOrderDistributeRecordMapper;




    private final String DISTRIBUTE_LOCK_KEY = "distributeLock";


    private final static String IMPORTANT = "important";




    @Autowired


    private RedisCache redisCache;




    /**


     * 添加


@@ -56,6 +81,9 @@


    @Override


    public Result add(WorkOrderForm form) {


        WorkOrder entity = WorkOrderForm.getEntityByForm(form, null);


        entity.setWorkOrderNo(IdUtils.timeAddRandomNO(5));


        entity.setCreateTime(DateUtils.getNowDate());


        entity.setStatus(WorkOrderStatusEnum.WAIT_DISTRIBUTE);


        if(baseMapper.insert(entity) > 0) {


            return Result.ok("添加成功");


        }


@@ -85,23 +113,70 @@


    @Transactional(rollbackFor = Exception.class)


    public Result auditing(WorkOrderAuditingForm form) {


        WorkOrder workOrder = baseMapper.selectById(form.getId());


        if (Objects.isNull(workOrder)) {


            throw new ServiceException("工单不存在");


        }


        // 工单状态


        if ("pass".equals(form.getAuditingResult())) {


            workOrder.setStatus(WorkOrderStatusEnum.FINISH);


        } else {


            workOrder.setStatus(WorkOrderStatusEnum.AUDITING);


        }


        workOrder.setStatus(form.getAuditingResult());


        baseMapper.updateById(workOrder);


        // 添加一条审核记录


        WorkOrderAuditingRecord workOrderAuditingRecord = new WorkOrderAuditingRecord();


        workOrderAuditingRecord.setWorkOrderId(workOrder.getId());


        workOrderAuditingRecord.setAuditingUser(1);


        workOrderAuditingRecord.setResult(form.getAuditingResult());


        workOrderAuditingRecord.setAuditingUser(SecurityUtils.getLoginUser().getUserId().intValue());


        workOrderAuditingRecord.setResult(form.getAuditingResult().getDesc());


        workOrderAuditingRecord.setRemark(form.getAuditingRemark());


        workOrderAuditingRecordMapper.insert(workOrderAuditingRecord);


        // 添加新通知


        Notify notify = Notify.genEntityByUnit(NotifyTypeEnum.WORK_ORDER,


                form.getAuditingResult().getDesc(),


                workOrder.getUnitId(),


                UrgentLevelEnum.WARNING,


                workOrder.getWorkOrderNo());


        notifyService.save(notify);


        // 同步点位状态


        if (form.getAuditingResult() == WorkOrderStatusEnum.AUDITING_SUCCESS) {


            ywPointService.updateRecovery(Collections.singletonList(workOrder.getPointId()), 0);


        }


        return Result.ok("操作成功");


    }




    @Override


    @Transactional(rollbackFor = Exception.class)


    public Result batchAuditing(WorkOrderBatchAuditingForm form) {


        // 根据故障类型获取列表


        List<WorkOrder> list = new LambdaQueryChainWrapper<>(baseMapper)


                .in(WorkOrder::getStatus, WorkOrderStatusEnum.YW_HANDLE.getValue())


                .in(WorkOrder::getErrorType, form.getErrorTypes())


                .select(WorkOrder::getId, WorkOrder::getUnitId, WorkOrder::getWorkOrderNo)


                .list();


        if (list.isEmpty()) {


            return Result.error("没有工单可以审核");


        }


        List<Integer> ids = list.stream().map(WorkOrder::getId).collect(Collectors.toList());


        // 工单状态


        LambdaUpdateWrapper<WorkOrder> lambdaUpdateWrapper = new LambdaUpdateWrapper<>();


        lambdaUpdateWrapper.in(WorkOrder::getId, ids);


        lambdaUpdateWrapper.set(WorkOrder::getStatus, form.getAuditingResult());


        baseMapper.update(lambdaUpdateWrapper);


        // 添加多条审核记录


        List<WorkOrderAuditingRecord> workOrderAuditingRecords = new ArrayList<>();


        for (Integer id : ids) {


            WorkOrderAuditingRecord workOrderAuditingRecord = new WorkOrderAuditingRecord();


            workOrderAuditingRecord.setWorkOrderId(id);


            workOrderAuditingRecord.setAuditingUser(SecurityUtils.getLoginUser().getUserId().intValue());


            workOrderAuditingRecord.setResult(form.getAuditingResult().getDesc());


            workOrderAuditingRecord.setRemark(form.getAuditingRemark());


            workOrderAuditingRecords.add(workOrderAuditingRecord);


        }


        workOrderAuditingRecordService.saveBatch(workOrderAuditingRecords);


        // 添加新通知


        List<Notify> notifies = new ArrayList<>();


        for (WorkOrder workOrder : list) {


            Notify notify = Notify.genEntityByUnit(NotifyTypeEnum.WORK_ORDER,


                    form.getAuditingResult().getDesc(),


                    workOrder.getUnitId(),


                    UrgentLevelEnum.WARNING,


                    workOrder.getWorkOrderNo());


            notifies.add(notify);


        }


        notifyService.saveBatch(notifies);


        return Result.ok("操作成功");


    }




@@ -113,15 +188,32 @@


            throw new ServiceException("工单不存在");


        }


        // 工单状态


//        workOrder.setStatus("commit-condition");


        workOrder.setStatus(WorkOrderStatusEnum.YW_HANDLE);


        workOrder.setYwHandleTime(LocalDateTime.now());


        baseMapper.updateById(workOrder);


        // 添加一条运维情况记录


        WorkOrderYwConditionRecord workOrderYwConditionRecord = new WorkOrderYwConditionRecord();


        workOrderYwConditionRecord.setWorkOrderId(workOrder.getId());


        workOrderYwConditionRecord.setCommitUser(1);


        workOrderYwConditionRecord.setCommitUser(SecurityUtils.getLoginUser().getUserId().intValue());


        workOrderYwConditionRecord.setYwCondition(form.getYwCondition());


        workOrderYwConditionRecord.setYwProofMaterials(form.getYwProofMaterials());


        workOrderYwConditionRecordMapper.insert(workOrderYwConditionRecord);


        return Result.ok("操作成功");


    }




    @Override


    public Result selectYwConditionByYwId(Integer id) {


        List<WorkOrderYwConditionRecordVO> ywConditionList = workOrderYwConditionRecordMapper.selectYwConditionByYwId(id);


        return Result.ok().data(ywConditionList);


    }




    @Override


    public Result selectYwAuditingListByYwId(String id) {


        return Result.ok().data(


                new LambdaQueryChainWrapper<>(workOrderAuditingRecordMapper)


                        .eq(WorkOrderAuditingRecord::getWorkOrderId, id)


                        .orderByAsc(WorkOrderAuditingRecord::getCreateTime)


                        .list());


    }




    @Override


@@ -167,74 +259,140 @@


     */


    @Override


    public Result page(WorkOrderQuery query) {


        IPage<WorkOrder> page = new LambdaQueryChainWrapper<>(baseMapper)


                .eq(StringUtils.hasText(query.getWorkOrderNo()), WorkOrder::getWorkOrderNo, query.getWorkOrderNo())


                .ne(WorkOrder::getStatus, WorkOrderStatusEnum.WAIT_DISTRIBUTE)


                .between(Objects.nonNull(query.getStart()) && Objects.nonNull(query.getEnd()),


                        WorkOrder::getYwHandleTime,


                        DateUtils.getDayStart(query.getStart()),


                        DateUtils.getDayEnd(query.getEnd()))


                .orderByDesc(WorkOrder::getCreateTime)


                .page(PageUtil.getPage(query, WorkOrder.class));




        List<WorkOrderVO> vos = page.getRecords().stream()


                .map(


                        entity -> {


                            WorkOrderVO vo = WorkOrderVO.getVoByEntity(entity, null);


                            YwUnit unit = ywUnitMapper.selectById(vo.getUnitId());


                            if (Objects.nonNull(unit)) {


                                vo.setUnitName(unit.getUnitName());


                            }


                            YwPeople ywPeople = ywPeopleMapper.selectById(vo.getYwPeopleId());


                            if (Objects.nonNull(ywPeople)) {


                                vo.setYwPeopleName(ywPeople.getYwPersonName());


                            }


                            return vo;


                        }


                )


                .collect(Collectors.toList());


        return Result.ok().data(vos).total(page.getTotal());


        IPage<WorkOrderVO> page = PageUtil.getPage(query, WorkOrderVO.class);


        baseMapper.page(page, query);


        return Result.ok().data(page.getRecords()).total(page.getTotal());


    }




    @Override


    public Result distributePage(DistributeWorkOrderQuery query) {


        IPage<WorkOrder> page = new LambdaQueryChainWrapper<>(baseMapper)


                .eq(Objects.nonNull(query.getUnitId()), WorkOrder::getUnitId, query.getUnitId())


                .eq(WorkOrder::getStatus, WorkOrderStatusEnum.WAIT_DISTRIBUTE)


                .orderByDesc(WorkOrder::getCreateTime)


                .page(PageUtil.getPage(query, WorkOrder.class));


        List<WorkOrderVO> vos = page.getRecords().stream()


                .map(


                        entity -> {


                            WorkOrderVO vo = WorkOrderVO.getVoByEntity(entity, null);


                            YwUnit unit = ywUnitMapper.selectById(vo.getUnitId());


                            if (Objects.nonNull(unit)) {


                                vo.setUnitName(unit.getUnitName());


                            }


                            YwPeople ywPeople = ywPeopleMapper.selectById(vo.getYwPeopleId());


                            if (Objects.nonNull(ywPeople)) {


                                vo.setYwPeopleName(ywPeople.getYwPersonName());


                            }


                            return vo;


                        }


                )


                .collect(Collectors.toList());


        return Result.ok().data(vos).total(page.getTotal());


        IPage<WorkOrderVO> page = PageUtil.getPage(query, WorkOrderVO.class);


        baseMapper.distributePage(page, query);


        page.getRecords().stream().forEach(workOrder -> {


            if (StringUtils.hasText(workOrder.getPointTag()) && workOrder.getPointTag().contains(IMPORTANT)) {


                workOrder.setImportant(Boolean.TRUE);


            }


        });


        return Result.ok().data(page).total(page.getTotal());


    }




    @Override


    @Transactional


    public Result distributeFast(DistributeWorkOrderVO data) {


        List<WorkOrder> willDistributeList = new ArrayList<>(32);


        // 如果是自定义时间范围


        if (FastDistributeTypeEnum.CUSTOM == data.getFastWay()) {


        // 获取当前时间


        LocalDateTime now = LocalDateTime.now(ZoneId.systemDefault());


        data.setEnd(now);


        switch (data.getFastWay()) {


            case LAST_HALF_HOUR:


                data.setStart(now.minusMinutes(30));


                break;


            case LAST_HOUR:


                data.setStart(now.minusHours(1));


                break;


            case LAST_TWO_HOUR:


               data.setStart(now.minusHours(2));


               break;


            case LAST_DAY:


                data.setStart(now.minusDays(1));


                break;


           }




        // 查询符合条件的工单


        List<WorkOrder> list = new LambdaQueryChainWrapper<>(baseMapper)


                .select(WorkOrder::getId, WorkOrder::getPointId)


                .eq(WorkOrder::getStatus, WorkOrderStatusEnum.WAIT_DISTRIBUTE)


                .eq(Objects.nonNull(data.getUnitId()), WorkOrder::getUnitId, data.getUnitId())


                .eq(WorkOrder::getErrorType, data.getErrorType())


                .between(WorkOrder::getCreateTime, data.getStart(), data.getEnd())


                .orderByDesc(WorkOrder::getCreateTime)


                .last("limit " + data.getFastNumLimit())


                .list();


        List<Integer> ids = list.stream().map(WorkOrder::getId).toList();


        List<Integer> pointIds = list.stream().map(WorkOrder::getPointId).toList();




        if (ids.isEmpty()) { return Result.error("没有符合条件的工单"); }


        if (!getDistributeLock()) { return Result.error("此刻有人下发中，为避免冲突，请稍后重试"); }


        try {


            new LambdaUpdateChainWrapper<>(baseMapper)


                    .set(WorkOrder::getStatus, WorkOrderStatusEnum.DISTRIBUTED)


                    .in(WorkOrder::getId, ids)


                    .update();


            addDistributeRecord(ids, WorkOrderDistributeWayEnum.FAST_DISTRIBUTE);


            // 同步点位状态


            ywPointService.updateRecovery(pointIds, 1);


            return Result.ok("成功下发" + ids.size() + "条工单");


        } catch (Exception e) {


            return Result.error("操作失败");


        } finally {


            distributeUnLock();


        }


    }




        new LambdaUpdateChainWrapper<>(baseMapper)


                .set(WorkOrder::getStatus, WorkOrderStatusEnum.DISTRIBUTED)


                .update();


    @Override


    @Transactional


    public Result selectedIdsDistribute(DistributeWorkOrderQuery query) {


        WorkOrderDistributeWayEnum distributeWayEnum = WorkOrderDistributeWayEnum.SELECTED_DISTRIBUTE;


        if (!getDistributeLock()) { return Result.error("此刻有人下发中，为避免冲突，请稍后重试"); }


        try {


            if (query.getIds().isEmpty()) {


                query.setIds(new LambdaQueryChainWrapper<>(baseMapper)


                        .eq(WorkOrder::getStatus, WorkOrderStatusEnum.WAIT_DISTRIBUTE)


                        .eq(Objects.nonNull(query.getUnitId()), WorkOrder::getUnitId, query.getUnitId())


                        .select(WorkOrder::getId)


                        .list()


                        .stream()


                        .map(WorkOrder::getId)


                        .collect(Collectors.toList()));


                distributeWayEnum = WorkOrderDistributeWayEnum.ALL_DISTRIBUTE;


            }


            if (query.getIds().isEmpty()) { return Result.error("没有工单待下发"); }


            new LambdaUpdateChainWrapper<>(baseMapper)


                    .set(WorkOrder::getStatus, WorkOrderStatusEnum.DISTRIBUTED)


                    .in(WorkOrder::getId, query.getIds())


                    .update();


            addDistributeRecord(query.getIds(), distributeWayEnum);


            // 同步点位状态


            List<Integer> pointIds = new LambdaQueryChainWrapper<>(baseMapper).select(WorkOrder::getPointId).in(WorkOrder::getId, query.getIds()).list().stream().map(WorkOrder::getPointId).toList();


            ywPointService.updateRecovery(pointIds, 1);


            return Result.ok("成功下发" + query.getIds().size() + "条工单");


        } catch (Exception e) {


            return Result.error("操作失败");


        } finally {


            distributeUnLock();


        }


    }




        return Result.ok("操作成功");


    /**


     * 申请工单下发锁


     * @return 工单下发锁申请结果


     */


    public synchronized Boolean getDistributeLock() {


        if (Objects.isNull(redisCache.getCacheObject(DISTRIBUTE_LOCK_KEY))) {


            redisCache.setCacheObject(DISTRIBUTE_LOCK_KEY, "1");


            return true;


        } else {


            return false;


        }


    }




    /**


     * 工单下发锁释放


     */


    public synchronized void distributeUnLock() {


        redisCache.deleteObject(DISTRIBUTE_LOCK_KEY);


    }




    /**


     * 添加工单下发记录


     * @param ids 工单id


     */


    private void addDistributeRecord(List<Integer> ids, WorkOrderDistributeWayEnum distributeWay) {


        LoginUser loginUser = SecurityUtils.getLoginUser();


        workOrderDistributeRecordMapper.insertBatch(


                ids.stream()


                        .map(id -> new WorkOrderDistributeRecord(id, loginUser.getUserId(), distributeWay))


                        .toList()


        );


    }




    /**


@@ -265,4 +423,33 @@


                .collect(Collectors.toList());


        return Result.ok().data(vos);


    }




    @Override


    public Result screenWorkOrder(ScreenQuery query) {


        ScreenWorkOrderVO screen = baseMapper.screenWorkOrder(query);


        return Result.ok().data(screen);


    }




    @Override


    public Map<String, Object> home(HomeQuery monitorQuery) {


        Map<String, Object> dataMap = new HashMap<>();


        Map<String, Object> data1 = new HashMap<>();


        Map<String, Object> data2 = new HashMap<>();


        Map<String, Object> data3 = new HashMap<>();


        List<Map<String, Object>> home = baseMapper.home(monitorQuery);


        if (ObjectUtils.isNotEmpty(home)) {


            for (Map<String, Object> map : home) {


                if (Objects.nonNull(map.get("dateType")) && StringUtils.hasText(map.get("dateType").toString())) {


                    data1.put(map.get("dateType").toString(), map.get("num1"));


                    data2.put(map.get("dateType").toString(), map.get("num2"));


                    data3.put(map.get("dateType").toString(), map.get("num3"));


                }


            }


            dataMap.put("name", home.get(0).get("name"));


        }


        dataMap.put("complete", data1);


        dataMap.put("waiting", data2);


        dataMap.put("pending", data3);


        return dataMap;


    }


}