Parcourir la source

标题相似临时过滤

wangyunpeng il y a 6 mois
Parent
commit
b126fcd5e4

+ 30 - 0
long-article-recommend-service/src/main/java/com/tzld/longarticle/recommend/server/common/enums/LongArticleTextSimilarityStatusEnum.java

@@ -0,0 +1,30 @@
+package com.tzld.longarticle.recommend.server.common.enums;
+
+import lombok.Getter;
+
+@Getter
+public enum LongArticleTextSimilarityStatusEnum {
+
+    WAITING(0, "未判断"),
+    USEFUL(1, "可用"),
+    UNUSEFUL(2, "不可用"),
+    ;
+
+    //0-未判断 1-可用 2-不可用
+    private int code;
+    private String msg;
+
+    LongArticleTextSimilarityStatusEnum(int code, String msg) {
+        this.code = code;
+        this.msg = msg;
+    }
+
+    public static LongArticleTextSimilarityStatusEnum getByCode(int code) {
+        for (LongArticleTextSimilarityStatusEnum statusEnum : LongArticleTextSimilarityStatusEnum.values()) {
+            if (statusEnum.getCode() == code) {
+                return statusEnum;
+            }
+        }
+        return null;
+    }
+}

+ 2 - 1
long-article-recommend-service/src/main/java/com/tzld/longarticle/recommend/server/service/recommend/recall/RecallService.java

@@ -7,6 +7,7 @@ import com.google.common.cache.CacheBuilder;
 import com.google.common.collect.Lists;
 import com.tzld.longarticle.recommend.server.common.CostMonitor;
 import com.tzld.longarticle.recommend.server.common.ThreadPoolFactory;
+import com.tzld.longarticle.recommend.server.common.enums.LongArticleTextSimilarityStatusEnum;
 import com.tzld.longarticle.recommend.server.common.enums.StatusEnum;
 import com.tzld.longarticle.recommend.server.common.enums.aigc.PublishPlanInputSourceTypesEnum;
 import com.tzld.longarticle.recommend.server.common.enums.recommend.*;
@@ -219,7 +220,7 @@ public class RecallService implements ApplicationContextAware {
         }
         content.removeIf(o -> !Objects.equals(o.getSourceType(), PublishPlanInputSourceTypesEnum.longArticleVideoPoolSource.getVal())
                 && (Objects.isNull(textStatusMap.get(o.getSourceId()))
-                || textStatusMap.get(o.getSourceId()) != 1));
+                || textStatusMap.get(o.getSourceId()) != LongArticleTextSimilarityStatusEnum.USEFUL.getCode()));
     }
 
     private void setVideoContent(List<Content> contentList) {