From f0ec80a9eb4d12aca25f311fecf8f6e5edf8c577 Mon Sep 17 00:00:00 2001 From: fuliqi <fuliqi@qq.com> Date: 星期三, 06 十一月 2024 16:47:00 +0800 Subject: [PATCH] 工单加搜索table、一些搜索bug --- ycl-server/src/main/java/com/ycl/platform/service/impl/WorkOrderServiceImpl.java | 366 +++++++++++++++++++++++++++++++++++++++------------- 1 files changed, 273 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 6f47e90..241e65b 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.*; @@ -48,13 +54,26 @@ import org.springframework.beans.BeanUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Value; +import org.springframework.context.ApplicationContext; import org.springframework.data.redis.core.RedisTemplate; +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 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 +101,9 @@ private final SysConfigMapper configMapper; private final ReportMapper reportMapper; private final WorkOrderCheckImgMapper workOrderCheckImgMapper; + private final WorkOrderWhiteMapper workOrderWhiteMapper; + private final DeviceInfoMapper deviceInfoMapper; + private final ApplicationContext applicationContext; @Value("${rtsp.server:http://127.0.0.1:7788}") private String rtspServer; @@ -93,12 +115,16 @@ @Override @Transactional(rollbackFor = Exception.class) - //TODO锛氭竻绌篟edis瀹氭椂浠诲姟 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 +176,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 +217,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)); // 鏌ュ嚭閲嶇偣鐐逛綅銆佹櫘閫氱偣浣嶇殑澶勭悊鏃堕棿 @@ -200,6 +232,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 +274,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(); @@ -249,12 +288,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 +327,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 +342,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.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("娣诲姞鎴愬姛"); + // 鏌ヨ鍑虹櫧鍚嶅崟鍒楄〃 + 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("娣诲姞澶辫触"); } /** @@ -422,6 +464,7 @@ @Override @Transactional(rollbackFor = Exception.class) public Result ywCondition(WorkOrderYWConditionForm form) { + WorkOrder workOrder = baseMapper.selectById(form.getId()); if (Objects.isNull(workOrder)) { throw new ServiceException("宸ュ崟涓嶅瓨鍦�"); @@ -438,9 +481,37 @@ 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); @@ -507,7 +578,9 @@ @Override public Result page(WorkOrderQuery query) { IPage<WorkOrderVO> page = PageUtil.getPage(query, WorkOrderVO.class); - query.setUnitId(SecurityUtils.getUnitId()); + if(query.getUnitId()==null) { + query.setUnitId(SecurityUtils.getUnitId()); + } query.setStart(DateUtils.getDayStart(query.getStart())); query.setEnd(DateUtils.getDayEnd(query.getEnd())); baseMapper.page(page, query); @@ -558,7 +631,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 +693,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 +825,77 @@ @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); + log.info("璁块棶璺緞{}",url); 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 +949,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 +963,7 @@ } @Override - public Result processImg(String workOrderNo) { + public Result processImg (String workOrderNo){ WorkOrder workOrder = new LambdaQueryChainWrapper<>(baseMapper) .eq(WorkOrder::getWorkOrderNo, workOrderNo) .one(); @@ -922,7 +1004,7 @@ } @Override - public Result detailByNo(String workOrderNo) { + public Result detailByNo (String workOrderNo){ WorkOrderDetailVO workOrder = baseMapper.detailByNo(workOrderNo); // 鏄惁鎶ュ boolean hasReport = new LambdaQueryChainWrapper<>(reportMapper) @@ -942,4 +1024,102 @@ 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.setEnd(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