From 70dfa087a9f61ecc33f3b5bc618202c2023e38f3 Mon Sep 17 00:00:00 2001
From: xiangpei <xiangpei@timesnew.cn>
Date: 星期一, 19 八月 2024 15:45:26 +0800
Subject: [PATCH] 优云接口工单生成

---
 ycl-server/src/main/java/com/ycl/platform/service/impl/WorkOrderServiceImpl.java |  485 ++++++++++++++++++++++++++++++++++++++++++++++++-----
 1 files changed, 439 insertions(+), 46 deletions(-)

diff --git a/ycl-server/src/main/java/com/ycl/platform/service/impl/WorkOrderServiceImpl.java b/ycl-server/src/main/java/com/ycl/platform/service/impl/WorkOrderServiceImpl.java
index 26b738e..66e3094 100644
--- a/ycl-server/src/main/java/com/ycl/platform/service/impl/WorkOrderServiceImpl.java
+++ b/ycl-server/src/main/java/com/ycl/platform/service/impl/WorkOrderServiceImpl.java
@@ -1,33 +1,50 @@
 package com.ycl.platform.service.impl;
 
-import com.ycl.platform.domain.entity.WorkOrder;
-import com.ycl.platform.domain.entity.YwPeople;
-import com.ycl.platform.domain.entity.YwUnit;
-import com.ycl.platform.mapper.WorkOrderMapper;
-import com.ycl.platform.mapper.YwPeopleMapper;
-import com.ycl.platform.mapper.YwUnitMapper;
-import com.ycl.platform.service.WorkOrderService;
-import com.ycl.system.Result;
-import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
-import com.ycl.platform.domain.form.WorkOrderForm;
-import com.ycl.platform.domain.vo.WorkOrderVO;
-import com.ycl.platform.domain.query.WorkOrderQuery;
-import java.util.List;
-
+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.*;
+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.domain.vo.screen.WorkOrderRegionVO;
+import com.ycl.platform.domain.vo.screen.WorkOrderTotalVO;
+import com.ycl.platform.mapper.*;
+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.domain.SysConfig;
+import com.ycl.system.mapper.SysConfigMapper;
+import com.ycl.system.model.LoginUser;
 import com.ycl.system.page.PageUtil;
-import org.springframework.stereotype.Service;
-import org.springframework.security.core.context.SecurityContextHolder;
+import com.ycl.utils.DateUtils;
+import com.ycl.utils.SecurityUtils;
+import com.ycl.utils.redis.RedisCache;
+import com.ycl.utils.uuid.IdUtils;
+import enumeration.general.*;
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+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.plugins.pagination.Page;
-import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
-import lombok.RequiredArgsConstructor;
+import org.springframework.util.CollectionUtils;
 import org.springframework.util.StringUtils;
+
+import java.time.LocalDateTime;
+import java.time.ZoneId;
+import java.util.*;
+import java.util.stream.Collectors;
 
 /**
  * 宸ュ崟 鏈嶅姟瀹炵幇绫�
@@ -35,14 +52,122 @@
  * @author xp
  * @since 2024-03-05
  */
+@Slf4j
 @Service
 @RequiredArgsConstructor
 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 TMonitorMapper monitorMapper;
+    private final SysConfigMapper configMapper;
 
