|
@@ -71,23 +71,22 @@ public class ScreenshotServiceImpl extends ServiceImpl<ScreenshotMapper, Screens
|
|
|
List<String> textContents = (List<String>) picResultMap.get("wordsList");
|
|
|
screenshot.setPicContext((String) picResultMap.get("picContent"));
|
|
|
Set<Object> members = redisUtil.members(Constant.COMMON_SOFTWARE_KEYWORDS);
|
|
|
+ List<String> exeprocessList = ProcessUtil.getExeprocessListfromProcessStr(screenshotvo.getProcessList());
|
|
|
for (String textContent : textContents) {
|
|
|
for (Object member : members) {
|
|
|
//由于redis里存储的member是一个PicContentKeywords类型的json字符串,所以取出关键字内容比较
|
|
|
JSONObject jsonMember = JSON.parseObject((String) member);
|
|
|
- if (textContent.contains(jsonMember.getString("content"))) {
|
|
|
+ if (textContent.toLowerCase().contains(jsonMember.getString("content").toLowerCase())) {
|
|
|
log.info("图片文字中包含了关键词--->"+jsonMember.getString("content"));
|
|
|
log.info("图片文字识别出的类型--->"+jsonMember.getInteger("type"));
|
|
|
- //确定图片是哪个类型的图片
|
|
|
- screenshot.setPicType(jsonMember.getInteger("type"));
|
|
|
+ //找到对应关键字,确定图片是哪个类型的图片
|
|
|
+ //结合进程判断
|
|
|
+ if(exeprocessList.contains(jsonMember.getString("processName").toLowerCase())){
|
|
|
+ log.info("图片中关键字对应的进程名--->"+jsonMember.getString("processName"));
|
|
|
+ screenshot.setPicType(jsonMember.getInteger("type"));
|
|
|
+ }
|
|
|
}
|
|
|
}
|
|
|
- }
|
|
|
- //结合进程判断
|
|
|
- List<String> exeprocessList = ProcessUtil.getExeprocessListfromProcessStr(screenshotvo.getProcessList());
|
|
|
- for (String process : exeprocessList) {
|
|
|
-
|
|
|
-
|
|
|
}
|
|
|
screenshot.setIsHandle(1);
|
|
|
screenshotMapper.insert(screenshot);
|