ArticleSensitive jpa

long-article-recommend-service/src/main/java/com/tzld/longarticle/recommend/server/model/remote/ArticleSensitive.java

@@ -1,13 +0,0 @@
-package com.tzld.longarticle.recommend.server.model.remote;
-
-import lombok.Data;
-
-@Data
-public class ArticleSensitive {
-    private String md5;
-    private String title;
-    private String sensitive;
-    private String label;
-    private String subLabel;
-    private String response;
-}

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

@@ -7,8 +7,8 @@ import com.tencentcloudapi.common.profile.HttpProfile;
 import com.tencentcloudapi.tms.v20201229.TmsClient;
 import com.tencentcloudapi.tms.v20201229.models.TextModerationRequest;
 import com.tencentcloudapi.tms.v20201229.models.TextModerationResponse;
-import com.tzld.longarticle.recommend.server.model.remote.ArticleSensitive;
-import com.tzld.longarticle.recommend.server.repository.mapper.crawler.CrawlerBaseMapper;
+import com.tzld.longarticle.recommend.server.repository.crawler.ArticleSensitiveRepository;
+import com.tzld.longarticle.recommend.server.repository.entity.crawler.ArticleSensitive;
 import com.tzld.longarticle.recommend.server.util.Md5Util;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -23,7 +23,7 @@ import java.util.Objects;
 public class ArticleSensitiveRemoteService {
 
     @Autowired
-    CrawlerBaseMapper crawlerBaseMapper;
+    ArticleSensitiveRepository articleSensitiveRepository;
 
     @Value("${tecent.cloud.secret.id:AKIDbLGZox42mZaloeIo16zeJrDRUelkbFfw}")
     private String tecentCloudSecretId;
@@ -77,12 +77,13 @@ public class ArticleSensitiveRemoteService {
     }
 
     public ArticleSensitive getArticleSensitive(String md5) {
-        return crawlerBaseMapper.getArticleSensitiveByKey(md5);
+        return articleSensitiveRepository.getOne(md5);
     }
 
     public void insertArticleSensitive(String md5, String title, String sensitive, String label, String subLabel, String response) {
         // 插入
-        crawlerBaseMapper.insertArticleSensitive(md5, title, sensitive, label, subLabel, response);
+        ArticleSensitive insert = new ArticleSensitive(md5, title, sensitive, label, subLabel, response);
+        articleSensitiveRepository.save(insert);
     }
 
 

long-article-recommend-service/src/main/java/com/tzld/longarticle/recommend/server/repository/crawler/ArticleSensitiveRepository.java

@@ -0,0 +1,10 @@
+package com.tzld.longarticle.recommend.server.repository.crawler;
+
+import com.tzld.longarticle.recommend.server.repository.entity.crawler.ArticleSensitive;
+import org.springframework.data.jpa.repository.JpaRepository;
+import org.springframework.stereotype.Repository;
+
+@Repository
+public interface ArticleSensitiveRepository extends JpaRepository<ArticleSensitive, String> {
+
+}

long-article-recommend-service/src/main/java/com/tzld/longarticle/recommend/server/repository/entity/crawler/ArticleSensitive.java

@@ -0,0 +1,32 @@
+package com.tzld.longarticle.recommend.server.repository.entity.crawler;
+
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+
+import javax.persistence.Column;
+import javax.persistence.Entity;
+import javax.persistence.Id;
+import javax.persistence.Table;
+import java.io.Serializable;
+
+@Data
+@AllArgsConstructor
+@NoArgsConstructor
+@Entity
+@Table(name = "article_sensitive")
+public class ArticleSensitive implements Serializable {
+    @Id
+    private String md5;
+    @Column(name = "title")
+    private String title;
+    @Column(name = "sensitive")
+    private String sensitive;
+    @Column(name = "label")
+    private String label;
+    @Column(name = "sub_label")
+    private String subLabel;
+    @Column(name = "response")
+    private String response;
+
+}

long-article-recommend-service/src/main/java/com/tzld/longarticle/recommend/server/repository/mapper/crawler/CrawlerBaseMapper.java

@@ -1,7 +1,7 @@
 package com.tzld.longarticle.recommend.server.repository.mapper.crawler;
 
 import com.tzld.longarticle.recommend.server.model.remote.Article;
-import com.tzld.longarticle.recommend.server.model.remote.ArticleSensitive;
+import com.tzld.longarticle.recommend.server.repository.entity.crawler.ArticleSensitive;
 import com.tzld.longarticle.recommend.server.service.recall.ContentCategory;
 import org.apache.ibatis.annotations.Param;
 

long-article-recommend-service/src/main/resources/mapper/crawler/CrawlerBaseMapper.xml

@@ -9,7 +9,7 @@
 
 
     <select id="getArticleSensitiveByKey"
-            resultType="com.tzld.longarticle.recommend.server.model.remote.ArticleSensitive">
+            resultType="com.tzld.longarticle.recommend.server.repository.entity.crawler.ArticleSensitive">
         select md5, title, `sensitive`, label, sub_label, response from article_sensitive where md5 = #{md5}
     </select>