From 3e9a6da99aae968123ee7bca352fc08becd7f1f2 Mon Sep 17 00:00:00 2001
From: xiangpei <xiangpei@timesnew.cn>
Date: 星期四, 29 八月 2024 17:29:21 +0800
Subject: [PATCH] 工单过程图接口更换

---
 ycl-server/src/main/java/com/ycl/platform/service/impl/WorkOrderServiceImpl.java |  145 ++++++++++++++++++++++++++++++++----------------
 1 files changed, 96 insertions(+), 49 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 908915b..a8f2c12 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,6 +1,7 @@
 package com.ycl.platform.service.impl;
 
 import com.alibaba.fastjson2.JSON;
+import com.alibaba.fastjson2.JSONObject;
 import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
@@ -12,10 +13,7 @@
 import com.ycl.platform.domain.entity.*;
 import com.ycl.platform.domain.form.*;
 import com.ycl.platform.domain.query.*;
-import com.ycl.platform.domain.vo.DeviceInfoVO;
-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.*;
 import com.ycl.platform.domain.vo.screen.ScreenWorkOrderVO;
 import com.ycl.platform.domain.vo.screen.WorkOrderRegionVO;
 import com.ycl.platform.domain.vo.screen.WorkOrderTotalVO;
@@ -37,6 +35,7 @@
 import com.ycl.utils.http.HttpUtils;
 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;
@@ -44,9 +43,9 @@
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.lang3.ObjectUtils;
-import org.bytedeco.javacv.FFmpegFrameGrabber;
-import org.bytedeco.javacv.Frame;
-import org.bytedeco.javacv.Java2DFrameConverter;
+import org.bytedeco.javacv.*;
+import org.bytedeco.opencv.global.opencv_imgcodecs;
+import org.bytedeco.opencv.opencv_core.Mat;
 import org.springframework.beans.BeanUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Value;
@@ -57,6 +56,7 @@
 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;
@@ -121,10 +121,16 @@
         List<WorkOrderErrorType> waitAddErrorTypeList = new ArrayList<>(48);
         Integer updateNum = 0;
         Date now = new Date();
+        // 鍥犳晠闅滅被鍨嬩笉涓�鑷磋�岃鏇存柊鐘舵�佺殑宸ュ崟
+        List<WorkOrder> willUpdateStatusWorkOrderList = new ArrayList<>(48);
+        // 鏇存敼宸ュ崟绫诲瀷鑰岃澧炲姞鐨勭郴缁熻繍缁村鐞嗕俊鎭�
+        List<WorkOrderYwConditionRecord> willAddMsg = new ArrayList<>(48);
+        // 鍗冲皢瑕佹坊鍔犵殑閿欒绫诲瀷
+        List<WorkOrderErrorType> willAddErrorType = new ArrayList<>(96);
         for (WorkOrder workOrder : workOrderList) {
             WorkOrder databaseWorkOrder = mapping.get(workOrder.getSerialNumber());
             if (Objects.nonNull(databaseWorkOrder)) {
-                List<String> errorNameList = workOrderErrorTypeService.getBaseMapper().getNotFinishWorkOrderErrors(workOrder.getWorkOrderNo());
+                List<String> errorNameList = workOrderErrorTypeService.getBaseMapper().getNotFinishWorkOrderErrors(databaseWorkOrder.getWorkOrderNo());
                 List<String> errorTypes = workOrder.getErrorTypeList();
                 if (errorNameList.containsAll(errorTypes)) {
                     // 濡傛灉锛屽浗鏍囩爜銆佹晠闅滅被鍨嬮兘涓�鏍凤紝鍒欒烦杩囦笉澶勭悊
@@ -138,20 +144,21 @@
                             workOrderErrorType.setCreateTime(now);
                             workOrderErrorType.setUpdateTime(now);
                             workOrderErrorType.setErrorName(errorType);
-                            workOrderErrorTypeService.save(workOrderErrorType);
+                            willAddErrorType.add(workOrderErrorType);
                         }
                     }
                     databaseWorkOrder.setStatus(WorkOrderStatusEnum.DISTRIBUTED);
                     databaseWorkOrder.setUpdateTime(now);
-                    baseMapper.updateById(databaseWorkOrder);
+                    willUpdateStatusWorkOrderList.add(databaseWorkOrder);
                     updateNum++;
                     // 鍚屾椂鏂板涓�涓繍缁村鐞嗕俊鎭紝琛ㄦ槑姝ゅ伐鍗曡璋冩暣
                     WorkOrderYwConditionRecord ywRecord = new WorkOrderYwConditionRecord();
-                    ywRecord.setWorkOrderId(databaseWorkOrder.getId());
+                    ywRecord.setWorkOrderNo(databaseWorkOrder.getWorkOrderNo());
                     ywRecord.setCommitUser(1);
                     ywRecord.setYwCondition("鏁呴殰绫诲瀷鏇存柊锛屽伐鍗曠姸鎬佽皟鏁翠负寰呭鐞�");
                     ywRecord.setCreateTime(new Date());
-                    workOrderYwConditionRecordMapper.insert(ywRecord);
+                    ywRecord.setSysMsg(Boolean.TRUE);
+                    willAddMsg.add(ywRecord);
                 }
             } else {
                 workOrder.setCreateTime(new Date());
@@ -163,6 +170,16 @@
 //                }
                 waitAddList.add(workOrder);
             }
