From b5df1cee65f3e9dcf554efcd3f1642a8c79d1e76 Mon Sep 17 00:00:00 2001
From: fuliqi <fuliqi@qq.com>
Date: 星期一, 30 九月 2024 02:12:24 +0800
Subject: [PATCH] 点位监测更换对象+存入mongo+提取utils

---
 ycl-server/src/main/java/com/ycl/platform/service/impl/WorkOrderServiceImpl.java |  243 ++++++++++++++++++++++++++++++++++++++++++------
 1 files changed, 210 insertions(+), 33 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 a946f0c..c8b9498 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,5 +1,6 @@
 package com.ycl.platform.service.impl;
 
+import com.alibaba.excel.annotation.format.DateTimeFormat;
 import com.alibaba.fastjson2.JSON;
 import com.alibaba.fastjson2.JSONObject;
 import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
@@ -37,10 +38,8 @@
 import com.ycl.utils.redis.RedisCache;
 import com.ycl.utils.uuid.IdUtils;
 import constant.Constants;
-import enumeration.general.NotifyTypeEnum;
-import enumeration.general.UrgentLevelEnum;
-import enumeration.general.WorkOrderDistributeWayEnum;
-import enumeration.general.WorkOrderStatusEnum;
+import constant.RedisConstant;
+import enumeration.general.*;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.lang3.ObjectUtils;
@@ -50,12 +49,19 @@
 import org.springframework.beans.BeanUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Value;
+import org.springframework.data.redis.core.RedisTemplate;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 import org.springframework.util.Assert;
 import org.springframework.util.CollectionUtils;
 import org.springframework.util.StringUtils;
 
+import java.net.HttpURLConnection;
+import java.net.URL;
+import java.net.URLDecoder;
+import java.net.URLEncoder;
+import java.nio.charset.StandardCharsets;
+import java.text.SimpleDateFormat;
 import java.time.LocalDateTime;
 import java.time.ZoneId;
 import java.util.*;
