From e9af9f5cfeddbe5c0b33a3060b8ea6364c51e744 Mon Sep 17 00:00:00 2001 From: fuliqi <fuliqi@qq.com> Date: 星期五, 15 十一月 2024 11:37:10 +0800 Subject: [PATCH] 白名单故障类型 --- ycl-server/src/main/java/com/ycl/platform/service/impl/WorkOrderServiceImpl.java | 456 +++++++++++++++++++++++++++++++++++++++++--------------- 1 files changed, 335 insertions(+), 121 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 dad7b84..96e8e1b 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,17 +1,23 @@ package com.ycl.platform.service.impl; -import com.alibaba.excel.annotation.format.DateTimeFormat; +import com.alibaba.excel.EasyExcel; +import com.alibaba.excel.ExcelWriter; +import com.alibaba.excel.write.metadata.WriteSheet; import com.alibaba.fastjson2.JSON; +import com.alibaba.fastjson2.JSONArray; import com.alibaba.fastjson2.JSONObject; +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; import com.baomidou.mybatisplus.extension.conditions.update.LambdaUpdateChainWrapper; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.ycl.config.PlatformConfig; -import com.ycl.config.ServerConfig; import com.ycl.exception.ServiceException; import com.ycl.platform.domain.entity.*; +import com.ycl.platform.domain.excel.ErrorExport; +import com.ycl.platform.domain.excel.PointExport; +import com.ycl.platform.domain.excel.WorkOrderWhiteExport; import com.ycl.platform.domain.form.*; import com.ycl.platform.domain.query.*; import com.ycl.platform.domain.vo.*; @@ -23,13 +29,11 @@ import com.ycl.platform.service.WorkOrderAuditingRecordService; import com.ycl.platform.service.WorkOrderService; import com.ycl.platform.service.YwPointService; -import com.ycl.platform.wvp.StreamContent; import com.ycl.platform.wvp.WVPResult; import com.ycl.system.Result; import com.ycl.system.domain.SysConfig; import com.ycl.system.entity.SysDictData; import com.ycl.system.mapper.SysConfigMapper; -import com.ycl.system.mapper.SysDictDataMapper; import com.ycl.system.model.LoginUser; import com.ycl.system.page.PageUtil; import com.ycl.utils.DateUtils; @@ -39,38 +43,37 @@ import com.ycl.utils.uuid.IdUtils; import constant.Constants; import constant.RedisConstant; +import enumeration.ErrorType; import enumeration.general.*; +import jakarta.servlet.http.HttpServletResponse; import lombok.RequiredArgsConstructor; import lombok.extern.slf4j.Slf4j; import org.apache.commons.lang3.ObjectUtils; -import org.bytedeco.javacv.*; +import org.bytedeco.javacv.FFmpegFrameGrabber; +import org.bytedeco.javacv.Frame; +import org.bytedeco.javacv.FrameGrabber; +import org.bytedeco.javacv.OpenCVFrameConverter; 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; -import org.springframework.data.redis.core.RedisTemplate; +import org.springframework.context.ApplicationContext; +import org.springframework.scheduling.annotation.Async; 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 pojo.CascadeOption; -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.io.OutputStream; import java.time.LocalDateTime; import java.time.ZoneId; import java.util.*; import java.util.concurrent.TimeUnit; +import java.util.function.Function; import java.util.stream.Collectors; /** @@ -91,9 +94,13 @@ private final NotifyService notifyService; private final WorkOrderDistributeRecordMapper workOrderDistributeRecordMapper; private final WorkOrderErrorTypeServiceImpl workOrderErrorTypeService; + private final WorkOrderErrorTypeMapper workOrderErrorTypeMapper; + private final WorkOrderWhiteMapper workOrderWhiteMapper; private final SysConfigMapper configMapper; private final ReportMapper reportMapper; private final WorkOrderCheckImgMapper workOrderCheckImgMapper; + private final DeviceInfoMapper deviceInfoMapper; + private final ApplicationContext applicationContext; @Value("${rtsp.server:http://127.0.0.1:7788}") private String rtspServer; @@ -104,15 +111,19 @@ private RedisCache redisCache; @Override - @Transactional(rollbackFor = Exception.class) public synchronized Boolean innerAddWorkOrder(List<WorkOrder> workOrderList) { + //閬垮厤鍧�(浜嬪姟鍔犻攣浼氬け鏁堛�佹柟娉曞唴閮ㄨ皟鐢ㄤ簨鍔′細澶辨晥) + WorkOrderServiceImpl self = applicationContext.getBean(WorkOrderServiceImpl.class); + return self.batchAddWorkOrder(workOrderList); + } + + @Transactional(rollbackFor = Exception.class) + public Boolean batchAddWorkOrder(List<WorkOrder> workOrderList) { int total = workOrderList.size(); - workOrderList.stream().filter(item -> { + //娓呯悊鏁版嵁 + workOrderList = workOrderList.stream().filter(item -> { return StringUtils.hasText(item.getSerialNumber()) && Objects.nonNull(item.getStatus()) && !CollectionUtils.isEmpty(item.getErrorTypeList()); - }); - if (CollectionUtils.isEmpty(workOrderList)) { - return Boolean.TRUE; - } + }).collect(Collectors.toList()); // 鏍规嵁鍥芥爣鐮佸幓閲� workOrderList = workOrderList.stream() .collect(Collectors.toMap( @@ -120,6 +131,12 @@ p -> p, (existing, replacement) -> existing // 鍐茬獊鏃朵繚鐣欑涓�涓� )).values().stream().collect(Collectors.toList()); + //妫�鏌ョ櫧鍚嶅崟 + workOrderList = checkWhite(workOrderList); + + if (CollectionUtils.isEmpty(workOrderList)) { + return Boolean.TRUE; + } List<String> serialNumberList = workOrderList.stream().map(WorkOrder::getSerialNumber).collect(Collectors.toList()); // 鏌ュ嚭鏁版嵁搴撲腑鍥芥爣鐮佸搴旂殑鏈畬鎴愮殑宸ュ崟 List<WorkOrder> inDatabaseWorkOrderList = baseMapper.getNotFinishedWorkOrders(serialNumberList); @@ -161,7 +178,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++; @@ -199,9 +219,12 @@ 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) + .isNotNull(YwPoint::getUnitId) .list(); Map<String, YwPoint> pointMapping = pointList.stream().collect(Collectors.toMap(YwPoint::getSerialNumber, point -> point)); // 鏌ュ嚭閲嶇偣鐐逛綅銆佹櫘閫氱偣浣嶇殑澶勭悊鏃堕棿 @@ -211,6 +234,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 { @@ -234,23 +276,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.workOrderNO(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())); - } // 淇濆瓨閿欒绫诲瀷 for (String errorType : workOrder.getErrorTypeList()) { WorkOrderErrorType workOrderErrorType = new WorkOrderErrorType(); @@ -260,12 +290,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); // 淇濆瓨宸ュ崟鍜屾晠闅滅被鍨� @@ -294,6 +320,7 @@ return Boolean.TRUE; } + /** * 娣诲姞 * @@ -302,6 +329,16 @@ */ @Override public Result add(WorkOrderForm form) { + //妫�鏌ョ櫧鍚嶅崟 + List<WorkOrder> data = new ArrayList<>(); + WorkOrder workOrder = WorkOrderForm.getEntityByForm(form, null); + data.add(workOrder); + data = checkWhite(data); + if (CollectionUtils.isEmpty(data)) { + return Result.error("璇ヨ澶囩殑鏁呴殰绫诲瀷鍦ㄧ櫧鍚嶅崟涓紝涓嶈兘娣诲姞宸ュ崟"); + } + WorkOrder entity = data.get(0); + //鏌edis浠婃棩宸ュ崟鏁伴噺 int workOrderNum = 0; //UUID浣滀负value锛屼繚璇佷笂閿佺殑绾跨▼鑷繁瑙i攣 @@ -317,7 +354,7 @@ } else { return Result.error("宸ュ崟姝e湪娣诲姞锛岃绋嶅悗鍐嶈瘯"); } - WorkOrder entity = WorkOrderForm.getEntityByForm(form, null); + entity.setCreateTime(DateUtils.getNowDate()); entity.setStatus(WorkOrderStatusEnum.WAIT_DISTRIBUTE); Date now = new Date(); @@ -332,6 +369,7 @@ } return Result.error("娣诲姞澶辫触"); } + /** * 淇敼 @@ -385,14 +423,10 @@ @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, WorkOrder::getSerialNumber) - .list(); + //鏍规嵁宸ュ崟缂栧彿鑾峰彇宸ュ崟 + List<WorkOrder> list = baseMapper.selectByNos(form.getWorkOrderNumbers()); if (list.isEmpty()) { - return Result.error("娌℃湁宸ュ崟鍙互瀹℃牳"); + return Result.error("娌℃湁寰呭鏍稿伐鍗�"); } List<String> workOrderNoList = list.stream().map(WorkOrder::getWorkOrderNo).collect(Collectors.toList()); List<String> serialNumbers = list.stream().map(WorkOrder::getSerialNumber).toList(); @@ -433,6 +467,7 @@ @Override @Transactional(rollbackFor = Exception.class) public Result ywCondition(WorkOrderYWConditionForm form) { + WorkOrder workOrder = baseMapper.selectById(form.getId()); if (Objects.isNull(workOrder)) { throw new ServiceException("宸ュ崟涓嶅瓨鍦�"); @@ -449,13 +484,45 @@ workOrderYwConditionRecord.setYwProofMaterials(form.getYwProofMaterials()); workOrderYwConditionRecord.setSysMsg(Boolean.FALSE); workOrderYwConditionRecordMapper.insert(workOrderYwConditionRecord); + //寮傛鑾峰彇鍥剧墖 + WorkOrderServiceImpl self = applicationContext.getBean(WorkOrderServiceImpl.class); + self.getImage(workOrder); return Result.ok("鎿嶄綔鎴愬姛"); + } + + @Async("threadPoolTaskExecutor") + public void getImage(WorkOrder workOrder) { + List<DeviceInfo> gbDevices = new LambdaQueryChainWrapper<>(deviceInfoMapper) + .orderByDesc(DeviceInfo::getUpdateTime) + .last("limit 1") + .list(); + if (CollectionUtils.isEmpty(gbDevices)) { + return; + } + // 鍥芥爣璁惧鐨勭紪鐮佸氨鏄彇瑙嗛娴佺殑璁惧缂栫爜锛屽浗鏍囪澶囧氨涓�涓�傚浗鏍囪澶囩殑姣忎竴涓�氶亾浠h〃涓�涓憚鍍忓ご锛屼篃灏辨槸璁惧id鏄彇娴佺殑閫氶亾id + String frameImg = null; + try { + log.info("鍥芥爣骞冲彴:{},璁惧缂栫爜:{},宸ュ崟鍙�:{}", gbDevices.get(0).getDeviceId(), workOrder.getSerialNumber(), workOrder.getWorkOrderNo()); + frameImg = this.getFrameImgByDevice(gbDevices.get(0).getDeviceId(), workOrder.getSerialNumber(), workOrder.getWorkOrderNo()); + } catch (Exception e) { + e.printStackTrace(); + } + if (StringUtils.hasText(frameImg)) { + WorkOrderCheckImg img = new WorkOrderCheckImg(); + img.setWorkOrderNo(workOrder.getWorkOrderNo()); + img.setImgUrl(frameImg); + img.setCreateTime(new Date()); + workOrderCheckImgMapper.insert(img); + } } @Override public List<WorkOrderYwConditionRecordVO> selectYwConditionByYwId(String workOrderNo) { List<WorkOrderYwConditionRecordVO> ywConditionList = workOrderYwConditionRecordMapper.selectYwConditionByYwId(workOrderNo); ywConditionList.stream().forEach(item -> { + if (item.getUserId().equals(1)) { + item.setUnitName("绠$悊鍛�"); + } if (Objects.nonNull(item.getSysMsg()) && item.getSysMsg()) { item.setUnitName("绯荤粺娑堟伅"); } @@ -503,6 +570,19 @@ */ @Override public Result removeById(String id) { + WorkOrder workOrder = baseMapper.selectById(id); + String workOrderNo = workOrder.getWorkOrderNo(); + //鍒犻櫎宸ュ崟瀹℃牳璁板綍 + workOrderAuditingRecordMapper.delete(new QueryWrapper<WorkOrderAuditingRecord>().eq("work_order_no", workOrderNo)); + //鍒犻櫎宸ュ崟鍥剧墖璁板綍 + workOrderCheckImgMapper.delete(new QueryWrapper<WorkOrderCheckImg>().eq("work_order_no", workOrderNo)); + //鍒犻櫎宸ュ崟涓嬪彂璁板綍 + workOrderDistributeRecordMapper.delete(new QueryWrapper<WorkOrderDistributeRecord>().eq("work_order_no", workOrderNo)); + //鍒犻櫎宸ュ崟鏁呴殰绫诲瀷 + workOrderErrorTypeMapper.delete(new QueryWrapper<WorkOrderErrorType>().eq("work_order_no", workOrderNo)); + //鍒犻櫎宸ュ崟鎯呭喌璁板綍 + workOrderYwConditionRecordMapper.delete(new QueryWrapper<WorkOrderYwConditionRecord>().eq("work_order_no", workOrderNo)); + if (baseMapper.deleteById(id) > 0) { return Result.ok("鍒犻櫎鎴愬姛"); } @@ -518,14 +598,27 @@ @Override 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())); + if (query.getUnitId() == null) { + query.setUnitId(SecurityUtils.getUnitId()); + } + if (query.getStart() != null) query.setStart(DateUtils.getDayStart(query.getStart())); + if (query.getEnd() != null) query.setEnd(DateUtils.getDayEnd(query.getEnd())); baseMapper.page(page, query); if (!CollectionUtils.isEmpty(page.getRecords())) { page.getRecords().stream().forEach(item -> { if (StringUtils.hasText(item.getErrorType())) { - item.setErrorTypeList(List.of(item.getErrorType().split(","))); + List<String> errorTypeList = new ArrayList<>(List.of(item.getErrorType().split(","))); + int index = errorTypeList.indexOf(query.getErrorTypeLabel()); + if (index != -1) { + // 淇濆瓨瑕佺Щ鍔ㄧ殑鍏冪礌 + String firstOccurrence = errorTypeList.get(index); + // 浠庡垪琛ㄤ腑绉婚櫎璇ュ厓绱狅紙绗竴娆″嚭鐜扮殑浣嶇疆锛� + errorTypeList.remove(index); + // 灏嗚鍏冪礌娣诲姞鍒板垪琛ㄧ殑棣栦綅 + errorTypeList.add(0, firstOccurrence); + } + // 璁剧疆鏇存柊鍚庣殑鍒楄〃 + item.setErrorTypeList(errorTypeList); } if (StringUtils.hasText(item.getImgListStr())) { item.setImgList(List.of(item.getImgListStr().split(","))); @@ -538,6 +631,8 @@ @Override public Result distributePage(DistributeWorkOrderQuery query) { IPage<WorkOrderVO> page = PageUtil.getPage(query, WorkOrderVO.class); + if (query.getStart() != null) query.setStart(DateUtils.getDayStart(query.getStart())); + if (query.getEnd() != null) query.setEnd(DateUtils.getDayEnd(query.getEnd())); baseMapper.distributePage(page, query); return Result.ok().data(page).total(page.getTotal()); } @@ -763,61 +858,70 @@ @Override - public synchronized String getFrameImgByDevice(String deviceId, String channelId, String workOrderNo) throws Exception { - String url = String.format(this.rtspServer + "/api/play/start/%s/%s", deviceId, channelId); + public String getFrameImgByDevice(String deviceId, String channelId, String workOrderNo) throws Exception { + String url = String.format(this.rtspServer + "/api/play/start/img/%s/%s", deviceId, channelId); + log.info("璁块棶璺緞{}", url); String result = HttpUtils.sendGet(url); - log.info("鎷垮埌鍙栨祦鍝嶅簲缁撴灉锛�" + result); + 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 = 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()); + 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); @@ -878,13 +982,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) @@ -892,7 +996,7 @@ } @Override - public Result processImg (String workOrderNo){ + public Result processImg(String workOrderNo) { WorkOrder workOrder = new LambdaQueryChainWrapper<>(baseMapper) .eq(WorkOrder::getWorkOrderNo, workOrderNo) .one(); @@ -933,7 +1037,7 @@ } @Override - public Result detailByNo (String workOrderNo){ + public Result detailByNo(String workOrderNo) { WorkOrderDetailVO workOrder = baseMapper.detailByNo(workOrderNo); // 鏄惁鎶ュ boolean hasReport = new LambdaQueryChainWrapper<>(reportMapper) @@ -955,23 +1059,133 @@ } @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"); + public List<WorkOrderVO> export(WorkOrderExportQuery query) { + if (query.getUnitId() == null) { + query.setUnitId(SecurityUtils.getUnitId()); } - if (query.getEnd() == null) { - query.setStart(format.format(DateUtils.getDayEnd(new Date()))); - } else { - query.setEnd(query.getEnd() + " 23:59:59"); - } + if (query.getStart() != null) query.setStart(query.getStart() + " 00:00:00"); + if (query.getEnd() != null) query.setEnd(query.getEnd() + " 23:59:59"); - List<WorkOrderVO> export = baseMapper.export(query); - - System.out.println(export); - return export; + return baseMapper.export(query); } + + /** + * 妫�娴嬪伐鍗曟寜閽� + * + * @param workOrderNo + * @return + */ + @Override + public Result checkImage(String workOrderNo, String serialNumber) { + // 鏌ュ嚭鍥芥爣璁惧锛屽氨涓�鏉℃暟鎹� + List<DeviceInfo> gbDevices = new LambdaQueryChainWrapper<>(deviceInfoMapper) + .orderByDesc(DeviceInfo::getUpdateTime) + .last("limit 1") + .list(); + if (!CollectionUtils.isEmpty(gbDevices)) { + try { + String imgUrl = this.getFrameImgByDevice(gbDevices.get(0).getDeviceId(), serialNumber, workOrderNo); + if (StringUtils.hasText(imgUrl)) { + WorkOrderCheckImg img = new WorkOrderCheckImg(); + img.setWorkOrderNo(workOrderNo); + img.setImgUrl(imgUrl); + img.setCreateTime(new Date()); + workOrderCheckImgMapper.insert(img); + } + return Result.ok().data(imgUrl); + } catch (Exception e) { + e.printStackTrace(); + } + } + return Result.ok(); + } + + @Override + @Transactional(rollbackFor = Exception.class) + public Result batchDeleteWorkOrder(List<String> workOrderNos) { + if (!CollectionUtils.isEmpty(workOrderNos)) { + //鍒犻櫎宸ュ崟瀹℃牳璁板綍 + workOrderAuditingRecordMapper.delete(new QueryWrapper<WorkOrderAuditingRecord>().in("work_order_no", workOrderNos)); + //鍒犻櫎宸ュ崟鍥剧墖璁板綍 + workOrderCheckImgMapper.delete(new QueryWrapper<WorkOrderCheckImg>().in("work_order_no", workOrderNos)); + //鍒犻櫎宸ュ崟涓嬪彂璁板綍 + workOrderDistributeRecordMapper.delete(new QueryWrapper<WorkOrderDistributeRecord>().in("work_order_no", workOrderNos)); + //鍒犻櫎宸ュ崟鏁呴殰绫诲瀷 + workOrderErrorTypeMapper.delete(new QueryWrapper<WorkOrderErrorType>().in("work_order_no", workOrderNos)); + //鍒犻櫎宸ュ崟鎯呭喌璁板綍 + workOrderYwConditionRecordMapper.delete(new QueryWrapper<WorkOrderYwConditionRecord>().in("work_order_no", workOrderNos)); + //鍒犻櫎宸ュ崟 + this.baseMapper.delete(new QueryWrapper<WorkOrder>().in("work_order_no", workOrderNos)); + } + return Result.ok(); + } + + @Override + public Result errorAll() { + List<CascadeOption> results = new ArrayList<>(); + CascadeOption video = new CascadeOption(); + video.setLabel("瑙嗛"); + video.setValue("VIDEO"); + List<ErrorType> videoErr = ErrorType.getErrorTypesByCategory("VIDEO"); + List<ErrorType> videoErr1 = ErrorType.getErrorTypesByCategory("COMMON"); + videoErr.addAll(videoErr1); + List<CascadeOption> videoChildren = videoErr.stream().map(item -> CascadeOption.builder() + .value(item.getValue()) + .label(item.getDesc()) + .build()).collect(Collectors.toList()); + video.setChildren(videoChildren); + + CascadeOption car = new CascadeOption(); + car.setLabel("杞﹁締"); + car.setValue("CAR"); + List<ErrorType> carErr = ErrorType.getErrorTypesByCategory("CAR"); + List<ErrorType> carErr1 = ErrorType.getErrorTypesByCategory("COMMON"); + List<ErrorType> carErr2 = ErrorType.getErrorTypesByCategory("CARORFACE"); + carErr.addAll(carErr1); + carErr.addAll(carErr2); + List<CascadeOption> carChildren = carErr.stream().map(item -> CascadeOption.builder() + .value(item.getValue()) + .label(item.getDesc()) + .build()).collect(Collectors.toList()); + car.setChildren(carChildren); + + CascadeOption face = new CascadeOption(); + face.setLabel("浜鸿劯"); + face.setValue("FACE"); + List<ErrorType> faceErr = ErrorType.getErrorTypesByCategory("FACE"); + List<ErrorType> faceErr1 = ErrorType.getErrorTypesByCategory("COMMON"); + List<ErrorType> faceErr2 = ErrorType.getErrorTypesByCategory("CARORFACE"); + faceErr.addAll(faceErr1); + faceErr.addAll(faceErr2); + List<CascadeOption> faceChildren = faceErr.stream().map(item -> CascadeOption.builder() + .value(item.getValue()) + .label(item.getDesc()) + .build()).collect(Collectors.toList()); + face.setChildren(faceChildren); + + results.add(video); + results.add(car); + results.add(face); + return Result.ok().data(results); + } + + //鐧藉悕鍗曟牎楠� + private List<WorkOrder> checkWhite(List<WorkOrder> workOrderList) { + // 鏌ヨ鍑虹櫧鍚嶅崟鍒楄〃 + Map<String, WorkOrderWhite> whiteMap = workOrderWhiteMapper.selectList().stream().collect(Collectors.toMap(WorkOrderWhite::getSerialNumber, Function.identity())); + workOrderList = workOrderList.stream() + .filter(workOrder -> { + List<String> errorTypeList = workOrder.getErrorTypeList(); + WorkOrderWhite white = whiteMap.get(workOrder.getSerialNumber()); + if (white != null) { + List<String> whiteErrorListSet = Arrays.asList(white.getErrorType().split(",")); + errorTypeList.removeIf(whiteErrorListSet::contains); + } + return !errorTypeList.isEmpty(); // 鍙繚鐣� errorTypeList 闈炵┖鐨� WorkOrder + }) + .collect(Collectors.toList()); + return workOrderList; + } + + } -- Gitblit v1.8.0