Selaa lähdekoodia

screen prompt

wangyunpeng 3 viikkoa sitten
vanhempi
commit
f5ae972d97

+ 2 - 0
core/src/main/java/com/tzld/supply/dao/mapper/supply/spider/ext/SpiderMapperExt.java

@@ -11,4 +11,6 @@ public interface SpiderMapperExt {
     List<SpiderContent> getRoughScreenSpiderContentHasContent();
 
     List<SpiderContent> getRoughScreenSpiderContentNoContent();
+
+    List<SpiderContent> getExportContentList(List<Integer> statusList, Integer aiRoughStatus, Long startTime, Long endTime);
 }

+ 2 - 2
core/src/main/java/com/tzld/supply/job/ContentScreenJob.java

@@ -112,8 +112,8 @@ public class ContentScreenJob {
             "\n" +
             "请严格按以下 JSON 格式输出,不要输出任何说明或注释:\n" +
             "{\n" +
-            "\"is_suitable\": true/false,\n" +
-            "\"reason\": \"简要说明判断依据\"\n" +
+            "    \"is_suitable\": true/false,\n" +
+            "    \"reason\": \"简要说明判断依据\"\n" +
             "}\n" +
             "\n" +
             "输入内容:\n" +

+ 7 - 6
core/src/main/java/com/tzld/supply/job/SpiderContentFeishuJob.java

@@ -3,12 +3,13 @@ package com.tzld.supply.job;
 import cn.hutool.core.collection.CollectionUtil;
 import com.alibaba.fastjson.JSONObject;
 import com.google.common.collect.Lists;
+import com.tzld.supply.common.enums.SpiderContentScreenStatusEnum;
 import com.tzld.supply.common.enums.SpiderContentStatusEnum;
 import com.tzld.supply.dao.mapper.supply.spider.SpiderContentMapper;
+import com.tzld.supply.dao.mapper.supply.spider.ext.SpiderMapperExt;
 import com.tzld.supply.model.entity.PrecisionScreenEntity;
 import com.tzld.supply.model.entity.feishu.SpiderContentEntity;
 import com.tzld.supply.model.po.supply.spider.SpiderContent;
-import com.tzld.supply.model.po.supply.spider.SpiderContentExample;
 import com.tzld.supply.util.DateUtils;
 import com.tzld.supply.util.feishu.FeiShu;
 import com.tzld.supply.util.feishu.FeishuExcelUtil;
@@ -33,6 +34,8 @@ public class SpiderContentFeishuJob {
 
     @Autowired
     private SpiderContentMapper spiderContentMapper;
+    @Autowired
+    private SpiderMapperExt spiderMapperExt;
 
 
     @XxlJob("spiderContentFeishuJob")
@@ -42,11 +45,9 @@ public class SpiderContentFeishuJob {
             startTime = DateUtils.getStartOfDay(param, "yyyyMMdd");
         }
         Long endTime = startTime + 86400 * 1000;
-        SpiderContentExample example = new SpiderContentExample();
-        example.createCriteria().andStatusIn(Lists.newArrayList(2, 3))
-                .andAiRoughStatusEqualTo(1)
-                .andCreateTimeBetween(startTime, endTime);
-        List<SpiderContent> dataList = spiderContentMapper.selectByExampleWithBLOBs(example);
+        List<SpiderContent> dataList = spiderMapperExt.getExportContentList(
+                Lists.newArrayList(SpiderContentStatusEnum.PASSED.getCode(), SpiderContentStatusEnum.ABANDONED.getCode()),
+                SpiderContentScreenStatusEnum.PASSED.getCode(), startTime, endTime);
         if (CollectionUtil.isEmpty(dataList)) {
             return ReturnT.SUCCESS;
         }

+ 12 - 0
core/src/main/resources/mapper/supply/spider/ext/SpiderMapperExt.xml

@@ -23,4 +23,16 @@
         where `status` = 1 and ai_rough_status = 1 and ai_precision_status = 0 and content is null
     </select>
 
+    <select id="getExportContentList" resultType="com.tzld.supply.model.po.supply.spider.SpiderContent">
+        select *
+        from spider_content
+        where `status` in
+        <foreach collection="statusList" item="item" open="(" close=")" separator=",">
+            #{item}
+        </foreach>
+        and ai_rough_status = #{aiRoughStatus}
+        and create_time between #{startTime} and #{endTime}
+        and content is not null
+    </select>
+
 </mapper>