backup

barrage-crawler-server/src/main/java/com/zhiweidata/automatictest/barragecrawlerserver/createtasks/BiliBiliCreateTaskJob.java

@@ -2,10 +2,13 @@ package com.zhiweidata.automatictest.barragecrawlerserver.createtasks;
 
 import com.fasterxml.jackson.databind.ObjectMapper;
 import java.net.URL;
+import java.util.List;
 import java.util.Map;
 import lombok.extern.slf4j.Slf4j;
 import org.quartz.JobExecutionContext;
 
+import static java.util.Objects.requireNonNull;
+
 /**
  * 从B站直播的首页获得直播间的地址，并发送到弹幕采集服务器进行测试
  *
@@ -22,6 +25,13 @@ public class BiliBiliCreateTaskJob extends CreateTaskJob {
     public void execute(JobExecutionContext jobExecutionContext) {
         try {
             Map<?, ?> liveJson = new ObjectMapper().readValue(new URL("https://api.live.bilibili.com/xlive/web-interface/v1/index/getList?platform=web"), Map.class);
+            liveJson = (Map<?, ?>) requireNonNull(liveJson.get("data"));
+            final List<?> roomList = (List<?>) liveJson.get("room_list");
+            roomList.forEach(liveRoom -> {
+                Map<?, ?> roomMap = (Map<?, ?>) liveRoom;
+                this.liveTitle = (String) roomMap.get("title");
+                super.createCollectionTask((String) roomMap.get("link"));
+            });
         } catch (Exception e) {
             log.error(e.getMessage(), e);
         }