From b91945ea32378a71d1e57356af6e16703a1ceaa7 Mon Sep 17 00:00:00 2001
From: fuliqi <fuliqi@qq.com>
Date: 星期五, 01 十一月 2024 14:52:16 +0800
Subject: [PATCH] 工单只下发考核的、阈值bug

---
 ycl-server/src/main/java/com/ycl/platform/service/impl/WorkOrderServiceImpl.java |  327 +++++++++++++++++++++++++++++++++++++++---------------
 1 files changed, 234 insertions(+), 93 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 7fd7d38..5859cbc 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,7 +1,10 @@
 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.query.LambdaQueryWrapper;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
@@ -11,6 +14,7 @@
 import com.ycl.config.ServerConfig;
 import com.ycl.exception.ServiceException;
 import com.ycl.platform.domain.entity.*;
+import com.ycl.platform.domain.excel.PointExport;
 import com.ycl.platform.domain.form.*;
 import com.ycl.platform.domain.query.*;
 import com.ycl.platform.domain.vo.*;
@@ -36,6 +40,8 @@
 import com.ycl.utils.http.HttpUtils;
 import com.ycl.utils.redis.RedisCache;
 import com.ycl.utils.uuid.IdUtils;
+import constant.ApiConstants;
+import constant.CheckConstants;
 import constant.Constants;
 import constant.RedisConstant;
 import enumeration.general.*;
@@ -55,6 +61,17 @@
 import org.springframework.util.CollectionUtils;
 import org.springframework.util.StringUtils;
 
+import javax.imageio.ImageIO;
+import javax.swing.*;
+import java.awt.image.BufferedImage;
+import java.io.File;
+import java.io.IOException;
+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.*;
@@ -82,6 +99,7 @@
     private final SysConfigMapper configMapper;
     private final ReportMapper reportMapper;
     private final WorkOrderCheckImgMapper workOrderCheckImgMapper;
+    private final WorkOrderWhiteMapper workOrderWhiteMapper;
     @Value("${rtsp.server:http://127.0.0.1:7788}")
     private String rtspServer;
 
@@ -93,12 +111,16 @@
 
     @Override
     @Transactional(rollbackFor = Exception.class)
-    //TODO锛歊edis鍔犻攣锛屼粠Redis璇诲彇褰撴棩宸ュ崟鏁版嵁閲�
     public synchronized Boolean innerAddWorkOrder(List<WorkOrder> workOrderList) {
         int total = workOrderList.size();
-        workOrderList.stream().filter(item -> {
+        // 鏌ヨ鍑虹櫧鍚嶅崟鍒楄〃
+        List<String> serialNumbers = workOrderWhiteMapper.selectList().stream().map(WorkOrderWhite::getSerialNumber).collect(Collectors.toList());
+        // 閬嶅巻宸ュ崟鍒楄〃锛屽垽鏂槸鍚﹀湪鐧藉悕鍗曚腑
+        workOrderList = workOrderList.stream().filter(item -> !serialNumbers.contains(item.getSerialNumber()))
+                        .collect(Collectors.toList());
+        workOrderList = workOrderList.stream().filter(item -> {
             return StringUtils.hasText(item.getSerialNumber()) && Objects.nonNull(item.getStatus()) && !CollectionUtils.isEmpty(item.getErrorTypeList());
-        });
+        }).collect(Collectors.toList());
         if (CollectionUtils.isEmpty(workOrderList)) {
             return Boolean.TRUE;
         }
@@ -150,7 +172,10 @@
                             willAddErrorType.add(workOrderErrorType);
                         }
                     }
-                    databaseWorkOrder.setStatus(WorkOrderStatusEnum.DISTRIBUTED);
+                    //閬垮厤浠d笅鍙戠殑宸ュ崟鐩存帴鍙樻垚涓嬪彂
+                    if(!databaseWorkOrder.getStatus().equals(WorkOrderStatusEnum.WAIT_DISTRIBUTE)) {
+                        databaseWorkOrder.setStatus(WorkOrderStatusEnum.DISTRIBUTED);
+                    }
                     databaseWorkOrder.setUpdateTime(now);
                     willUpdateStatusWorkOrderList.add(databaseWorkOrder);
                     updateNum++;
@@ -188,9 +213,11 @@
             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)
