From 7ef4892f9f24f941aca37e6b3991b808a0aca619 Mon Sep 17 00:00:00 2001
From: zhanghua <314079846@qq.com>
Date: 星期五, 08 九月 2023 11:16:35 +0800
Subject: [PATCH] 优化
---
ycl-platform/src/main/java/com/ycl/service/video/impl/VideoAlarmReportServiceImpl.java | 200 +++++++++++++++++++++++++++++++++++++++++--------
1 files changed, 166 insertions(+), 34 deletions(-)
diff --git a/ycl-platform/src/main/java/com/ycl/service/video/impl/VideoAlarmReportServiceImpl.java b/ycl-platform/src/main/java/com/ycl/service/video/impl/VideoAlarmReportServiceImpl.java
index 4cac604..5873897 100644
--- a/ycl-platform/src/main/java/com/ycl/service/video/impl/VideoAlarmReportServiceImpl.java
+++ b/ycl-platform/src/main/java/com/ycl/service/video/impl/VideoAlarmReportServiceImpl.java
@@ -1,20 +1,27 @@
package com.ycl.service.video.impl;
+import com.alibaba.druid.util.StringUtils;
import com.alibaba.fastjson.JSON;
+import com.alibaba.fastjson.JSONObject;
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
-import com.ycl.dto.video.AlarmDataParam;
-import com.ycl.dto.video.AlarmParam;
-import com.ycl.dto.video.CameraInfParam;
+import com.ycl.dto.video.*;
import com.ycl.entity.video.VideoAlarmReport;
import com.ycl.mapper.video.VideoAlarmReportMapper;
import com.ycl.service.caseHandler.IViolationsService;
+import com.ycl.service.oss.OssService;
import com.ycl.service.video.IVideoAlarmReportService;
import com.ycl.utils.MediaFileUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
+import javax.imageio.ImageIO;
import javax.xml.bind.DatatypeConverter;
+import java.awt.*;
+import java.awt.image.BufferedImage;
import java.io.ByteArrayInputStream;
+import java.io.ByteArrayOutputStream;
+import java.io.IOException;
import java.io.InputStream;
import java.text.ParseException;
import java.text.SimpleDateFormat;
@@ -23,6 +30,7 @@
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
+import java.util.function.Consumer;
/**
* VideoAlarmReportServiceImpl
@@ -38,6 +46,9 @@
private IViolationsService violationsService;
@Autowired
+ private OssService ossService;
+
+ @Autowired
public void setMediaFileUtil(MediaFileUtil mediaFileUtil) {
this.mediaFileUtil = mediaFileUtil;
}
@@ -47,48 +58,169 @@
this.violationsService = violationsService;
}
+// @Override
+// public void save(AlarmParam alarmParam) {
+// if (alarmParam != null && alarmParam.getAlarmData() != null) {
+// CameraInfParam cameraInf = alarmParam.getCameraInf();
+// SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd'T'HH:mm:ss.SSSXXX");
+// String picData = "";
+// if (!StringUtils.isEmpty(alarmParam.getPicData())) {
+//// String[] strings = alarmParam.getPicData().split(",");
+// String extension = "jpg";
+//
+// byte[] data = DatatypeConverter.parseBase64Binary(alarmParam.getPicData());
+// InputStream inputStream = new ByteArrayInputStream(data);
+//
+// try {
+// if (!StringUtils.isEmpty(alarmParam.getAlarmData().getAlarmObject().toString())) {
+// String alarmObject = JSON.toJSONString(alarmParam.getAlarmData().getAlarmObject());
+// if (!StringUtils.isEmpty(alarmObject)) {
+// BufferedImage image = ImageIO.read(inputStream);
+//
+// Graphics2D g2 = (Graphics2D) image.getGraphics();
+// g2.setColor(Color.RED);
+// g2.setStroke(new BasicStroke(3.0f));
+// List<AlarmObject> alarmObjects = JSON.parseArray(alarmObject, AlarmObject.class);
+// alarmObjects.forEach(o -> {
+// o.getObjectRect().forEach(rect -> {
+// g2.drawRect(rect.getX(), rect.getY(), rect.getW(), rect.getH());
+// });
+// });
+// ByteArrayOutputStream os = new ByteArrayOutputStream();
+// ImageIO.write(image, extension, os);
+// inputStream = new ByteArrayInputStream(os.toByteArray());
+// }
+// }
+// picData = mediaFileUtil.savePicture(inputStream, extension);
+// } catch (Exception e) {
+// e.printStackTrace();
+// }
+// }
+// List<VideoAlarmReport> videoAlarmReports = new ArrayList<>();
+//
+// AlarmDataParam alarmData = alarmParam.getAlarmData();
+// List<VideoAlarmReport> list = getByAlarmId(alarmData.getAlarmID());
+// try {
+// Date date = sdf.parse(alarmData.getAlarmTime());
+// Instant instant = date.toInstant();
+// ZoneId zoneId = ZoneId.systemDefault();
+// VideoAlarmReport videoAlarmReport = VideoAlarmReport.builder()
+// .ip(cameraInf.getIp()).port(cameraInf.getPort())
+// .user(cameraInf.getUser()).password(cameraInf.getPassword())
+// .channel(cameraInf.getChannel()).platResourceId(cameraInf.getPlatResourceID())
+// .picData(picData).alarmTime(instant.atZone(zoneId).toLocalDateTime())
+// .alarmId(alarmData.getAlarmID()).algoType(alarmData.getAlgoType())
+// .alarmType(alarmData.getAlarmType()).alarmName(alarmData.getAlarmName())
+// .algoName(alarmData.getAlgoName()).alarmObj(JSON.toJSONString(alarmData.getAlarmObject()))
+// .build();
+// if (list.size() == 0) {
+//
+// videoAlarmReports.add(videoAlarmReport);
+//
+// this.saveBatch(videoAlarmReports, videoAlarmReports.size());
+//
+// violationsService.saveFromVideo(videoAlarmReports);
+//
+// } else {
+// videoAlarmReport.setId(list.get(0).getId());
+// this.updateById(videoAlarmReport);
+// }
+// } catch (ParseException e) {
+// e.printStackTrace();
+// }
+// }
+// }
+
@Override
public void save(AlarmParam alarmParam) {
- if (alarmParam != null && alarmParam.getAlarmData().size() > 0) {
+ if (alarmParam != null && alarmParam.getAlarmData() != null) {
+// 鍙栧嚭CameraInf
CameraInfParam cameraInf = alarmParam.getCameraInf();
SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd'T'HH:mm:ss.SSSXXX");
String picData = "";
- String[] strings = alarmParam.getPicData().split(",");
- String extension = "jpg";
+ if (!StringUtils.isEmpty(alarmParam.getPicData())) {
+// String[] strings = alarmParam.getPicData().split(",");
+ String extension = "jpg";
- byte[] data = DatatypeConverter.parseBase64Binary(strings[1]);
- InputStream inputStream = new ByteArrayInputStream(data);
- try {
- picData = mediaFileUtil.savePicture(inputStream, extension);
- List<VideoAlarmReport> videoAlarmReports = new ArrayList<>();
- for (int i = 0; i < alarmParam.getAlarmData().size(); i++) {
- AlarmDataParam alarmData = alarmParam.getAlarmData().get(i);
- /******************** 澶勭悊鍥剧墖 **********************/
- try {
- Date date = sdf.parse(alarmData.getAlarmTime());
- Instant instant = date.toInstant();
- ZoneId zoneId = ZoneId.systemDefault();
+// 鎸夌収Base64缂栫爜瑙ｆ瀽鏁版嵁
+ byte[] data = DatatypeConverter.parseBase64Binary(alarmParam.getPicData());
+ InputStream inputStream = new ByteArrayInputStream(data);
+ try {
+// 濡傛灉List<AlarmDataParam>鍚堢悊
+ if (alarmParam.getAlarmData() != null && alarmParam.getAlarmData().size() != 0) {
- VideoAlarmReport videoAlarmReport = VideoAlarmReport.builder()
- .ip(cameraInf.getIp()).port(cameraInf.getPort())
- .user(cameraInf.getUser()).password(cameraInf.getPassword())
- .channel(cameraInf.getChannel()).platResourceId(cameraInf.getPlatResourceID())
- .picData(picData).alarmTime(instant.atZone(zoneId).toLocalDateTime())
- .alarmId(alarmData.getAlarmID()).algoType(alarmData.getAlgoType())
- .algoName(alarmData.getAlgoName()).alarmObj(JSON.toJSONString(alarmData.getAlarmObject()))
- .build();
+ BufferedImage image = ImageIO.read(inputStream);
+ Graphics2D g2 = (Graphics2D) image.getGraphics();
+ g2.setColor(Color.RED);
+ g2.setStroke(new BasicStroke(3.0f));
+
+ List<AlarmDataParam> alarmData = alarmParam.getAlarmData();
+
+ alarmData.forEach(new Consumer<AlarmDataParam>() {
+ @Override
+ public void accept(AlarmDataParam alarmDataParam) {
+ List<AlarmObject> alarmObject = alarmDataParam.getAlarmObject();
+ alarmObject.forEach(new Consumer<AlarmObject>() {
+ @Override
+ public void accept(AlarmObject alarmObject) {
+ ObjectRect rect = alarmObject.getObjectRect();
+ g2.drawRect(rect.getX(), rect.getY(), rect.getW(), rect.getH());
+ }
+ });
+ }
+ });
+
+ ByteArrayOutputStream os = new ByteArrayOutputStream();
+ ImageIO.write(image, extension, os);
+ inputStream = new ByteArrayInputStream(os.toByteArray());
+
+ }
+ picData = ossService.uploadImages(inputStream, extension);
+ } catch (Exception e) {
+ System.out.println("淇濆瓨鎶ヨ鏁版嵁澶辫触---鍥剧墖涓婁紶澶辫触锛�" + e.getMessage());
+// e.printStackTrace();
+ }
+ }
+ List<VideoAlarmReport> videoAlarmReports = new ArrayList<>();
+
+ List<AlarmDataParam> alarmData = alarmParam.getAlarmData();
+ for (AlarmDataParam alarmDatum : alarmData) {
+ List<VideoAlarmReport> list = getByAlarmId(alarmDatum.getAlarmID());
+ try {
+ Date date = sdf.parse(alarmDatum.getAlarmTime());
+ Instant instant = date.toInstant();
+ ZoneId zoneId = ZoneId.systemDefault();
+ VideoAlarmReport videoAlarmReport = VideoAlarmReport.builder()
+ .ip(cameraInf.getIp()).port(cameraInf.getPort())
+ .user(cameraInf.getUser()).password(cameraInf.getPassword())
+ .channel(cameraInf.getChannel()).platResourceId(cameraInf.getPlatResourceID())
+ .picData(picData).alarmTime(instant.atZone(zoneId).toLocalDateTime())
+ .alarmId(alarmDatum.getAlarmID()).algoType(alarmDatum.getAlgoType())
+ .alarmType(alarmDatum.getAlarmType()).alarmName(alarmDatum.getAlarmName())
+ .algoName(alarmDatum.getAlgoName()).alarmObj(JSON.toJSONString(alarmDatum.getAlarmObject()))
+ .build();
+ if (list.size() == 0) {
+
videoAlarmReports.add(videoAlarmReport);
- } catch (ParseException e) {
- e.printStackTrace();
- }
- }
- this.saveBatch(videoAlarmReports, videoAlarmReports.size());
+ this.saveBatch(videoAlarmReports, videoAlarmReports.size());
- violationsService.saveFromVideo(videoAlarmReports);
- } catch (Exception e) {
- e.printStackTrace();
+ violationsService.saveFromVideo(videoAlarmReports);
+
+ } else {
+ videoAlarmReport.setId(list.get(0).getId());
+ this.updateById(videoAlarmReport);
+ }
+ } catch (ParseException e) {
+ System.out.println("淇濆瓨鎶ヨ鏁版嵁澶辫触锛�" + e.getMessage());
+// e.printStackTrace();
+ }
}
}
}
+
+ private List<VideoAlarmReport> getByAlarmId(String alarmId) {
+ LambdaQueryWrapper<VideoAlarmReport> queryWrapper = new LambdaQueryWrapper<VideoAlarmReport>().eq(VideoAlarmReport::getAlarmId, alarmId);
+ return this.list(queryWrapper);
+ }
}
--
Gitblit v1.8.0