+    private final String DISTRIBUTE_LOCK_KEY = "distributeLock";
+    private final static String IMPORTANT = "important";
+
+    @Autowired
+    private RedisCache redisCache;
+
+    @Override
+    @Transactional(rollbackFor = Exception.class)
+    public Boolean innerAddWorkOrder(List<WorkOrder> workOrderList) {
+        int total = workOrderList.size();
+        workOrderList.stream().filter(item -> {
+            return StringUtils.hasText(item.getSerialNumber()) && Objects.nonNull(item.getStatus()) && StringUtils.hasText(item.getErrorType());
+        });
+        if (CollectionUtils.isEmpty(workOrderList)) {
+            return Boolean.TRUE;
+        }
+        List<String> serialNumberList = workOrderList.stream().map(WorkOrder::getSerialNumber).collect(Collectors.toList());
+        // 鏌ュ嚭鏁版嵁搴撲腑鍥芥爣鐮佸搴旂殑鏈畬鎴愮殑宸ュ崟
+        List<WorkOrder> inDatabaseWorkOrderList = new LambdaQueryChainWrapper<>(baseMapper)
+                .select(WorkOrder::getId, WorkOrder::getSerialNumber, WorkOrder::getStatus, WorkOrder::getErrorType)
+                .in(WorkOrder::getSerialNumber, serialNumberList)
+                .ne(WorkOrder::getStatus, WorkOrderStatusEnum.AUDITING_SUCCESS)
+                .list();
+        Map<String, WorkOrder> mapping = inDatabaseWorkOrderList.stream().collect((Collectors.toMap(WorkOrder::getSerialNumber, workOrder -> workOrder)));
+        List<WorkOrder> waitAddList = new ArrayList<>(48);
+        Integer updateNum = 0;
+        // TODO 鏌ヨ鏄惁鎶ュ杩囷紝鏇存柊鏁呴殰绫诲瀷
+        for (WorkOrder workOrder : workOrderList) {
+            WorkOrder databaseWorkOrder = mapping.get(workOrder.getSerialNumber());
+            if (Objects.nonNull(databaseWorkOrder)) {
+                String errorType = databaseWorkOrder.getErrorType();
+                if (workOrder.getErrorType().equals(errorType)) {
+                    // 濡傛灉锛屽浗鏍囩爜銆佹晠闅滅被鍨嬮兘涓�鏍凤紝鍒欒烦杩囦笉澶勭悊
+                    continue;
+                } else {
+                    // 閿欒绫诲瀷涓嶄竴鏍凤紝灏辨洿鏂伴敊璇被鍨嬶紝骞朵笖閲嶇疆宸ュ崟鐘舵�佷负寰呭鐞�
+                    errorType = errorType + "," + workOrder.getErrorType();
+                    databaseWorkOrder.setErrorType(errorType);
+                    databaseWorkOrder.setStatus(WorkOrderStatusEnum.DISTRIBUTED);
+                    Date now = new Date();
+                    databaseWorkOrder.setUpdateTime(now);
+                    baseMapper.updateById(databaseWorkOrder);
+                    updateNum++;
+                    // 鍚屾椂鏂板涓�涓繍缁村鐞嗕俊鎭紝琛ㄦ槑姝ゅ伐鍗曡璋冩暣 TODO
+                }
+            } else {
+                workOrder.setCreateTime(new Date());
+                workOrder.setUpdateTime(new Date());
+                waitAddList.add(workOrder);
+            }
+        }
+        if (CollectionUtils.isEmpty(waitAddList)) {
+            return Boolean.TRUE;
+        }
+        List<String> willAddSerialNumber = waitAddList.stream().map(WorkOrder::getSerialNumber).collect(Collectors.toList());
+        List<YwPoint> pointList = new LambdaQueryChainWrapper<>(ywPointService.getBaseMapper())
+                .select(YwPoint::getUnitId, YwPoint::getSerialNumber, YwPoint::getImportantTag, YwPoint::getImportantTag, YwPoint::getProvinceTag, YwPoint::getImportantCommandImageTag)
+                .in(YwPoint::getSerialNumber, willAddSerialNumber)
+                .list();
+        Map<String, YwPoint> pointMapping = pointList.stream().collect(Collectors.toMap(YwPoint::getSerialNumber, point -> point));
+        // 鏌ュ嚭閲嶇偣鐐逛綅銆佹櫘閫氱偣浣嶇殑澶勭悊鏃堕棿
+        SysConfig important = configMapper.checkConfigKeyUnique("important.wordkorder.time");
+        SysConfig normal = configMapper.checkConfigKeyUnique("normal.wordkorder.alarm.time");
+        // 濡傛灉鍗冲皢鐢熸垚宸ュ崟锛屼絾鏄澶囧浗鏍囩爜鏌ヤ笉鍒扮偣浣嶏紝鍒欎笉娣诲姞锛�
+        List<WorkOrder> notAddList = new ArrayList<>();
+        for (WorkOrder workOrder : waitAddList) {
+            YwPoint point = pointMapping.get(workOrder.getSerialNumber());
+            if (Objects.isNull(point)) {
+                notAddList.add(workOrder);
+                continue;
+            }
+            workOrder.setWorkOrderNo(IdUtils.timeAddRandomNO(5));
+            workOrder.setUnitId(Math.toIntExact(point.getUnitId()));
+            if (point.getImportantTag() || point.getImportantCommandImageTag()) {
+                workOrder.setProcessingPeriod(Integer.valueOf(important.getConfigValue()));
+            } else {
+                workOrder.setProcessingPeriod(Integer.valueOf(normal.getConfigValue()));
+            }
+        }
+        waitAddList.removeAll(notAddList);
+        if (CollectionUtils.isEmpty(waitAddList)) {
+            return Boolean.TRUE;
+        }
+        this.saveBatch(waitAddList);
+        // 濡傛灉鏄洿鎺ヤ笅鍙戯紝娣诲姞涓嬪彂璁板綍
+        if (WorkOrderStatusEnum.DISTRIBUTED.equals(waitAddList.get(0).getStatus())) {
+            List<WorkOrderDistributeRecord> distributedRecordList = waitAddList.stream().map(item -> {
+                WorkOrderDistributeRecord workOrderDistributeRecord = new WorkOrderDistributeRecord();
+                workOrderDistributeRecord.setWorkOrderId(item.getId());
+                workOrderDistributeRecord.setDistributeWay(WorkOrderDistributeWayEnum.DIRECT_DISTRIBUTE);
+                workOrderDistributeRecord.setUserId(1L);
+                Date now = new Date();
+                workOrderDistributeRecord.setCreateTime(now);
+                workOrderDistributeRecord.setUpdateTime(now);
+                return workOrderDistributeRecord;
+            }).collect(Collectors.toList());
+            workOrderDistributeRecordMapper.insertBatch(distributedRecordList);
+        }
+        log.info("浼犲叆宸ュ崟鎬绘暟: {}锛屽疄闄呮坊鍔犲伐鍗曟暟锛歿}, 瀹為檯淇敼宸ュ崟鏁帮細{}", total, waitAddList.size(), updateNum);
+        return Boolean.TRUE;
+    }
 
     /**
      * 娣诲姞
@@ -52,6 +177,12 @@
     @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);
+        Date now = new Date();
+        entity.setCreateTime(now);
+        entity.setUpdateTime(now);
         if(baseMapper.insert(entity) > 0) {
             return Result.ok("娣诲姞鎴愬姛");
         }
@@ -67,14 +198,132 @@
     public Result update(WorkOrderForm form) {
 
         WorkOrder entity = baseMapper.selectById(form.getId());
-
         // 涓虹┖鎶汭llegalArgumentException锛屽仛鍏ㄥ眬寮傚父澶勭悊
         Assert.notNull(entity, "璁板綍涓嶅瓨鍦�");
         BeanUtils.copyProperties(form, entity);
+        Date now = new Date();
+        entity.setUpdateTime(now);
         if (baseMapper.updateById(entity) > 0) {
             return Result.ok("淇敼鎴愬姛");
         }
         return Result.error("淇敼澶辫触");
+    }
+
+    @Override
+    @Transactional(rollbackFor = Exception.class)
+    public Result auditing(WorkOrderAuditingForm form) {
+        WorkOrder workOrder = baseMapper.selectById(form.getId());
+        // 宸ュ崟鐘舵��
+        workOrder.setStatus(form.getAuditingResult());
+        baseMapper.updateById(workOrder);
+        // 娣诲姞涓�鏉″鏍歌褰�
+        WorkOrderAuditingRecord workOrderAuditingRecord = new WorkOrderAuditingRecord();
+        workOrderAuditingRecord.setWorkOrderId(workOrder.getId());
+        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(Arrays.asList(workOrder.getSerialNumber()), 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("鎿嶄綔鎴愬姛");
+    }
+
+    @Override
+    @Transactional(rollbackFor = Exception.class)
+    public Result ywCondition(WorkOrderYWConditionForm form) {
+        WorkOrder workOrder = baseMapper.selectById(form.getId());
+        if (Objects.isNull(workOrder)) {
+            throw new ServiceException("宸ュ崟涓嶅瓨鍦�");
+        }
+        // 宸ュ崟鐘舵��
+        workOrder.setStatus(WorkOrderStatusEnum.YW_HANDLE);
+        workOrder.setYwHandleTime(LocalDateTime.now());
+        baseMapper.updateById(workOrder);
+        // 娣诲姞涓�鏉¤繍缁存儏鍐佃褰�
+        WorkOrderYwConditionRecord workOrderYwConditionRecord = new WorkOrderYwConditionRecord();
+        workOrderYwConditionRecord.setWorkOrderId(workOrder.getId());
+        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
+    public Result ywResult(WorkOrderYWResultForm form) {
+        return null;
+    }
+
+    @Override
+    public Result checkResult(WorkOrderCheckResultForm form) {
+        return null;
     }
 
     /**
@@ -110,30 +359,135 @@
      */
     @Override
     public Result page(WorkOrderQuery query) {
+        IPage<WorkOrderVO> page = PageUtil.getPage(query, WorkOrderVO.class);
+        baseMapper.page(page, query);
+        return Result.ok().data(page.getRecords()).total(page.getTotal());
+    }
 
