UnivieController.java 2.34 KB
package com.canrd.webmagic.controller;

import com.canrd.webmagic.common.constant.ServerResult;
import com.canrd.webmagic.common.utils.KeywordUtil;
import com.canrd.webmagic.processor.Science4SpjSearchPageProcessor;
import com.canrd.webmagic.processor.UnivieSearchPageProcessor;
import com.canrd.webmagic.processor.download.SeleniumDownloader;
import org.apache.logging.log4j.core.util.UuidUtil;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
import us.codecraft.webmagic.Spider;

import javax.annotation.Resource;

/**
 * nature-文章信息(NatureArticle)表控制层
 *
 * @author makejava
 * @since 2024-04-07 18:39:41
 */
@RestController
@RequestMapping("/univie/ac")
public class UnivieController {

    @Resource
    private UnivieSearchPageProcessor univieSearchPageProcessor;

    @Resource
    private SeleniumDownloader seleniumDownloader;

    /**
     * @return
     */
    @GetMapping("/start")
    public ServerResult start(@RequestParam(value = "indexSize") Integer indexSize) {
        for (int i = 0; i <= indexSize; i++) {
            Spider.create(univieSearchPageProcessor)
                    .addUrl("https://www.univie.ac.at/suche/?&currentPage=" + i)
                    // 开启5个线程执行,并开始爬取
                    .setUUID(UuidUtil.getTimeBasedUuid().toString())
//                    .setDownloader(seleniumDownloader)
                    .thread(5).run();
        }

        return ServerResult.success();
    }


    /**
     * @return
     */
    @GetMapping("/search")
    public ServerResult search(@RequestParam(value = "indexSize") Integer indexSize) {
        for (int i = 1; i <= indexSize; i++) {
            for (String keyword : KeywordUtil.getKeyWordList()) {
                Spider.create(univieSearchPageProcessor)
                        .addUrl("https://www.univie.ac.at/suche/?q=" + keyword + "&currentPage=" + i)
                        // 开启5个线程执行,并开始爬取
                        .setUUID(UuidUtil.getTimeBasedUuid().toString())
//                        .setDownloader(seleniumDownloader)
                        .thread(5).run();
            }
        }

        return ServerResult.success();
    }

}