@@ -73,7 +79,6 @@
 @RequiredArgsConstructor
 public class WorkOrderServiceImpl extends ServiceImpl<WorkOrderMapper, WorkOrder> implements WorkOrderService {
 
-    private final WorkOrderMapper workOrderMapper;
     private final YwPointService ywPointService;
     private final WorkOrderAuditingRecordMapper workOrderAuditingRecordMapper;
     private final WorkOrderAuditingRecordService workOrderAuditingRecordService;
@@ -82,10 +87,8 @@
     private final WorkOrderDistributeRecordMapper workOrderDistributeRecordMapper;
     private final WorkOrderErrorTypeServiceImpl workOrderErrorTypeService;
     private final SysConfigMapper configMapper;
-    private final ServerConfig serverConfig;
     private final ReportMapper reportMapper;
-    private final SysDictDataMapper dictDataMapper;
-
+    private final WorkOrderCheckImgMapper workOrderCheckImgMapper;
     @Value("${rtsp.server:http://127.0.0.1:7788}")
     private String rtspServer;
 
@@ -143,7 +146,7 @@
                     continue;
                 } else {
                     for (String errorType : errorTypes) {
-                        if (! errorNameList.contains(errorType)) {
+                        if (!errorNameList.contains(errorType)) {
                             // 閿欒绫诲瀷涓嶄竴鏍凤紝灏辨柊澧炰竴涓敊璇被鍨嬶紝骞朵笖閲嶇疆宸ュ崟鐘舵�佷负寰呭鐞�
                             WorkOrderErrorType workOrderErrorType = new WorkOrderErrorType();
                             workOrderErrorType.setWorkOrderNo(databaseWorkOrder.getWorkOrderNo());
@@ -201,13 +204,40 @@
         SysConfig normal = configMapper.checkConfigKeyUnique("normal.wordkorder.alarm.time");
         // 濡傛灉鍗冲皢鐢熸垚宸ュ崟锛屼絾鏄澶囧浗鏍囩爜鏌ヤ笉鍒扮偣浣嶏紝鍒欎笉娣诲姞锛�
         List<WorkOrder> notAddList = new ArrayList<>();
+        //鏌edis浠婃棩宸ュ崟鏁伴噺
+        int workOrderNum = 0;
+        //UUID浣滀负value锛屼繚璇佷笂閿佺殑绾跨▼鑷繁瑙i攣
+        String requestId = IdUtils.fastSimpleUUID();
+        try {
+            for (int i = 0; i < 3; i++) {
+                boolean result = redisCache.acquireLock(RedisConstant.WORKORDER_NUM_LOCK, requestId, 10000);
+                if (result) {
+                    //鏌ヤ粖鏃ュ伐鍗曢噺
+                    Object redisNum = redisCache.getCacheObject(RedisConstant.WORKORDER_NUM);
+                    workOrderNum = redisNum == null ? 0 : (Integer) redisNum;
+                    break;
+                } else {
+                    if (i == 2) {
+                        log.error("閿佽鍗犵敤");
+                        return Boolean.FALSE;
+                    }
+                    //绛夊緟涓�娈垫椂闂村悗缁х画
+                    Thread.sleep(5000);
+                }
+            }
+        } catch (InterruptedException e) {
+            log.error("鑾峰彇閿佸紓甯�");
+            return Boolean.FALSE;
+        }
         for (WorkOrder workOrder : waitAddList) {
             YwPoint point = pointMapping.get(workOrder.getSerialNumber());
             if (Objects.isNull(point)) {
                 notAddList.add(workOrder);
                 continue;
             }
-            workOrder.setWorkOrderNo(IdUtils.randomNO(now));
+            //鏁板瓧鍓嶉潰琛�0
+            workOrderNum++;
+            workOrder.setWorkOrderNo(IdUtils.workOrderNO(now, String.format("%05d", workOrderNum)));
             if (Objects.nonNull(point.getUnitId())) {
                 workOrder.setUnitId(Math.toIntExact(point.getUnitId()));
             }
@@ -230,9 +260,12 @@
         if (CollectionUtils.isEmpty(waitAddList)) {
             return Boolean.TRUE;
         }
+        //璁板綍宸ュ崟鏁�
+        redisCache.setCacheObject(RedisConstant.WORKORDER_NUM, workOrderNum);
+        redisCache.releaseLock(RedisConstant.WORKORDER_NUM_LOCK, requestId);
         // 淇濆瓨宸ュ崟鍜屾晠闅滅被鍨�
         baseMapper.addMany(waitAddList);
-        if (! CollectionUtils.isEmpty(waitAddErrorTypeList)) {
+        if (!CollectionUtils.isEmpty(waitAddErrorTypeList)) {
             workOrderErrorTypeService.getBaseMapper().addMany(waitAddErrorTypeList);
         }
         // 濡傛灉鏄洿鎺ヤ笅鍙戯紝娣诲姞涓嬪彂璁板綍
@@ -258,22 +291,38 @@
 
     /**
      * 娣诲姞
+     *
      * @param form
      * @return
      */
     @Override
     public Result add(WorkOrderForm form) {
+        //鏌edis浠婃棩宸ュ崟鏁伴噺
+        int workOrderNum = 0;
+        //UUID浣滀负value锛屼繚璇佷笂閿佺殑绾跨▼鑷繁瑙i攣
+        String requestId = IdUtils.fastSimpleUUID();
+        boolean result = redisCache.acquireLock(RedisConstant.WORKORDER_NUM_LOCK, requestId, 10000);
+        if (result) {
+            //鏌ヤ粖鏃ュ伐鍗曢噺
+            Object redisNum = redisCache.getCacheObject(RedisConstant.WORKORDER_NUM);
+            workOrderNum = redisNum == null ? 0 : (Integer) redisNum;
+            workOrderNum++;
+            redisCache.setCacheObject(RedisConstant.WORKORDER_NUM, workOrderNum);
+            redisCache.releaseLock(RedisConstant.WORKORDER_NUM_LOCK, requestId);
+        } else {
+            return Result.error("宸ュ崟姝e湪娣诲姞锛岃绋嶅悗鍐嶈瘯");
+        }
         WorkOrder entity = WorkOrderForm.getEntityByForm(form, null);
         entity.setCreateTime(DateUtils.getNowDate());
         entity.setStatus(WorkOrderStatusEnum.WAIT_DISTRIBUTE);
         Date now = new Date();
         entity.setCreateTime(now);
         entity.setUpdateTime(now);
-        entity.setWorkOrderNo(IdUtils.randomNO(now));
+        entity.setWorkOrderNo(IdUtils.workOrderNO(now, String.format("%05d", workOrderNum)));
         entity.setErrorType(String.join(",", form.getErrorType()));
         List<WorkOrderErrorType> workOrderErrorTypes = form.getErrorType().stream().map(errorType -> new WorkOrderErrorType(entity.getWorkOrderNo(), errorType)).toList();
         workOrderErrorTypeService.getBaseMapper().insertWorkOrderErrorTypeList(workOrderErrorTypes);
-        if(baseMapper.insert(entity) > 0) {
+        if (baseMapper.insert(entity) > 0) {
             return Result.ok("娣诲姞鎴愬姛");
         }
         return Result.error("娣诲姞澶辫触");
@@ -281,6 +330,7 @@
 
     /**
      * 淇敼
+     *
      * @param form
      * @return
      */
@@ -402,7 +452,7 @@
         List<WorkOrderYwConditionRecordVO> ywConditionList = workOrderYwConditionRecordMapper.selectYwConditionByYwId(workOrderNo);
         ywConditionList.stream().forEach(item -> {
             if (Objects.nonNull(item.getSysMsg()) && item.getSysMsg()) {
-                item.setCommitUserName("绯荤粺娑堟伅");
+                item.setUnitName("绯荤粺娑堟伅");
             }
         });
         return ywConditionList;
@@ -411,9 +461,9 @@
     @Override
     public List<WorkOrderAuditingRecord> selectYwAuditingListByYwId(String workOrderNo) {
         return new LambdaQueryChainWrapper<>(workOrderAuditingRecordMapper)
-                        .eq(WorkOrderAuditingRecord::getWorkOrderNo, workOrderNo)
-                        .orderByAsc(WorkOrderAuditingRecord::getCreateTime)
-                        .list();
+                .eq(WorkOrderAuditingRecord::getWorkOrderNo, workOrderNo)
+                .orderByAsc(WorkOrderAuditingRecord::getCreateTime)
+                .list();
     }
 
     @Override
@@ -428,12 +478,13 @@
 
     /**
      * 鎵归噺鍒犻櫎
+     *
      * @param ids
      * @return
      */
     @Override
     public Result remove(List<String> ids) {
-        if(baseMapper.deleteBatchIds(ids) > 0) {
+        if (baseMapper.deleteBatchIds(ids) > 0) {
             return Result.ok("鍒犻櫎鎴愬姛");
         }
         return Result.error("鍒犻櫎澶辫触");
@@ -441,12 +492,13 @@
 
     /**
      * id鍒犻櫎
+     *
      * @param id
      * @return
      */
     @Override
     public Result removeById(String id) {
-        if(baseMapper.deleteById(id) > 0) {
+        if (baseMapper.deleteById(id) > 0) {
             return Result.ok("鍒犻櫎鎴愬姛");
         }
         return Result.error("鍒犻櫎澶辫触");
@@ -454,6 +506,7 @@
 
     /**
      * 鍒嗛〉鏌ヨ
+     *
      * @param query
      * @return
      */
@@ -461,8 +514,10 @@
     public Result page(WorkOrderQuery query) {
         IPage<WorkOrderVO> page = PageUtil.getPage(query, WorkOrderVO.class);
         query.setUnitId(SecurityUtils.getUnitId());
+        query.setStart(DateUtils.getDayStart(query.getStart()));
+        query.setEnd(DateUtils.getDayEnd(query.getEnd()));
         baseMapper.page(page, query);
-        if (! CollectionUtils.isEmpty(page.getRecords())) {
+        if (!CollectionUtils.isEmpty(page.getRecords())) {
             page.getRecords().stream().forEach(item -> {
                 if (StringUtils.hasText(item.getErrorType())) {
                     item.setErrorTypeList(List.of(item.getErrorType().split(",")));
@@ -493,14 +548,14 @@
                 data.setStart(now.minusHours(1));
                 break;
             case LAST_TWO_HOUR:
-               data.setStart(now.minusHours(2));
-               break;
+                data.setStart(now.minusHours(2));
+                break;
             case LAST_DAY:
                 data.setStart(now.minusDays(1));
                 break;
             default:
                 break;
-           }
+        }
         if (Objects.isNull(data.getStart())) {
             throw new RuntimeException("鏃犳硶鐢熸垚蹇�熶笅鍙戠殑鏃堕棿鑼冨洿锛岃閫夋嫨姝g‘鐨勫揩閫熶笅鍙戞柟寮�");
         }
@@ -509,7 +564,7 @@
                 .select(WorkOrder::getSerialNumber, WorkOrder::getWorkOrderNo)
                 .eq(WorkOrder::getStatus, WorkOrderStatusEnum.WAIT_DISTRIBUTE)
                 .eq(Objects.nonNull(data.getUnitId()), WorkOrder::getUnitId, data.getUnitId())
-                .eq(WorkOrder::getErrorType, data.getErrorType())
+                .in(WorkOrder::getErrorType, data.getErrorType())
                 .between(WorkOrder::getCreateTime, data.getStart(), data.getEnd())
                 .orderByDesc(WorkOrder::getCreateTime)
                 .last("limit " + data.getFastNumLimit())
@@ -517,8 +572,12 @@
         List<String> workOrderNoList = list.stream().map(WorkOrder::getWorkOrderNo).toList();
         List<String> serialNumberList = list.stream().map(WorkOrder::getSerialNumber).toList();
 
-        if (workOrderNoList.isEmpty()) { return Result.error("娌℃湁绗﹀悎鏉′欢鐨勫伐鍗�"); }
-        if (!getDistributeLock()) { return Result.error("姝ゅ埢鏈変汉涓嬪彂涓紝涓洪伩鍏嶅啿绐侊紝璇风◢鍚庨噸璇�"); }
+        if (workOrderNoList.isEmpty()) {
+            return Result.error("娌℃湁绗﹀悎鏉′欢鐨勫伐鍗�");
+        }
+        if (!getDistributeLock()) {
+            return Result.error("姝ゅ埢鏈変汉涓嬪彂涓紝涓洪伩鍏嶅啿绐侊紝璇风◢鍚庨噸璇�");
+        }
         try {
             new LambdaUpdateChainWrapper<>(baseMapper)
                     .set(WorkOrder::getStatus, WorkOrderStatusEnum.DISTRIBUTED)
@@ -539,7 +598,9 @@
     @Transactional
     public Result selectedIdsDistribute(DistributeWorkOrderQuery query) {
         WorkOrderDistributeWayEnum distributeWayEnum = WorkOrderDistributeWayEnum.SELECTED_DISTRIBUTE;
-        if (!getDistributeLock()) { return Result.error("姝ゅ埢鏈変汉涓嬪彂涓紝涓洪伩鍏嶅啿绐侊紝璇风◢鍚庨噸璇�"); }
+        if (!getDistributeLock()) {
+            return Result.error("姝ゅ埢鏈変汉涓嬪彂涓紝涓洪伩鍏嶅啿绐侊紝璇风◢鍚庨噸璇�");
+        }
         try {
             if (query.getWorkOrderNOList().isEmpty()) {
                 query.setWorkOrderNOList(new LambdaQueryChainWrapper<>(baseMapper)
@@ -552,7 +613,9 @@
                         .collect(Collectors.toList()));
                 distributeWayEnum = WorkOrderDistributeWayEnum.ALL_DISTRIBUTE;
             }
-            if (query.getWorkOrderNOList().isEmpty()) { return Result.error("娌℃湁宸ュ崟寰呬笅鍙�"); }
+            if (query.getWorkOrderNOList().isEmpty()) {
+                return Result.error("娌℃湁宸ュ崟寰呬笅鍙�");
+            }
             new LambdaUpdateChainWrapper<>(baseMapper)
                     .set(WorkOrder::getStatus, WorkOrderStatusEnum.DISTRIBUTED)
                     .in(WorkOrder::getWorkOrderNo, query.getWorkOrderNOList())
@@ -563,6 +626,7 @@
             ywPointService.updateRecovery(serialNumberList, 1);
             return Result.ok("鎴愬姛涓嬪彂" + query.getWorkOrderNOList().size() + "鏉″伐鍗�");
         } catch (Exception e) {
+            System.out.println(e.getMessage());
             return Result.error("鎿嶄綔澶辫触");
         } finally {
             distributeUnLock();
@@ -571,6 +635,7 @@
 
     /**
      * 鐢宠宸ュ崟涓嬪彂閿�
+     *
      * @return 宸ュ崟涓嬪彂閿佺敵璇风粨鏋�
      */
     public synchronized Boolean getDistributeLock() {
@@ -591,6 +656,7 @@
 
     /**
      * 娣诲姞宸ュ崟涓嬪彂璁板綍
+     *
      * @param workOrderNoList 宸ュ崟id
      */
     private void addDistributeRecord(List<String> workOrderNoList, WorkOrderDistributeWayEnum distributeWay) {
@@ -604,6 +670,7 @@
 
     /**
      * 鏍规嵁id鏌ユ壘
+     *
      * @param id
      * @return
      */
@@ -618,6 +685,7 @@
 
     /**
      * 鍒楄〃
+     *
      * @return
      */
     @Override
@@ -667,12 +735,85 @@
 
     @Override
     public List<WorkOrderRegionVO> workOrderRegion(DashboardQuery dashboardQuery) {
-        return baseMapper.workOrderRegion(dashboardQuery);
+        //鍒濆鍖栨墍鏈夊尯鍩熸暟鎹�
+        List<WorkOrderRegionVO> vos = new ArrayList<>();
+        for (AreaDeptEnum value : AreaDeptEnum.values()) {
+            WorkOrderRegionVO vo = new WorkOrderRegionVO();
+            vo.setArea(value.getName());
+            vo.setDoneNum(0);
+            vo.setTodoNum(0);
+            vos.add(vo);
+        }
+        List<WorkOrderRegionVO> workOrderRegionVOS = baseMapper.workOrderRegion(dashboardQuery);
+        for (WorkOrderRegionVO vo : vos) {
+            for (WorkOrderRegionVO workOrder : workOrderRegionVOS) {
+                //璧嬪��
+                if (vo.getArea().equals(workOrder.getArea())) {
+                    BeanUtils.copyProperties(workOrder, vo);
+                }
+            }
+        }
+        return vos;
     }
 
 
     @Override
-    public String getFrameImgByDevice(String deviceId, String channelId, String workOrderNo) {
+    public synchronized String getFrameImgByDevice(String deviceId, String channelId, String workOrderNo) {
+        String url = String.format(this.rtspServer + "/api/play/start/%s/%s", deviceId, channelId);
+        String result = HttpUtils.sendGet(url);
+        WVPResult wvpResult = JSON.parseObject(result, WVPResult.class);
+        String imgUrl = null;
+        if (wvpResult.getCode() == 0) {
+            JSONObject data = (JSONObject) wvpResult.getData();
+            String rtspUrl = data.getString("fmp4"); // 鍙杕p4鍦板潃
+            if (StringUtils.hasText(rtspUrl)) {
+                System.out.println("鐩爣鍦板潃锛�" + rtspUrl);
+                FFmpegFrameGrabber grabber = null;
+                try {
+                    grabber = new FFmpegFrameGrabber(rtspUrl);
+//                    grabber.setOption("rtsp_transport", "tcp"); // 浣跨敤tcp鐨勬柟寮忥紝涓嶇劧浼氫涪鍖呭緢涓ラ噸
+//                    grabber.setVideoOption("probesize", "10000"); // 璁剧疆鎹曡幏鍒嗘瀽鐨勬渶澶у瓧鑺�
+                    //璁剧疆10s瓒呮椂
+                    grabber.setTimeout(10000);
+                    grabber.start();
+                    Frame frame = grabber.grabImage(); // 鐩存帴鎹曡幏涓�甯�
+                    if (frame != null) {
+                        System.out.println("鎴愬姛鎹曡幏涓�甯�");
+                        // 灏咶rame杞崲涓篗at
+                        OpenCVFrameConverter.ToMat converter = new OpenCVFrameConverter.ToMat();
+                        Mat mat = converter.convertToMat(frame);
+
+                        imgUrl = workOrderNo + "_" + IdUtils.fastSimpleUUID() + ".png";
+                        // 鐢熸垚鍥剧墖璺緞
+                        String imgPath = PlatformConfig.getProfile() + "/" + imgUrl;
+                        System.out.println("鍥剧墖淇濆瓨鍦板潃锛�" + imgPath);
+                        imgUrl = Constants.RESOURCE_PREFIX + "/" + imgUrl;
+                        // 淇濆瓨鍥剧墖
+                        opencv_imgcodecs.imwrite(imgPath, mat);
+                    } else {
+                        System.out.println("鏈崟鑾峰埌甯�");
+                    }
+                } catch (FrameGrabber.Exception e) {
+                    e.printStackTrace();
+                } finally {
+                    if (grabber != null) {
+                        try {
+                            grabber.stop(); // 鍋滄鎹曡幏
+                        } catch (FrameGrabber.Exception e) {
+                            e.printStackTrace();
+                        }
+                    }
+                }
+            }
+        } else {
+            System.out.println("璇锋眰澶辫触锛岄敊璇爜锛�" + wvpResult.getCode() + "--" + wvpResult.getMsg());
+        }
+        System.out.println("鍥剧墖URL锛�" + imgUrl);
+        return imgUrl;
+    }
+
+    @Override
+    public void saveFrameImgByDevice(String deviceId, String channelId, String workOrderNo) {
         String url = String.format(this.rtspServer + "/api/play/start/%s/%s", deviceId, channelId);
         String result = HttpUtils.sendGet(url);
         WVPResult wvpResult = JSON.parseObject(result, WVPResult.class);
@@ -685,6 +826,8 @@
                 FFmpegFrameGrabber grabber = null;
                 try {
                     grabber = FFmpegFrameGrabber.createDefault(rtspUrl);
+                    //璁剧疆10s瓒呮椂
+                    grabber.setTimeout(10000);
                     grabber.start();
                     Frame frame = grabber.grabImage(); // 鐩存帴鎹曡幏涓�甯�
                     if (frame != null) {
@@ -721,7 +864,13 @@
             System.out.println("璇锋眰澶辫触锛岄敊璇爜锛�" + wvpResult.getCode());
         }
         System.out.println("鍥剧墖URL锛�" + imgUrl);
-        return imgUrl;
+        if (StringUtils.hasText(imgUrl)) {
+            WorkOrderCheckImg img = new WorkOrderCheckImg();
+            img.setWorkOrderNo(workOrderNo);
+            img.setImgUrl(imgUrl);
+            img.setCreateTime(new Date());
+            workOrderCheckImgMapper.insert(img);
+        }
     }
 
     @Override
@@ -761,7 +910,7 @@
         if (Objects.nonNull(beforeRecord)) {
             if (now.before(beforeRecord.getBeginCreateTime())) {
                 process.setBeforeReportMsg("浜嬪墠鎶ュ宸插け鏁�");
-            } else if (now.after(beforeRecord.getEndCreateTime())){
+            } else if (now.after(beforeRecord.getEndCreateTime())) {
                 process.setBeforeReportMsg("浜嬪墠鎶ュ鏈敓鏁�");
             } else {
                 process.setBeforeReportMsg("宸蹭簨鍓嶆姤澶�");
@@ -770,7 +919,7 @@
         if (Objects.nonNull(afterRecord)) {
             if (now.before(afterRecord.getBeginCreateTime())) {
                 process.setAfterReportMsg("浜嬪悗鎶ュ宸插け鏁�");
-            } else if (now.after(afterRecord.getEndCreateTime())){
+            } else if (now.after(afterRecord.getEndCreateTime())) {
                 process.setAfterReportMsg("浜嬪悗鎶ュ鏈敓鏁�");
             } else {
                 process.setAfterReportMsg("宸蹭簨鍚庢姤澶�");
@@ -791,6 +940,34 @@
         List<SysDictData> errorList = workOrderErrorTypeService.getBaseMapper().getErrorList(workOrder.getWorkOrderNo());
         List<String> errList = errorList.stream().map(SysDictData::getDictLabel).collect(Collectors.toList());
         workOrder.setErrorTypeList(errList);
+        // 妫�娴嬪浘鐗�
+        List<WorkOrderCheckImg> imgList = new LambdaQueryChainWrapper<>(workOrderCheckImgMapper)
+                .eq(WorkOrderCheckImg::getWorkOrderNo, workOrderNo)
+                .orderByDesc(WorkOrderCheckImg::getCreateTime)
+                .last("limit 20")
+                .list();
+        workOrder.setImgList(imgList);
         return Result.ok().data(workOrder);
     }
+
+    @Override
+    public List<WorkOrderVO> export(WorkOrderExportQuery query) {
+        query.setUnitId(SecurityUtils.getUnitId());
+        SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
+        if (query.getStart() == null) {
+            query.setStart(format.format(DateUtils.getDayStart(new Date())));
+        } else {
+            query.setStart(query.getStart() + " 00:00:00");
+        }
+        if (query.getEnd() == null) {
+            query.setStart(format.format(DateUtils.getDayEnd(new Date())));
+        } else {
+            query.setEnd(query.getEnd() + " 23:59:59");
+        }
+
+        List<WorkOrderVO> export = baseMapper.export(query);
+
+        System.out.println(export);
+        return export;
+    }
 }

--
Gitblit v1.8.0