-        IPage<WorkOrder> page = new LambdaQueryChainWrapper<>(baseMapper)
-                .eq(StringUtils.hasText(query.getWorkOrderNo()), WorkOrder::getWorkOrderNo, query.getWorkOrderNo())
-                .between(Objects.nonNull(query.getStart()) && Objects.nonNull(query.getEnd()),WorkOrder::getYwHandleTime, query.getStart(), query.getEnd())
+    @Override
+    public Result distributePage(DistributeWorkOrderQuery query) {
+        IPage<WorkOrderVO> page = PageUtil.getPage(query, WorkOrderVO.class);
+        baseMapper.distributePage(page, query);
+        return Result.ok().data(page).total(page.getTotal());
+    }
+
+    @Override
+    @Transactional
+    public Result distributeFast(DistributeWorkOrderVO data) {
+        // 鑾峰彇褰撳墠鏃堕棿
+        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::getSerialNumber, WorkOrder::getSerialNumber)
+                .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)
-                .page(PageUtil.getPage(query, WorkOrder.class));
+                .last("limit " + data.getFastNumLimit())
+                .list();
+        List<Integer> ids = list.stream().map(WorkOrder::getId).toList();
+        List<String> serialNumberList = list.stream().map(WorkOrder::getSerialNumber).toList();
 
