From 042936ba89e122bece0e4ce595c68f859b9e9bed Mon Sep 17 00:00:00 2001
From: 龚焕茏 <2842157468@qq.com>
Date: 星期三, 17 四月 2024 17:53:43 +0800
Subject: [PATCH] 工单下发:新增互斥锁、下发记录

---
 ycl-server/src/main/java/com/ycl/platform/service/impl/WorkOrderServiceImpl.java |  299 +++++++++++++++++++++++++++++++++++++++++++++++++++++++----
 1 files changed, 276 insertions(+), 23 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..670853e 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,42 @@
 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.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.query.WorkOrderQuery;
+import com.ycl.platform.domain.vo.DistributeWorkOrderVO;
+import com.ycl.platform.domain.vo.WorkOrderVO;
+import com.ycl.platform.mapper.*;
+import com.ycl.platform.service.NotifyService;
 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.metadata.IPage;
+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 enumeration.general.NotifyTypeEnum;
+import enumeration.general.WorkOrderStatusEnum;
+import lombok.RequiredArgsConstructor;
+import org.apache.commons.lang3.ArrayUtils;
 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.StringUtils;
+
+import java.time.LocalDateTime;
+import java.time.ZoneId;
+import java.util.List;
+import java.util.Objects;
+import java.util.concurrent.TimeUnit;
+import java.util.stream.Collectors;
 
 /**
  * 宸ュ崟 鏈嶅姟瀹炵幇绫�
@@ -42,7 +51,15 @@
     private final WorkOrderMapper workOrderMapper;
     private final YwUnitMapper ywUnitMapper;
     private final YwPeopleMapper ywPeopleMapper;
+    private final WorkOrderAuditingRecordMapper workOrderAuditingRecordMapper;
+    private final WorkOrderYwConditionRecordMapper workOrderYwConditionRecordMapper;
+    private final NotifyService notifyService;
+    private final WorkOrderDistributeRecordMapper workOrderDistributeRecordMapper;
 
+    private final String DISTRIBUTE_LOCK_KEY = "distributeLock";
+
+    @Autowired
+    private RedisCache redisCache;
 
     /**
      * 娣诲姞
@@ -52,6 +69,10 @@
     @Override
     public Result add(WorkOrderForm form) {
         WorkOrder entity = WorkOrderForm.getEntityByForm(form, null);
+        entity.setWorkOrderNo(DateUtils.dateTimeNow());
+        entity.setCreateTime(DateUtils.getNowDate());
+        entity.setStatus(WorkOrderStatusEnum.WAIT_DISTRIBUTE);
+        entity.setDeleted("0");
         if(baseMapper.insert(entity) > 0) {
             return Result.ok("娣诲姞鎴愬姛");
         }
@@ -75,6 +96,88 @@
             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);
+        // 娣诲姞鏂伴�氱煡
+        try {
+            Notify notify = new Notify(NotifyTypeEnum.WORK_ORDER, form.getAuditingResult().getDesc(), workOrder.getYwPeopleId(), "0", "0", workOrder.getId());
+            notifyService.save(notify);
+        } catch (Exception e) {
+            return Result.error("瀹℃牳鎴愬姛锛岄�氱煡杩愮淮浜哄憳澶辫触");
+        }
+        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);
+        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(String id) {
+        return Result.ok().data(
+                new LambdaQueryChainWrapper<>(workOrderYwConditionRecordMapper)
+                        .eq(WorkOrderYwConditionRecord::getWorkOrderId, id)
+                        .orderByAsc(WorkOrderYwConditionRecord::getCreateTime)
+                        .last("limit 1")
+                        .one());
+    }
+
+    @Override
+    public Result selectYwConditionListByYwId(String id) {
+        return Result.ok().data(
+                new LambdaQueryChainWrapper<>(workOrderYwConditionRecordMapper)
+                        .eq(WorkOrderYwConditionRecord::getWorkOrderId, id)
+                        .orderByAsc(WorkOrderYwConditionRecord::getCreateTime)
+                        .list());
+    }
+
+    @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,10 +213,14 @@
      */
     @Override
     public Result page(WorkOrderQuery query) {
-
         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())
+                .eq(StringUtils.hasText(query.getStatus().getValue()), WorkOrder::getStatus, query.getStatus())
+                .eq(StringUtils.hasText(query.getErrorType()), WorkOrder::getErrorType, query.getErrorType())
+                .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));
 
@@ -136,6 +243,152 @@
         return Result.ok().data(vos).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());
+    }
+
+    @Override
+    public Result distributeFast(DistributeWorkOrderVO data) {
+        // 鑾峰彇褰撳墠鏃堕棿
+        LocalDateTime now = LocalDateTime.now(ZoneId.systemDefault());
+        switch (data.getFastWay()) {
+            case LAST_HALF_HOUR:
+                data.setStart(now.minusMinutes(30));
+                data.setEnd(now);
+                break;
+            case LAST_HOUR:
+                data.setStart(now.minusHours(1));
+                data.setEnd(now);
+                break;
+            case LAST_TWO_HOUR:
+               data.setStart(now.minusHours(2));
+               data.setEnd(now);
+               break;
+            case LAST_DAY:
+                data.setStart(now.minusDays(1));
+                data.setEnd(now);
+                break;
+           }
+
+        // 鏌ヨ绗﹀悎鏉′欢鐨勫伐鍗�
+        List<Integer> ids = new LambdaQueryChainWrapper<>(baseMapper)
+                .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()
+                .stream()
+                .map(WorkOrder::getId)
+                .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);
+            return Result.ok("鎴愬姛涓嬪彂" + ids.size() + "鏉″伐鍗�");
+        } catch (Exception e) {
+            return Result.ok("鎿嶄綔澶辫触");
+        } finally {
+            distributeUnLock();
+        }
+    }
+
+    @Override
+    public Result selectedIdsDistribute(List<Integer> ids, Integer unitId) {
+        if (!getDistributeLock()) {
+            return Result.error("宸ュ崟涓嬪彂涓紝璇风◢鍚庨噸璇�");
+        }
+        try {
+            if (ids.isEmpty()) {
+                ids = new LambdaQueryChainWrapper<>(baseMapper)
+                        .eq(WorkOrder::getStatus, WorkOrderStatusEnum.WAIT_DISTRIBUTE)
+                        .eq(Objects.nonNull(unitId), WorkOrder::getUnitId, unitId)
+                        .select(WorkOrder::getId)
+                        .list()
+                        .stream()
+                        .map(WorkOrder::getId)
+                        .collect(Collectors.toList());
+            }
+            new LambdaUpdateChainWrapper<>(baseMapper)
+                    .set(WorkOrder::getStatus, WorkOrderStatusEnum.DISTRIBUTED)
+                    .in(WorkOrder::getId, ids)
+                    .eq(WorkOrder::getStatus, WorkOrderStatusEnum.WAIT_DISTRIBUTE)
+                    .eq(Objects.nonNull(unitId), WorkOrder::getUnitId, unitId)
+                    .update();
+            addDistributeRecord(ids);
+            return Result.ok("鎿嶄綔鎴愬姛");
+        } catch (Exception e) {
+            return Result.ok("鎿嶄綔澶辫触");
+        } 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) {
+        LoginUser loginUser = SecurityUtils.getLoginUser();
+        workOrderDistributeRecordMapper.insertBatch(
+                ids.stream()
+                        .map(id -> new WorkOrderDistributeRecord(id, loginUser.getUserId()))
+                        .toList()
+        );
+    }
+
     /**
      * 鏍规嵁id鏌ユ壘
      * @param id

--
Gitblit v1.8.0