票圈视频审核

long-article-recommend-service/src/main/java/com/tzld/longarticle/recommend/server/common/enums/cgi/PQVideoSensitiveLevelEnum.java

@@ -0,0 +1,34 @@
+package com.tzld.longarticle.recommend.server.common.enums.cgi;
+
+import lombok.Getter;
+
+import java.util.Objects;
+
+@Getter
+public enum PQVideoSensitiveLevelEnum {
+    NOT_CHECK(-1, "未检测"),
+    NORMAL(0, "正常"),
+    LOW_SENSITIVE(1, "初级敏感"),
+    MID_SENSITIVE(2, "中级敏感"),
+    HIGH_SENSITIVE(3, "严重敏感"),
+
+    other(999, "其他"),
+    ;
+    // -1 未检测 0 正常 1 初级敏感 2中级敏感 3 严重敏感
+    private final Integer level;
+    private final String description;
+
+    PQVideoSensitiveLevelEnum(Integer level, String description) {
+        this.level = level;
+        this.description = description;
+    }
+
+    public static PQVideoSensitiveLevelEnum from(Integer level) {
+        for (PQVideoSensitiveLevelEnum levelEnum : PQVideoSensitiveLevelEnum.values()) {
+            if (Objects.equals(levelEnum.level, level)) {
+                return levelEnum;
+            }
+        }
+        return other;
+    }
+}

long-article-recommend-service/src/main/java/com/tzld/longarticle/recommend/server/service/recommend/ArticleAuditService.java

@@ -6,6 +6,7 @@ import com.google.common.collect.Lists;
 import com.tzld.longarticle.recommend.server.common.enums.StatusEnum;
 import com.tzld.longarticle.recommend.server.common.enums.aigc.PushTypeEnum;
 import com.tzld.longarticle.recommend.server.common.enums.cgi.PQVideoAuditResultEnum;
+import com.tzld.longarticle.recommend.server.common.enums.cgi.PQVideoSensitiveLevelEnum;
 import com.tzld.longarticle.recommend.server.common.enums.recommend.ArticleDeleteStatusEnum;
 import com.tzld.longarticle.recommend.server.common.enums.recommend.FeishuRobotIdEnum;
 import com.tzld.longarticle.recommend.server.mapper.aigc.AigcBaseMapper;
@@ -128,21 +129,24 @@ public class ArticleAuditService {
             List<Long> taskIds = partition.stream().map(LongArticleVideoAudit::getTaskId).collect(Collectors.toList());
             Map<Long, LongArticleVideoAudit> map = partition.stream().collect(Collectors.toMap(LongArticleVideoAudit::getTaskId, Function.identity()));
             List<PQVideoAuditResult> resultList = pqVideoAuditResultService.getResult(taskIds);
-            resultList.forEach(result -> saveVideoAuditResult(map.get(result.getTaskId()), result.getAuditResult()));
+            resultList.forEach(result -> saveVideoAuditResult(map.get(result.getTaskId()), result));
         }
         return ReturnT.SUCCESS;
     }
 
 
-    public void saveVideoAuditResult(LongArticleVideoAudit longArticleVideoAudit, Integer auditResult) {
-        if (Objects.equals(auditResult, PQVideoAuditResultEnum.PASS.getStatus())) {
+    public void saveVideoAuditResult(LongArticleVideoAudit longArticleVideoAudit, PQVideoAuditResult result) {
+
+        if (Objects.equals(result.getAuditResult(), PQVideoAuditResultEnum.PASS.getStatus())) {
             // 审核通过，更新文章状态
             longArticleVideoAudit.setStatus(PQVideoAuditResultEnum.PASS.getStatus());
             longArticleVideoAudit.setFinishTimestamp(System.currentTimeMillis());
             longArticleVideoAuditRepository.save(longArticleVideoAudit);
-        } else if (Objects.equals(auditResult, PQVideoAuditResultEnum.REJECT.getStatus())) {
+        } else if (Objects.equals(result.getAuditResult(), PQVideoAuditResultEnum.REJECT.getStatus())) {
             // 审核不通过，删除文章
             longArticleVideoAudit.setStatus(PQVideoAuditResultEnum.REJECT.getStatus());
+            PQVideoSensitiveLevelEnum sensitiveLevelEnum = PQVideoSensitiveLevelEnum.from(result.getSensitiveLevel());
+            longArticleVideoAudit.setFailReason(sensitiveLevelEnum.getDescription());
             longArticleVideoAudit.setFinishTimestamp(System.currentTimeMillis());
             longArticleVideoAuditRepository.save(longArticleVideoAudit);
             // 构建删除文章记录 并保存