SciengineController.java 1.39 KB
package com.canrd.webmagic.controller;

import com.canrd.webmagic.common.constant.ServerResult;
import com.canrd.webmagic.processor.ScienginePcoessor;
import com.canrd.webmagic.processor.download.SeleniumDownloader;
import com.canrd.webmagic.processor.pipeline.ArticlePipeline;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
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.RestController;
import us.codecraft.webmagic.Request;
import us.codecraft.webmagic.Spider;

import javax.annotation.Resource;

@RestController
@RequestMapping("/sciengine/article")
@Api("Sciengine")
public class SciengineController {
    @Resource
    private ScienginePcoessor scienginePcoessor;

    @Resource
    private SeleniumDownloader seleniumDownloader;

    @Resource
    private ArticlePipeline articlePipeline;

    @GetMapping("/start")
    @ApiOperation("start")
    public ServerResult start()  {
        Spider.create(scienginePcoessor)
                .setUUID(UuidUtil.getTimeBasedUuid().toString())
                .addRequest(new Request("https://www.sciengine.com/plat/search?queryField_a=battery"))
                .addPipeline(articlePipeline)
                .thread(20).run();
        return ServerResult.success();
    }
}