+        }
+        if (willAddErrorType.size() > 0) {
+            workOrderErrorTypeService.getBaseMapper().addMany(willAddErrorType);
+        }
+        if (willAddMsg.size() > 0) {
+            workOrderYwConditionRecordMapper.insertMany(willAddMsg);
+        }
+        log.info("灏嗚鏇存柊鐨勫伐鍗曟暟锛�" + willUpdateStatusWorkOrderList.size());
+        if (willUpdateStatusWorkOrderList.size() > 0) {
+            this.baseMapper.updateMany(willUpdateStatusWorkOrderList);
         }
         if (CollectionUtils.isEmpty(waitAddList)) {
             return Boolean.TRUE;
@@ -285,7 +302,7 @@
         baseMapper.updateById(workOrder);
         // 娣诲姞涓�鏉″鏍歌褰�
         WorkOrderAuditingRecord workOrderAuditingRecord = new WorkOrderAuditingRecord();
-        workOrderAuditingRecord.setWorkOrderId(workOrder.getId());
+        workOrderAuditingRecord.setWorkOrderNo(workOrder.getWorkOrderNo());
         workOrderAuditingRecord.setAuditingUser(SecurityUtils.getLoginUser().getUserId().intValue());
         workOrderAuditingRecord.setResult(form.getAuditingResult().getDesc());
         workOrderAuditingRecord.setRemark(form.getAuditingRemark());
@@ -316,18 +333,18 @@
         if (list.isEmpty()) {
             return Result.error("娌℃湁宸ュ崟鍙互瀹℃牳");
         }
-        List<Integer> ids = list.stream().map(WorkOrder::getId).collect(Collectors.toList());
+        List<String> workOrderNoList = list.stream().map(WorkOrder::getWorkOrderNo).collect(Collectors.toList());
         List<String> serialNumbers = list.stream().map(WorkOrder::getSerialNumber).toList();
         // 宸ュ崟鐘舵��
         LambdaUpdateWrapper<WorkOrder> lambdaUpdateWrapper = new LambdaUpdateWrapper<>();
-        lambdaUpdateWrapper.in(WorkOrder::getId, ids);
+        lambdaUpdateWrapper.in(WorkOrder::getWorkOrderNo, workOrderNoList);
         lambdaUpdateWrapper.set(WorkOrder::getStatus, form.getAuditingResult());
         baseMapper.update(lambdaUpdateWrapper);
         // 娣诲姞澶氭潯瀹℃牳璁板綍
         List<WorkOrderAuditingRecord> workOrderAuditingRecords = new ArrayList<>();
-        for (Integer id : ids) {
+        for (String workOrderNo : workOrderNoList) {
             WorkOrderAuditingRecord workOrderAuditingRecord = new WorkOrderAuditingRecord();
-            workOrderAuditingRecord.setWorkOrderId(id);
+            workOrderAuditingRecord.setWorkOrderNo(workOrderNo);
             workOrderAuditingRecord.setAuditingUser(SecurityUtils.getLoginUser().getUserId().intValue());
             workOrderAuditingRecord.setResult(form.getAuditingResult().getDesc());
             workOrderAuditingRecord.setRemark(form.getAuditingRemark());
@@ -365,7 +382,7 @@
         baseMapper.updateById(workOrder);
         // 娣诲姞涓�鏉¤繍缁存儏鍐佃褰�
         WorkOrderYwConditionRecord workOrderYwConditionRecord = new WorkOrderYwConditionRecord();
-        workOrderYwConditionRecord.setWorkOrderId(workOrder.getId());
+        workOrderYwConditionRecord.setWorkOrderNo(workOrder.getWorkOrderNo());
         workOrderYwConditionRecord.setCommitUser(SecurityUtils.getLoginUser().getUserId().intValue());
         workOrderYwConditionRecord.setYwCondition(form.getYwCondition());
         workOrderYwConditionRecord.setYwProofMaterials(form.getYwProofMaterials());
@@ -375,23 +392,22 @@
     }
 
     @Override
-    public Result selectYwConditionByYwId(Integer id) {
-        List<WorkOrderYwConditionRecordVO> ywConditionList = workOrderYwConditionRecordMapper.selectYwConditionByYwId(id);
+    public List<WorkOrderYwConditionRecordVO> selectYwConditionByYwId(String workOrderNo) {
+        List<WorkOrderYwConditionRecordVO> ywConditionList = workOrderYwConditionRecordMapper.selectYwConditionByYwId(workOrderNo);
         ywConditionList.stream().forEach(item -> {
             if (Objects.nonNull(item.getSysMsg()) && item.getSysMsg()) {
                 item.setCommitUserName("绯荤粺娑堟伅");
             }
         });
-        return Result.ok().data(ywConditionList);
+        return ywConditionList;
     }
 
     @Override
-    public Result selectYwAuditingListByYwId(String id) {
-        return Result.ok().data(
-                new LambdaQueryChainWrapper<>(workOrderAuditingRecordMapper)
-                        .eq(WorkOrderAuditingRecord::getWorkOrderId, id)
+    public List<WorkOrderAuditingRecord> selectYwAuditingListByYwId(String workOrderNo) {
+        return new LambdaQueryChainWrapper<>(workOrderAuditingRecordMapper)
+                        .eq(WorkOrderAuditingRecord::getWorkOrderNo, workOrderNo)
                         .orderByAsc(WorkOrderAuditingRecord::getCreateTime)
-                        .list());
+                        .list();
     }
 
     @Override
@@ -645,38 +661,56 @@
 
 
     @Override
-    public String getFrameImgByDevice(String deviceId, String channelId) {
-        String url = String.format(this.rtspServer + "/start/%s/%s", deviceId, channelId);
+    public 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);
-        if (0 == wvpResult.getCode()) {
-            StreamContent data = (StreamContent) wvpResult.getData();
-            String rtspUrl = data.getRtsp();
+        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 {
-                    // 鍒涘缓FFmpegFrameGrabber瀵硅薄
-                    FFmpegFrameGrabber grabber = new FFmpegFrameGrabber(rtspUrl);
-                    grabber.start(); // 寮�濮嬫崟鑾�
+                    grabber = FFmpegFrameGrabber.createDefault(rtspUrl);
+                    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);
 
-                    Frame frame;
-                    if ((frame = grabber.grabFrame()) != null) {
-                        // 灏嗗抚杞崲涓哄浘鐗�
-                        Java2DFrameConverter converter = new Java2DFrameConverter();
-                        BufferedImage image = converter.convert(frame);
-                        File outputfile = new File(serverConfig.getUrl() + PlatformConfig.getUploadPath() + deviceId + IdUtils.fastSimpleUUID() + ".png");
-                        ImageIO.write(image, "png", outputfile);
-                        String imgPath = outputfile.getAbsolutePath();
-                        System.out.println("Saved " + imgPath);
-                        return imgPath;
+                        imgUrl = workOrderNo + "_" + IdUtils.fastSimpleUUID() + ".png";
+                        // 鐢熸垚鍥剧墖璺緞
+                        String imgPath = PlatformConfig.getUploadPath() + "/" + imgUrl;
+                        System.out.println("鍥剧墖淇濆瓨鍦板潃锛�" + imgPath);
+                        imgUrl = Constants.RESOURCE_PREFIX + "/" + imgUrl;
+                        // 淇濆瓨鍥剧墖
+                        opencv_imgcodecs.imwrite(imgPath, mat);
+                    } else {
+                        System.out.println("鏈崟鑾峰埌甯�");
                     }
-                    grabber.stop(); // 鍋滄鎹曡幏
-                    grabber.release(); // 閲婃斁璧勬簮
-                } catch (IOException e) {
+                } 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());
         }
-        return null;
+        System.out.println("鍥剧墖URL锛�" + imgUrl);
+        return imgUrl;
     }
 
     @Override
@@ -689,6 +723,19 @@
     public void updateImgById(Integer workOrderId, String imgPath) {
         new LambdaUpdateChainWrapper<>(baseMapper)
                 .eq(WorkOrder::getId, workOrderId)
-                .set(WorkOrder::getYwCheckResult, imgPath);
+                .set(WorkOrder::getYwCheckResult, imgPath)
+                .update();
+    }
+
+    @Override
+    public Result processImg(String workOrderNo) {
+        // 杩愮淮璁板綍
+        List<WorkOrderYwConditionRecordVO> workOrderYwConditionRecordVOS = this.selectYwConditionByYwId(workOrderNo);
+        // 瀹℃牳璁板綍
+        List<WorkOrderAuditingRecord> workOrderAuditingRecords = this.selectYwAuditingListByYwId(workOrderNo);
+        WorkOrderProcessVO process = new WorkOrderProcessVO();
+        process.setYwList(workOrderYwConditionRecordVOS);
+        process.setAuditingList(workOrderAuditingRecords);
+        return Result.ok().data(process);
     }
 }

--
Gitblit v1.8.0