zgyw.git - Gitblit

			@@ -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;
			@@ -55,6 +56,14 @@
			import org.springframework.util.CollectionUtils;
			import org.springframework.util.StringUtils;

			import javax.swing.*;
			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.*;
			@@ -93,7 +102,6 @@

			@Override
			@Transactional(rollbackFor = Exception.class)
			//TODO：Redis加锁，从Redis读取当日工单数据量
			public synchronized Boolean innerAddWorkOrder(List<WorkOrder> workOrderList) {
			int total = workOrderList.size();
			workOrderList.stream().filter(item -> {
			@@ -231,7 +239,7 @@
			}
			//数字前面补0
			workOrderNum++;
			workOrder.setWorkOrderNo(IdUtils.randomNO(now, String.format("%05d", workOrderNum)));
			workOrder.setWorkOrderNo(IdUtils.workOrderNO(now, String.format("%05d", workOrderNum)));
			if (Objects.nonNull(point.getUnitId())) {
			workOrder.setUnitId(Math.toIntExact(point.getUnitId()));
			}
			@@ -292,14 +300,14 @@
			@Override
			public Result add(WorkOrderForm form) {
			//查redis今日工单数量
			int workOrderNum = 1;
			int workOrderNum = 0;
			//UUID作为value，保证上锁的线程自己解锁
			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);
			@@ -312,7 +320,7 @@
			Date now = new Date();
			entity.setCreateTime(now);
			entity.setUpdateTime(now);
			entity.setWorkOrderNo(IdUtils.randomNO(now, String.format("%05d", workOrderNum)));
			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);
			@@ -558,7 +566,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 +628,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,27 +760,39 @@


			@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);
			log.error("拿到取流响应结果：" + 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"); // 取mp4地址
			if (0 == wvpResult.getCode()) {
			log.error("请求成功");
			StreamContent data = (StreamContent) wvpResult.getData();
			String rtspUrl = data.getRtsp();
			log.error("目标地址：" + rtspUrl);
			if (StringUtils.hasText(rtspUrl)) {
			System.out.println("目标地址：" + rtspUrl);
			FFmpegFrameGrabber grabber = null;
			try {
			grabber = FFmpegFrameGrabber.createDefault(rtspUrl);
			//设置10s超时
			grabber.setTimeout(10000);
			FFmpegFrameGrabber grabber = FFmpegFrameGrabber.createDefault(rtspUrl);
			grabber.setOption("rtsp_transport", "tcp"); // 使用tcp的方式，不然会丢包很严重
			grabber.setImageWidth(960);
			grabber.setImageHeight(540);
			grabber.start();
			Frame frame = grabber.grabImage(); // 直接捕获一帧
			if (frame != null) {
			System.out.println("成功捕获一帧");
			// 将Frame转换为Mat
			OpenCVFrameConverter.ToMat converter = new OpenCVFrameConverter.ToMat();
			CanvasFrame canvasFrame = new CanvasFrame("正茂");// 创建窗口
			canvasFrame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);// 设置窗口关闭程序是否也跟随关闭
			canvasFrame.setAlwaysOnTop(true);

			OpenCVFrameConverter.ToMat converter = new OpenCVFrameConverter.ToMat();
			int ex = 0;
			while (true) {
			if (ex > 1) {
			break;
			}
			Frame frame = grabber.grabImage();
			canvasFrame.showImage(frame);
			//程序到这里其实已经实现了预览的功能了，下面的方法就是将流保存成图片

			//opencv_core.Mat
			Mat mat = converter.convertToMat(frame);

			imgUrl = workOrderNo + "_" + IdUtils.fastSimpleUUID() + ".png";
			@@ -779,29 +800,25 @@
			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) {

			ex++;
			try {
			grabber.stop(); // 停止捕获
			} catch (FrameGrabber.Exception e) {
			e.printStackTrace();
			Thread.sleep(1000);
			}
			// 通常不需要调用release()，因为stop()会处理资源释放
			// grabber.release(); // 释放资源
			catch (InterruptedException e)
			{
			e.printStackTrace();
			} // 1000毫秒刷新一次圖像
			}
			grabber.stop(); // 停止捕获
			grabber.release(); // 释放资源
			} catch (IOException e) {
			e.printStackTrace();
			}
			}
			} else {
			System.out.println("请求失败，错误码：" + wvpResult.getCode());
			}
			System.out.println("图片URL：" + imgUrl);
			return imgUrl;
			}

			@@ -942,4 +959,25 @@
			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;
			}
			}