From 98f8160bd6531ef118d1bfcfb90368a3120ff70d Mon Sep 17 00:00:00 2001
From: baizonghao <1719256278@qq.com>
Date: 星期六, 01 四月 2023 13:22:45 +0800
Subject: [PATCH] AI算法效能，AI事件统计
---
ycl-platform/src/main/java/com/ycl/service/caseHandler/impl/ViolationsServiceImpl.java | 20 +++++++++++++++++++-
1 files changed, 19 insertions(+), 1 deletions(-)
diff --git a/ycl-platform/src/main/java/com/ycl/service/caseHandler/impl/ViolationsServiceImpl.java b/ycl-platform/src/main/java/com/ycl/service/caseHandler/impl/ViolationsServiceImpl.java
index 57ab292..8794a56 100644
--- a/ycl-platform/src/main/java/com/ycl/service/caseHandler/impl/ViolationsServiceImpl.java
+++ b/ycl-platform/src/main/java/com/ycl/service/caseHandler/impl/ViolationsServiceImpl.java
@@ -6,12 +6,15 @@
import com.ycl.entity.caseHandler.EventSource;
import com.ycl.entity.caseHandler.QuestionCategory;
import com.ycl.entity.caseHandler.Violations;
+import com.ycl.entity.dict.DataDictionary;
import com.ycl.entity.video.VideoAlarmReport;
import com.ycl.entity.video.VideoPoint;
+import com.ycl.enums.common.DictTypeEnum;
import com.ycl.mapper.caseHandler.BaseCaseMapper;
import com.ycl.mapper.caseHandler.ViolationsMapper;
import com.ycl.service.caseHandler.IBaseCaseService;
import com.ycl.service.caseHandler.IViolationsService;
+import com.ycl.service.dict.IDataDictionaryService;
import com.ycl.service.video.impl.IVideoPointService;
import com.ycl.vo.cockpit.enforcementEvents.VideoAndAreaVO;
import org.springframework.beans.factory.annotation.Autowired;
@@ -34,6 +37,7 @@
private IVideoPointService videoPointService;
private IBaseCaseService baseCaseService;
+ private IDataDictionaryService dataDictionaryService;
@Resource
BaseCaseMapper baseCaseMapper;
@@ -48,11 +52,18 @@
this.baseCaseService = baseCaseService;
}
+ @Autowired
+ public void setDataDictionaryService(IDataDictionaryService dataDictionaryService) {
+ this.dataDictionaryService = dataDictionaryService;
+ }
+
@Override
public void saveFromVideo(List<VideoAlarmReport> videoAlarmReports) {
for (VideoAlarmReport videoAlarmReport : videoAlarmReports) {
VideoPoint videoPoint = null;
- LambdaQueryWrapper<VideoPoint> queryWrapper = new LambdaQueryWrapper<VideoPoint>().eq(VideoPoint::getPlatResourceId, videoAlarmReport.getPlatResourceId());
+ LambdaQueryWrapper<VideoPoint> queryWrapper = new LambdaQueryWrapper<VideoPoint>()
+ .eq(VideoPoint::getPlatResourceId, videoAlarmReport.getPlatResourceId())
+ .or().eq(VideoPoint::getCode, videoAlarmReport.getPlatResourceId());
List<VideoPoint> pointList = videoPointService.list(queryWrapper);
if (pointList.size() > 0) {
videoPoint = pointList.get(0);
@@ -67,12 +78,19 @@
baseCase.setCommunityId(videoPoint.getCommunityId());
baseCase.setSite(videoPoint.getAddress());
baseCase.setGridId(videoPoint.getGridId());
+ baseCase.setSite(videoPoint.getAddress());
baseCaseService.save(baseCase);
violations.setId(baseCase.getId());
violations.setVideoAlarmReportId(videoAlarmReport.getId());
violations.setVideoPointId(videoPoint.getId());
violations.setDescription(videoAlarmReport.getAlgoName());
+
+ List<DataDictionary> list = dataDictionaryService.queryByCode(DictTypeEnum.EVENT_GRADE_TYPE.getCode(), videoAlarmReport.getAlarmName().trim());
+ if (list.size() > 0) {
+ violations.setGradeId(list.get(0).getId());
+ }
+
baseMapper.insert(violations);
}
}
--
Gitblit v1.8.0