backup

barrage-crawler-server/src/main/java/com/zhiweidata/automatictest/barragecrawlerserver/exporttasks/BiliBiliExportTaskJob.java

@@ -3,20 +3,25 @@ package com.zhiweidata.automatictest.barragecrawlerserver.exporttasks;
 import com.fasterxml.jackson.databind.ObjectMapper;
 import com.zhiweidata.automatictest.barragecrawlerserver.entity.BarrageCollectionTask;
 import com.zhiweidata.automatictest.barragecrawlerserver.entity.BarrageTaskResponse;
-import com.zhiweidata.automatictest.barragecrawlerserver.util.JsonPretty;
+import com.zhiweidata.automatictest.barragecrawlerserver.util.TimeUtil;
+import java.util.Date;
 import java.util.List;
 import java.util.Map;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.http.HttpResponse;
 import org.apache.http.client.methods.HttpGet;
 import org.apache.http.util.EntityUtils;
+import org.jetbrains.annotations.NotNull;
 import org.quartz.JobDataMap;
 import org.quartz.JobExecutionContext;
+import org.quartz.Trigger;
+import org.quartz.TriggerBuilder;
 
 import static com.zhiweidata.automatictest.barragecrawlerserver.config.BaseConfig.BARRAGE_COLLECTION_TASK_JOB_KEY;
 import static com.zhiweidata.automatictest.barragecrawlerserver.config.BaseConfig.BASE_URL;
-import static java.util.Objects.requireNonNull;
 import static com.zhiweidata.automatictest.publics.BeanContainer.HTTP_CLIENT;
+import static com.zhiweidata.automatictest.publics.BeanContainer.SCHEDULER;
+import static java.util.Objects.requireNonNull;
 
 /**
  * bilibili弹幕导出任务
@@ -55,9 +60,30 @@ public class BiliBiliExportTaskJob implements BarrageExportTaskJob {
             Map<?, ?> dataMap = (Map<?, ?>) requireNonNull(jsonMap.get("data"));
             List<?> dataList = (List<?>) requireNonNull(dataMap.get("list"));
             BarrageTaskResponse taskResponse = jsonMapper.convertValue(dataList.get(0), BarrageTaskResponse.class);
-            System.out.println(taskResponse);
+            switch (taskResponse.getStatus()) {
+                case ENDED:
+                    this.checkFile(taskResponse);
+                    break;
+                case FAIL:
+                    log.error("任务：{} 采集失败", taskResponse.getId());
+                    break;
+                case WAIT:
+                case COLLECTING:
+                    Date nextTime = TimeUtil.randmoDate(2);
+                    TriggerBuilder<Trigger> triggerBuilder = TriggerBuilder.newTrigger();
+                    triggerBuilder.startAt(nextTime);
+                    SCHEDULER.scheduleJob(context.getJobDetail(), triggerBuilder.build());
+                default:
+                    throw new RuntimeException("未知枚举：" + taskResponse.getStatus());
+            }
         } catch (Exception e) {
             log.error(e.getMessage(), e);
         }
     }
+
+    /**
+     * 下载excel文件，并检查excel文件
+     */
+    private void checkFile(@NotNull BarrageTaskResponse taskResponse) {
+    }
 }