+                .eq(YwPoint::getExamineStatus, Boolean.TRUE)
                 .list();
         Map<String, YwPoint> pointMapping = pointList.stream().collect(Collectors.toMap(YwPoint::getSerialNumber, point -> point));
         // 鏌ュ嚭閲嶇偣鐐逛綅銆佹櫘閫氱偣浣嶇殑澶勭悊鏃堕棿
@@ -200,6 +227,25 @@
         List<WorkOrder> notAddList = new ArrayList<>();
         //鏌edis浠婃棩宸ュ崟鏁伴噺
         int workOrderNum = 0;
+        for (WorkOrder workOrder : waitAddList) {
+            YwPoint point = pointMapping.get(workOrder.getSerialNumber());
+            if (Objects.isNull(point)) {
+                notAddList.add(workOrder);
+                continue;
+            }
+            if (Objects.nonNull(point.getUnitId())) {
+                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;
+        }
         //UUID浣滀负value锛屼繚璇佷笂閿佺殑绾跨▼鑷繁瑙i攣
         String requestId = IdUtils.fastSimpleUUID();
         try {
@@ -223,23 +269,11 @@
             log.error("鑾峰彇閿佸紓甯�");
             return Boolean.FALSE;
         }
+        //璁板綍宸ュ崟鏁�,琛ュ厖宸ュ崟鍙�
         for (WorkOrder workOrder : waitAddList) {
-            YwPoint point = pointMapping.get(workOrder.getSerialNumber());
-            if (Objects.isNull(point)) {
-                notAddList.add(workOrder);
-                continue;
-            }
             //鏁板瓧鍓嶉潰琛�0
             workOrderNum++;
-            workOrder.setWorkOrderNo(IdUtils.randomNO(now, String.format("%05d", workOrderNum)));
-            if (Objects.nonNull(point.getUnitId())) {
-                workOrder.setUnitId(Math.toIntExact(point.getUnitId()));
-            }
-            if (point.getImportantTag() || point.getImportantCommandImageTag()) {
-                workOrder.setProcessingPeriod(Integer.valueOf(important.getConfigValue()));
-            } else {
-                workOrder.setProcessingPeriod(Integer.valueOf(normal.getConfigValue()));
-            }
+            workOrder.setWorkOrderNo(IdUtils.workOrderNO(now, String.format("%05d", workOrderNum)));
             // 淇濆瓨閿欒绫诲瀷
             for (String errorType : workOrder.getErrorTypeList()) {
                 WorkOrderErrorType workOrderErrorType = new WorkOrderErrorType();
@@ -249,12 +283,8 @@
                 workOrderErrorType.setErrorName(errorType);
                 waitAddErrorTypeList.add(workOrderErrorType);
             }
+            log.info("鍗冲皢鐢熸垚鐨勫伐鍗�:{}",workOrder);
         }
-        waitAddList.removeAll(notAddList);
-        if (CollectionUtils.isEmpty(waitAddList)) {
-            return Boolean.TRUE;
-        }
-        //璁板綍宸ュ崟鏁�
         redisCache.setCacheObject(RedisConstant.WORKORDER_NUM, workOrderNum);
         redisCache.releaseLock(RedisConstant.WORKORDER_NUM_LOCK, requestId);
         // 淇濆瓨宸ュ崟鍜屾晠闅滅被鍨�
@@ -292,14 +322,14 @@
     @Override
     public Result add(WorkOrderForm form) {
         //鏌edis浠婃棩宸ュ崟鏁伴噺
-        int workOrderNum = 1;
+        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 ? 1 : (Integer) redisNum;
+            workOrderNum = redisNum == null ? 0 : (Integer) redisNum;
             workOrderNum++;
             redisCache.setCacheObject(RedisConstant.WORKORDER_NUM, workOrderNum);
             redisCache.releaseLock(RedisConstant.WORKORDER_NUM_LOCK, requestId);
@@ -307,19 +337,26 @@
             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, 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) {
-            return Result.ok("娣诲姞鎴愬姛");
+        // 鏌ヨ鍑虹櫧鍚嶅崟鍒楄〃
+        List<String> serialNumbers = workOrderWhiteMapper.selectList().stream().map(WorkOrderWhite::getSerialNumber).collect(Collectors.toList());
+        // 閬嶅巻宸ュ崟鍒楄〃锛屽垽鏂槸鍚﹀湪鐧藉悕鍗曚腑
+         if (serialNumbers.contains(entity.getSerialNumber())){
+            return Result.error("璇ヨ澶囧湪鐧藉悕鍗曚腑锛屼笉鑳芥坊鍔犲伐鍗�");
+        }else {
+            entity.setCreateTime(DateUtils.getNowDate());
+            entity.setStatus(WorkOrderStatusEnum.WAIT_DISTRIBUTE);
+            Date now = new Date();
+            entity.setCreateTime(now);
+            entity.setUpdateTime(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) {
+                return Result.ok("娣诲姞鎴愬姛");
+            }
+            return Result.error("娣诲姞澶辫触");
         }
-        return Result.error("娣诲姞澶辫触");
     }
 
     /**
@@ -558,7 +595,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())
@@ -620,6 +657,7 @@
             ywPointService.updateRecovery(serialNumberList, 1);
             return Result.ok("鎴愬姛涓嬪彂" + query.getWorkOrderNOList().size() + "鏉″伐鍗�");
         } catch (Exception e) {
+            System.out.println(e.getMessage());
             return Result.error("鎿嶄綔澶辫触");
         } finally {
             distributeUnLock();
@@ -751,69 +789,76 @@
 
 
     @Override
-    public String getFrameImgByDevice(String deviceId, String channelId, String workOrderNo) {
-        String url = String.format(this.rtspServer + "/api/play/start/%s/%s", deviceId, channelId);
+    public synchronized String getFrameImgByDevice(String deviceId, String channelId, String workOrderNo) throws Exception {
+        String url = String.format(this.rtspServer + "/api/play/start/img/%s/%s", deviceId, channelId);
         String result = HttpUtils.sendGet(url);
+        log.info("鎷垮埌鍙栨祦鍥剧墖鍝嶅簲缁撴灉锛�" + result);
         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 = FFmpegFrameGrabber.createDefault(rtspUrl);
-                    //璁剧疆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();
-                        }
-                        // 閫氬父涓嶉渶瑕佽皟鐢╮elease()锛屽洜涓簊top()浼氬鐞嗚祫婧愰噴鏀�
-                        // grabber.release(); // 閲婃斁璧勬簮
-                    }
-                }
-            }
-        } else {
-            System.out.println("璇锋眰澶辫触锛岄敊璇爜锛�" + wvpResult.getCode());
+        if(wvpResult.getCode() == 0) {
+            Object imgUrl = (Object) wvpResult.getData();
+            imgUrl = Objects.isNull(imgUrl) ? "" : imgUrl;
+            log.info("鎷垮埌鍙栨祦鍥剧墖锛�" + imgUrl);
+            return (String) imgUrl;
         }
-        System.out.println("鍥剧墖URL锛�" + imgUrl);
-        return imgUrl;
+        return null;
+//        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"); // 璁剧疆鎹曡幏鍒嗘瀽鐨勬渶澶у瓧鑺�
+//                    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) {
+    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);
         String imgUrl = null;
         if (wvpResult.getCode() == 0) {
             JSONObject data = (JSONObject) wvpResult.getData();
-            String rtspUrl = data.getString("fmp4"); // 鍙杕p4鍦板潃
+            String rtspUrl = data.getString("rtsp"); // 鍙杕p4鍦板潃
             if (StringUtils.hasText(rtspUrl)) {
                 System.out.println("鐩爣鍦板潃锛�" + rtspUrl);
                 FFmpegFrameGrabber grabber = null;
@@ -867,13 +912,13 @@
     }
 
     @Override
-    public List<DeviceInfoVO> hasErrorWorkOrderList(Date start, Date end) {
+    public List<DeviceInfoVO> hasErrorWorkOrderList (Date start, Date end){
         List<DeviceInfoVO> list = baseMapper.hasErrorWorkOrderList(start, end);
         return list;
     }
 
     @Override
-    public void updateImgById(Integer workOrderId, String imgPath) {
+    public void updateImgById (Integer workOrderId, String imgPath){
         new LambdaUpdateChainWrapper<>(baseMapper)
                 .eq(WorkOrder::getId, workOrderId)
                 .set(WorkOrder::getYwCheckResult, imgPath)
@@ -881,7 +926,7 @@
     }
 
     @Override
-    public Result processImg(String workOrderNo) {
+    public Result processImg (String workOrderNo){
         WorkOrder workOrder = new LambdaQueryChainWrapper<>(baseMapper)
                 .eq(WorkOrder::getWorkOrderNo, workOrderNo)
                 .one();
@@ -922,7 +967,7 @@
     }
 
     @Override
-    public Result detailByNo(String workOrderNo) {
+    public Result detailByNo (String workOrderNo){
         WorkOrderDetailVO workOrder = baseMapper.detailByNo(workOrderNo);
         // 鏄惁鎶ュ
         boolean hasReport = new LambdaQueryChainWrapper<>(reportMapper)
@@ -942,4 +987,100 @@
         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;
+    }
+
+    /**
+     * 宸ュ崟鐧藉悕鍗曞垪琛�
+     *
+     * @param query 鏌ヨ
+     * @return {@link List }<{@link WorkOrderWhite }>
+     * @author
+     */
+    @Override
+    public Result selectWorkOrderWhiteList(WorkOrderWhiteQuery query) {
+        IPage<WorkOrderWhite> page = PageUtil.getPage(query, WorkOrderWhite.class);
+        workOrderWhiteMapper.page(page, query);
+        return Result.ok().data(page.getRecords()).total(page.getTotal());
+    }
+
+    /**
+     * 娣诲姞宸ュ崟鐧藉悕鍗�
+     *
+     * @param workOrderWhite 鐧借壊宸ュ崟
+     * @return {@link Result }
+     * @author
+     */
+    @Override
+    public Result addWorkOrderWhite(WorkOrderWhite workOrderWhite) {
+        // 閫氳繃璁惧缂栫爜鏌ヨ璁惧淇℃伅
+        YwPoint ywPoint = ywPointService.selectBySerialNumber(workOrderWhite.getSerialNumber());
+        // 鏂板鐧藉悕鍗曡澶囪褰�
+        if (ywPoint == null){
+            return Result.error("璁惧涓嶅瓨鍦�");
+        }
+        // 妫�鏌ユ槸鍚﹀凡缁忓瓨鍦ㄨ鐧藉悕鍗�
+        WorkOrderWhite flag = workOrderWhiteMapper.selectBySerialNumber(workOrderWhite.getSerialNumber());
+        if (flag != null) {
+            return Result.error("璇ヨ澶囧凡瀛樺湪鐧藉悕鍗�");
+        } else {
+            workOrderWhite.setPointName(ywPoint.getPointName());
+            workOrderWhite.setCreateBy(SecurityUtils.getUsername());
+            workOrderWhiteMapper.insert(workOrderWhite);
+            return Result.ok();
+        }
+    }
+
+    /**
+     * 鎵归噺鍒犻櫎宸ュ崟鐧藉悕鍗�
+     *
+     * @param ids ids
+     * @author
+     */
+    @Override
+    public Result batchDeleteWorkOrderWhite(List<String> ids) {
+        workOrderWhiteMapper.batchDelete(ids);
+        return Result.ok();
+    }
+
+    /**
+     * 鎵归噺瀵煎叆鐧藉悕鍗�
+     *
+     * @param dataList 鏁版嵁鍒楄〃
+     * @author
+     */
+    @Override
+    public void batchImportWhite(List<PointExport> dataList) {
+        // 鑾峰緱鎵�鏈夐渶瑕佹柊澧炵殑鐧藉悕鍗曞璞�
+        List<WorkOrderWhite> whitelist = dataList.stream().map(pointExport ->
+                        new WorkOrderWhite(pointExport.getSerialNumber(), pointExport.getPointName(), SecurityUtils.getUsername()))
+                .collect(Collectors.toList());
+        // 鑾峰緱鎵�鏈夊凡瀛樺湪鐨勭櫧鍚嶅崟璁惧缂栫爜
+        List<String> serialNumbers = workOrderWhiteMapper.selectList().stream().map(WorkOrderWhite::getSerialNumber).collect(Collectors.toList());
+        // 绛涢�夊嚭鏂板銆佷慨鏀圭殑鐧藉悕鍗曞璞�
+        List<WorkOrderWhite> updateWhiteList = whitelist.stream().filter(white -> serialNumbers.contains(white.getSerialNumber())).collect(Collectors.toList());
+        List<WorkOrderWhite> insertWhiteList = whitelist.stream().filter(white ->!serialNumbers.contains(white.getSerialNumber())).collect(Collectors.toList());
+        // 鏂板/淇敼鐧藉悕鍗曡澶囪褰�
+        updateWhiteList.stream().forEach(white -> workOrderWhiteMapper.updateBySerialNumber(white));
+        insertWhiteList.stream().forEach(white -> workOrderWhiteMapper.insert(white));
+    }
 }

--
Gitblit v1.8.0