Bladeren bron

视频封面审核

wangyunpeng 5 maanden geleden
bovenliggende
commit
481f9670dd

+ 18 - 3
long-article-recommend-service/src/main/java/com/tzld/longarticle/recommend/server/model/entity/longArticle/LongArticleVideoAudit.java

@@ -5,17 +5,22 @@ import lombok.Data;
 import lombok.NoArgsConstructor;
 
 import javax.persistence.*;
+import java.io.Serializable;
 
 @Data
 @AllArgsConstructor
 @NoArgsConstructor
 @Entity
 @Table(name = "long_articles_video_audit")
+@IdClass(LongArticleVideoAudit.PK.class)
 public class LongArticleVideoAudit {
 
     @Id
     @Column(name = "video_id")
     private Long videoId;
+    @Id
+    @Column(name = "type")
+    private Integer type;
 
     @Column(name = "content_id")
     private String contentId;
@@ -23,9 +28,6 @@ public class LongArticleVideoAudit {
     @Column(name = "trace_id")
     private String traceId;
 
-    @Column(name = "type")
-    private Integer type;
-
     @Column(name = "oss_path")
     private String ossPath;
 
@@ -43,4 +45,17 @@ public class LongArticleVideoAudit {
 
     @Column(name = "finish_timestamp")
     private Long finishTimestamp;
+
+    @Data
+    public static class PK implements Serializable {
+
+        @Column(name = "video_id")
+        private Long videoId;
+        @Column(name = "type")
+        private Integer type;
+
+        public PK() {
+        }
+
+    }
 }

+ 1 - 1
long-article-recommend-service/src/main/java/com/tzld/longarticle/recommend/server/remote/alg/DangerFaceRecognizeService.java

@@ -58,7 +58,7 @@ public class DangerFaceRecognizeService {
                     String responseBody = EntityUtils.toString(responseEntity, "UTF-8");
                     PQResponse pqResponse = JSONObject.parseObject(responseBody, PQResponse.class);
                     if (pqResponse.getCode() == 0 && Objects.nonNull(pqResponse.getData())) {
-                        return (AlgFaceRecognizeResult) pqResponse.getData();
+                        return JSONObject.parseObject(JSONObject.toJSONString(pqResponse.getData()), AlgFaceRecognizeResult.class);
                     } else {
                         return null;
                     }

+ 1 - 1
long-article-recommend-service/src/main/java/com/tzld/longarticle/recommend/server/repository/longArticle/LongArticleVideoAuditRepository.java

@@ -7,7 +7,7 @@ import org.springframework.stereotype.Repository;
 import java.util.List;
 
 @Repository
-public interface LongArticleVideoAuditRepository extends JpaRepository<LongArticleVideoAudit, Long> {
+public interface LongArticleVideoAuditRepository extends JpaRepository<LongArticleVideoAudit, LongArticleVideoAudit.PK> {
     LongArticleVideoAudit getByVideoIdAndType(Long videoId, Integer type);
 
     List<LongArticleVideoAudit> getByVideoIdInAndType(List<Long> videoIds, Integer type);