-        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());
+        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(serialNumberList, 1);
+            return Result.ok("鎴愬姛涓嬪彂" + ids.size() + "鏉″伐鍗�");
+        } catch (Exception e) {
+            return Result.error("鎿嶄綔澶辫触");
+        } finally {
+            distributeUnLock();
+        }
+    }
+
+    @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<String> serialNumberList = new LambdaQueryChainWrapper<>(baseMapper).select(WorkOrder::getSerialNumber).in(WorkOrder::getId, query.getIds()).list().stream().map(WorkOrder::getSerialNumber).toList();
+            ywPointService.updateRecovery(serialNumberList, 1);
+            return Result.ok("鎴愬姛涓嬪彂" + query.getIds().size() + "鏉″伐鍗�");
+        } catch (Exception e) {
+            return Result.error("鎿嶄綔澶辫触");
+        } finally {
+            distributeUnLock();
+        }
+    }
+
+    /**
+     * 鐢宠宸ュ崟涓嬪彂閿�
+     * @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()
+        );
     }
 
     /**
@@ -164,4 +518,43 @@
                 .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;
+    }
+
+    @Override
+    public WorkOrderTotalVO workOrderTotal(DashboardQuery dashboardQuery) {
+        return baseMapper.workOrderTotal(dashboardQuery);
+    }
+
+    @Override
+    public List<WorkOrderRegionVO> workOrderRegion(DashboardQuery dashboardQuery) {
+        return baseMapper.workOrderRegion(dashboardQuery);
+    }
 }

--
Gitblit v1.8.0