2023年04月27 测试环境临时代码

middleware-automatic-center-client/src/main/java/com/zhiwei/middleware/automatic/server/pojo/TemplateTitleVo.java

@@ -24,7 +24,7 @@ public class TemplateTitleVo implements Serializable {
         return id;
     }
 
-    public void setId(String group) {
+    public void buildId(String group) {
         MessageDigest sMd5Digest = null;
         try {
             sMd5Digest = MessageDigest.getInstance("MD5");
@@ -37,6 +37,10 @@ public class TemplateTitleVo implements Serializable {
         this.id = numValue.toString(16);
     }
 
+    public void setId(String id) {
+        this.id = id;
+    }
+
     public String getTemplateTitle() {
         return templateTitle;
     }
@@ -104,6 +108,8 @@ public class TemplateTitleVo implements Serializable {
         this.status = TemplateStatus.运行中;
     }
 
+    public TemplateTitleVo() {}
+
     public void refreshMark() {
         this.updateTime = new Date();
     }

middleware-automatic-center-client/src/main/java/com/zhiwei/middleware/automatic/server/pojo/enums/TaskType.java

 package com.zhiwei.middleware.automatic.server.pojo.enums;
 
 public enum TaskType {
-    COMMON_ONE("common_one","common", "commonCache"),
-    COMMON_TWO("common_two","common", "commonCache"),
-    TEMPLATE("template", "template", null),
-    TEMPLATE_MODIFY("template_modify","template", null),
-    TEMPLATE_RESET("template_reset","template", null);
+    COMMON_ONE("common_one", "commonCache"),
+    COMMON_TWO("common_two", "commonCache"),
+    TEMPLATE("template", null),
+    TEMPLATE_MODIFY("template_modify", null),
+    TEMPLATE_RESET("template_reset", null),
+    TEMPLATE_RECORD("template_record", null);
 
     final String type;
-    final String name;
 
     final String cacheId;
 
-    TaskType(String type, String name, String cacheId) {
+    TaskType(String type, String cacheId) {
         this.type = type;
-        this.name = name;
         this.cacheId = cacheId;
     }
 
-    public String getName() {
-        return this.name;
-    }
-
     public String getType() {
         return this.type;
     }

middleware-automatic-center-server/src/main/java/com/zhiwei/middleware/automatic/server/mission/ScheduledMission.java

@@ -20,6 +20,8 @@ public class ScheduledMission {
 
     private final Logger log = LogManager.getLogger(ScheduledMission.class);
 
+    private static final Long MONTH = 1000L * 60 * 60 * 24 * 30;
+
     private final RedissonUtil redissonUtil;
 
     private final AsyncTask asyncTask;
@@ -60,6 +62,9 @@ public class ScheduledMission {
             for (String project : asyncTask.findAllGroup()) {
                 putTask(project, startTime, endTime);
             }
+
+            // 模板记录清除任务
+            putTaskByRecord();
         } catch (Exception e) {
             log.error("每天定时同步模板失败：", e);
         }
@@ -73,4 +78,10 @@ public class ScheduledMission {
         redissonUtil.putQueue(GenericAttribute.KEY, JSONObject.toJSONString(autoTask));
     }
 
+    private void putTaskByRecord() {
+        AutoTask autoTask = new AutoTask(TaskType.TEMPLATE_RECORD.getType());
+        autoTask.getParamSource().put(GenericAttribute.END_PARAM, System.currentTimeMillis() - MONTH);
+        redissonUtil.putQueue(GenericAttribute.KEY, JSONObject.toJSONString(autoTask));
+    }
+
 }

middleware-automatic-center-son/src/main/java/com/zhiwei/middleware/automatic/son/dao/TemplateRecordDao.java

@@ -33,4 +33,6 @@ public interface TemplateRecordDao {
      * @param query 条件
      */
     void removeTemplateRecord (Query query);
+
+    void tempRecord(TemplateTempRecord templateTempRecord);
 }

middleware-automatic-center-son/src/main/java/com/zhiwei/middleware/automatic/son/dao/TemplateTempRecord.java

+package com.zhiwei.middleware.automatic.son.dao;
+
+import org.springframework.data.mongodb.core.mapping.Document;
+
+@Document("automaticmark_template_record_temp")
+public class TemplateTempRecord {
+
+    private String id;
+
+    private String templateId;
+
+    private String templateTitle;
+
+    private String group;
+
+    private String url;
+
+    private String mtag;
+
+    public TemplateTempRecord() {}
+
+    public TemplateTempRecord(String id, String templateId, String templateTitle,
+                              String group, String url, String mtag) {
+        this.id = id;
+        this.templateId = templateId;
+        this.templateTitle = templateTitle;
+        this.group = group;
+        this.url = url;
+        this.mtag = mtag;
+    }
+
+    public String getId() {
+        return id;
+    }
+
+    public String getTemplateId() {
+        return templateId;
+    }
+
+    public String getTemplateTitle() {
+        return templateTitle;
+    }
+
+    public String getGroup() {
+        return group;
+    }
+
+    public String getUrl() {
+        return url;
+    }
+
+    public String getMtag() {
+        return mtag;
+    }
+}

middleware-automatic-center-son/src/main/java/com/zhiwei/middleware/automatic/son/dao/impl/TemplateRecordDaoImpl.java

@@ -2,6 +2,7 @@ package com.zhiwei.middleware.automatic.son.dao.impl;
 
 import com.zhiwei.middleware.automatic.server.pojo.TemplateRecord;
 import com.zhiwei.middleware.automatic.son.dao.TemplateRecordDao;
+import com.zhiwei.middleware.automatic.son.dao.TemplateTempRecord;
 import org.springframework.beans.factory.annotation.Qualifier;
 import org.springframework.data.mongodb.core.MongoTemplate;
 import org.springframework.data.mongodb.core.query.Query;
@@ -37,4 +38,9 @@ public class TemplateRecordDaoImpl implements TemplateRecordDao {
     public void removeTemplateRecord(Query query) {
         mongoTemplate.remove(query, TemplateRecord.class, "automaticmark_template_record_new");
     }
+
+    @Override
+    public void tempRecord(TemplateTempRecord templateTempRecord) {
+        mongoTemplate.insert(templateTempRecord);
+    }
 }

middleware-automatic-center-son/src/main/java/com/zhiwei/middleware/automatic/son/service/TemplateTitleService.java

@@ -2,6 +2,7 @@ package com.zhiwei.middleware.automatic.son.service;
 
 import com.zhiwei.middleware.automatic.server.pojo.TemplateRecord;
 import com.zhiwei.middleware.automatic.server.pojo.TemplateTitleVo;
+import com.zhiwei.middleware.automatic.son.dao.TemplateTempRecord;
 
 import java.util.List;
 import java.util.Map;
@@ -45,5 +46,4 @@ public interface TemplateTitleService {
      * @return 是否成功
      */
     boolean resetTemplate (String group, String templateTitle);
-
 }

middleware-automatic-center-son/src/main/java/com/zhiwei/middleware/automatic/son/service/impl/TemplateTitleServiceImpl.java

@@ -15,6 +15,7 @@ import com.zhiwei.middleware.automatic.son.service.TemplateTitleService;
 import com.zhiwei.middleware.automatic.son.util.MarkInfoUtil;
 import com.zhiwei.middleware.automatic.son.util.RedissonUtil;
 import com.zhiwei.middleware.automatic.son.util.Tools;
+import org.apache.commons.lang3.time.FastDateFormat;
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
 import org.elasticsearch.index.query.BoolQueryBuilder;
@@ -46,6 +47,9 @@ public class TemplateTitleServiceImpl implements TemplateTitleService {
 
     private final ThreadPoolTaskExecutor executor;
 
+    private static final FastDateFormat format = FastDateFormat.getInstance("yyyy-MM-dd");
+    private static final String AGGREGATION_FILE = "aggregation-";
+
     public TemplateTitleServiceImpl(RedissonUtil redissonUtil, EsDao esDao,
                                 IndexUtil.ESIndexes esIndexes, TemplateRecordDao templateRecordDao,
                                 DubboHandler dubboHandler,
@@ -61,7 +65,8 @@ public class TemplateTitleServiceImpl implements TemplateTitleService {
 
     @Override
     public Map<String, TemplateTitleVo> getTemplateTitleByProject(String project) {
-        Map<String, String> mapValue = redissonUtil.getMapValue(Tools.assembleKey(GenericAttribute.REDIS_MAP_KEY, project));
+//        Map<String, String> mapValue = redissonUtil.getMapValue(Tools.assembleKey(GenericAttribute.REDIS_MAP_KEY, project));
+        Map<String, String> mapValue = redissonUtil.getMapValue(Tools.assembleKey(getFileDayName(project)));
         if (Tools.isEmpty(mapValue)) {
             return new HashMap<>();
         }
@@ -185,4 +190,8 @@ public class TemplateTitleServiceImpl implements TemplateTitleService {
         query.must(should);
         return esDao.search(indexes, null, query, null, 0, 1000, null);
     }
+
+    private static String getFileDayName(String group) {
+        return AGGREGATION_FILE + format.format(new Date()) + ":" + group;
+    }
 }

middleware-automatic-center-son/src/main/java/com/zhiwei/middleware/automatic/son/task/service/TaskServiceCommon.java

@@ -10,6 +10,8 @@ import com.zhiwei.middleware.automatic.server.pojo.TemplateRecord;
 import com.zhiwei.middleware.automatic.server.pojo.TemplateTitleVo;
 import com.zhiwei.middleware.automatic.server.pojo.enums.TaskType;
 import com.zhiwei.middleware.automatic.server.pojo.enums.TemplateStatus;
+import com.zhiwei.middleware.automatic.son.dao.TemplateRecordDao;
+import com.zhiwei.middleware.automatic.son.dao.TemplateTempRecord;
 import com.zhiwei.middleware.automatic.son.dubbo.DubboHandler;
 import com.zhiwei.middleware.automatic.son.service.TemplateTitleService;
 import com.zhiwei.middleware.automatic.son.util.CosineSimilarity;
@@ -39,16 +41,20 @@ public class TaskServiceCommon implements TaskService {
 
     private final DubboHandler dubboHandler;
 
+    private final TemplateRecordDao templateRecordDao;
+
     private final ThreadPoolTaskExecutor autoMarkExecutor;
 
     private final static String COUNT_KEY = "count";
 
     public TaskServiceCommon(RedissonUtil redissonUtil, TemplateTitleService templateTitleService,
                              DubboHandler dubboHandler,
+                             TemplateRecordDao templateRecordDao,
                              @Qualifier("autMarkExecutor") ThreadPoolTaskExecutor autoMarkExecutor) {
         this.redissonUtil = redissonUtil;
         this.templateTitleService = templateTitleService;
         this.dubboHandler = dubboHandler;
+        this.templateRecordDao = templateRecordDao;
         this.autoMarkExecutor = autoMarkExecutor;
         TYPE_MAP.put(TaskType.COMMON_TWO.getType(), Pair.of(TaskType.COMMON_TWO, this::getMultiAutoInfo));
         TYPE_MAP.put(TaskType.COMMON_ONE.getType(), Pair.of(TaskType.COMMON_ONE, this::getOneAutoInfo));
@@ -182,33 +188,41 @@ public class TaskServiceCommon implements TaskService {
         List<MarkInfo> newList = markInfos.stream().filter(markInfo -> {
             String title = markInfo.getSourceObj().getString(GenericAttribute.ES_TITLE);
             return null != title && title.length() > 6;
-        }).collect(Collectors.toList());
-        for (MarkInfo markInfo : newList) {
-            JSONObject sourceObj = markInfo.getSourceObj();
-            String title = Tools.filterSymbol(sourceObj.getString(GenericAttribute.ES_TITLE));
-            Map<String, Object> similarMap = similarMapInfo(titleVoMap, title, group);
-            if (!similarMap.isEmpty()) {
-                // 填充数据
-                String aggreTitle = String.valueOf(similarMap.get("aggreTitle"));
-                TemplateTitleVo templateTitleVo = titleVoMap.get(aggreTitle);
-                String aggreTag = templateTitleVo.getMtag();
-                sourceObj.put(GenericAttribute.ES_M_TAG, aggreTag);
-                sourceObj.put(GenericAttribute.ES_M_PERSON, "自动化机器人");
-                sourceObj.put(GenericAttribute.ES_M_TIME, new Date().getTime());
-                log.info("项目:{} 模板标题:{} MarkSum:{} Tag:{}被标注标题：{}相似度：{}", group, aggreTitle, templateTitleVo.getMarkSum(), aggreTag,
-                        title, similarMap.get("similar"));
+        }).filter(e -> autoMark(group, e, titleVoMap)).collect(Collectors.toList());
+//        dubboHandler.markUpsert(newList);
+    }
+
+    private boolean autoMark(String group, MarkInfo markInfo, Map<String, TemplateTitleVo> titleVoMap) {
+        JSONObject sourceObj = markInfo.getSourceObj();
+        String title = Tools.filterSymbol(sourceObj.getString(GenericAttribute.ES_TITLE));
+        Map<String, Object> similarMap = similarMapInfo(titleVoMap, title, group);
+        if (!similarMap.isEmpty()) {
+            // 填充数据
+            String aggreTitle = String.valueOf(similarMap.get("aggreTitle"));
+            TemplateTitleVo templateTitleVo = titleVoMap.get(aggreTitle);
+            String aggreTag = templateTitleVo.getMtag();
+            sourceObj.put(GenericAttribute.ES_M_TAG, aggreTag);
+            sourceObj.put(GenericAttribute.ES_M_PERSON, "自动化机器人");
+            sourceObj.put(GenericAttribute.ES_M_TIME, new Date().getTime());
+            log.info("项目:{} 模板标题:{} MarkSum:{} Tag:{}被标注标题：{}相似度：{}", group, aggreTitle, templateTitleVo.getMarkSum(), aggreTag,
+                    title, similarMap.get("similar"));
 
-                // 刷新一下标注量和标注时间,
-                try {
-                    String[] updates = dubboHandler.getMupdates(markInfo.filterInfo());
-                    templateTitleVo.refreshMark();
-                    templateTitleService.insertTemplateRecord(new TemplateRecord(templateTitleVo.getId(), updates[0]));
-                    redissonUtil.putCount(Tools.assembleKey(COUNT_KEY, group, templateTitleVo.getId()), 1);
-                } catch (Exception e) {
-                    log.error("记录事件采集-标注数据特征值失败", e);
-                }
+            // 刷新一下标注量和标注时间,
+            try {
+                String[] updates = dubboHandler.getMupdates(markInfo.filterInfo());
+                templateTitleVo.refreshMark();
+                // 特征值记录
+                templateTitleService.insertTemplateRecord(new TemplateRecord(templateTitleVo.getId(), updates[0]));
+                // 模板标注count累加
+                redissonUtil.putCount(Tools.assembleKey(COUNT_KEY, group, templateTitleVo.getId()), 1);
+                // 测试环境临时添加，用于对比
+                templateRecordDao.tempRecord(new TemplateTempRecord());
+                return true;
+            } catch (Exception e) {
+                log.error("记录事件采集-标注数据特征值失败", e);
             }
         }
+        return false;
     }
 
     private Map<String, Object> similarMapInfo(Map<String, TemplateTitleVo> titleVoMap, String title, String group) {
@@ -216,7 +230,7 @@ public class TaskServiceCommon implements TaskService {
         Map<String, Object> similarMap = new HashMap<>();
         for (TemplateTitleVo templateTitleVo : titleVoMap.values()) {
             if (Objects.isNull(templateTitleVo.getId())) {
-                templateTitleVo.setId(group);
+                templateTitleVo.buildId(group);
             }
             // 过滤掉以重置的模板
             if (templateTitleVo.getStatus() == TemplateStatus.已重置 || Tools.isEmpty(templateTitleVo.getMtag())) {

middleware-automatic-center-son/src/main/java/com/zhiwei/middleware/automatic/son/task/service/TaskServiceTemplate.java

@@ -10,6 +10,7 @@ import com.zhiwei.middleware.automatic.server.pojo.TemplateTitleVo;
 import com.zhiwei.middleware.automatic.server.pojo.enums.TaskType;
 import com.zhiwei.middleware.automatic.server.pojo.enums.TemplateStatus;
 import com.zhiwei.middleware.automatic.son.dao.EsDao;
+import com.zhiwei.middleware.automatic.son.dao.TemplateRecordDao;
 import com.zhiwei.middleware.automatic.son.service.TemplateTitleService;
 import com.zhiwei.middleware.automatic.son.util.CosineSimilarity;
 import com.zhiwei.middleware.automatic.son.util.Tools;
@@ -24,6 +25,8 @@ import org.elasticsearch.index.query.QueryBuilders;
 import org.elasticsearch.search.SearchHit;
 import org.elasticsearch.search.builder.SearchSourceBuilder;
 import org.springframework.beans.factory.annotation.Qualifier;
+import org.springframework.data.mongodb.core.query.Criteria;
+import org.springframework.data.mongodb.core.query.Query;
 import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
 import org.springframework.stereotype.Service;
 
@@ -32,7 +35,7 @@ import java.util.*;
 import java.util.concurrent.ConcurrentHashMap;
 import java.util.stream.Collectors;
 
-@Service
+//@Service
 public class TaskServiceTemplate implements TaskService {
 
     private final Logger log = LogManager.getLogger(TaskServiceTemplate.class);
@@ -45,6 +48,8 @@ public class TaskServiceTemplate implements TaskService {
 
     private final IndexUtil.ESIndexes esIndexes;
 
+    private final TemplateRecordDao templateRecordDao;
+
     private final ThreadPoolTaskExecutor executor;
 
     /* 一天的秒数(为保留前一天文件) */
@@ -52,14 +57,17 @@ public class TaskServiceTemplate implements TaskService {
 
     public TaskServiceTemplate(TemplateTitleService templateTitleService,
                                EsDao esDao, IndexUtil.ESIndexes esIndexes,
+                               TemplateRecordDao templateRecordDao,
                                @Qualifier("templateExecutor") ThreadPoolTaskExecutor executor) {
         this.templateTitleService = templateTitleService;
         this.esDao = esDao;
         this.esIndexes = esIndexes;
+        this.templateRecordDao = templateRecordDao;
         this.executor = executor;
         TYPE_MAP.put(TaskType.TEMPLATE.getType(), Pair.of(TaskType.TEMPLATE, this::runTask));
         TYPE_MAP.put(TaskType.TEMPLATE_MODIFY.getType(), Pair.of(TaskType.TEMPLATE_MODIFY, this::templateModify));
         TYPE_MAP.put(TaskType.TEMPLATE_RESET.getType(), Pair.of(TaskType.TEMPLATE_RESET, this::templateReset));
+        TYPE_MAP.put(TaskType.TEMPLATE_RECORD.getType(), Pair.of(TaskType.TEMPLATE_RECORD, this::templateRecordClean));
     }
 
     @Override
@@ -82,6 +90,10 @@ public class TaskServiceTemplate implements TaskService {
         return false;
     }
 
+    private void templateRecordClean(JSONObject json) {
+        templateRecordDao.removeTemplateRecord(new Query(Criteria.where("createAt").lt(json.getLong(GenericAttribute.END_PARAM))));
+    }
+
     /**
      * 模板修改任务
      * @param json 参数
@@ -163,7 +175,7 @@ public class TaskServiceTemplate implements TaskService {
                     String title = e.getKey();
                     TemplateTitleVo templateTitleVo = e.getValue();
                     if (Objects.isNull(templateTitleVo.getId())) {
-                        templateTitleVo.setId(group);
+                        templateTitleVo.buildId(group);
                     }
                     long updateTime = templateTitleVo.getUpdateTime().getTime();
                     // 移除7天有效期外的数据

middleware-automatic-center-son/src/main/java/com/zhiwei/middleware/automatic/son/util/RedissonUtil.java

@@ -36,7 +36,8 @@ public class RedissonUtil {
     }
 
     public Map<String, String> getMapValue(String key) {
-        return redissonClient.getMap(redisKey(key));
+//        return redissonClient.getMap(redisKey(key));
+        return redissonClient.getMap(key);
     }
 
     public String getMapKeyValue(String key, String group) {

middleware-automatic-center-son/src/main/resources/redisson-local.yaml

@@ -6,14 +6,14 @@ singleServerConfig:
   retryAttempts: 3
   retryInterval: 1500
   subscriptionsPerConnection: 5
-  address: "redis://192.168.0.225:6379"
+  address: "redis://115.236.59.91:7386"
   subscriptionConnectionMinimumIdleSize: 1
   subscriptionConnectionPoolSize: 50
   connectionMinimumIdleSize: 32
   connectionPoolSize: 64
-  database: 4
+  database: 3
   dnsMonitoringInterval: 5000
 threads: 0
 nettyThreads: 0
-codec: !<org.redisson.codec.JsonJacksonCodec> {}
+codec: !<org.redisson.client.codec.StringCodec> {}
 transportMode: "NIO"
\ No newline at end of file

middleware-automatic-center-son/src/main/resources/redisson.yaml

@@ -15,5 +15,5 @@ singleServerConfig:
   dnsMonitoringInterval: 5000
 threads: 0
 nettyThreads: 0
-codec: !<org.redisson.codec.JsonJacksonCodec> {}
+codec: !<org.redisson.client.codec.StringCodec> {}
 transportMode: "NIO"
\ No newline at end of file