123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152 |
- <?php
- declare(strict_types=1);
- namespace App\Controller;
- use App\JsonRpc\CollectorServiceInterface;
- use App\Tools\PublicData;
- use Hyperf\Di\Annotation\Inject;
- use Hyperf\HttpServer\Annotation\AutoController;
- use App\Tools\Result;
- use Hyperf\Validation\Contract\ValidatorFactoryInterface;
- use App\Constants\ErrorCode;
- use Hyperf\Context\Context;
- use App\Service\GatherQueueService;
- use App\Service\ImportQueueService;
- /**
- * ClassCollectorController
- * @package App\Controller
- */
- class CollectorController extends AbstractController
- {
- #[Inject]
- protected ValidatorFactoryInterface $validationFactory;
- #[Inject]
- protected GatherQueueService $Gservice;
- #[Inject]
- protected ImportQueueService $Iservice;
- /**
- * @var CollectorServiceInterface
- */
- #[Inject]
- private $collectorServiceClient;
-
- /**
- * 添加网站
- * @return array
- */
- public function addWeb()
- {
- $requireData = $this->request->all();
- $validator = $this->validationFactory->make(
- $requireData,
- [
- 'name'=> 'required',
- 'url'=> 'required',
- ],
- [
- 'name.required' => '网站名称不能为空',
- 'url.required' => '网站地址不能为空',
- ]
- );
- if ($validator->fails()) {
- $errorMessage = $validator->errors()->first();
- return Result::error($errorMessage);
- }
-
- $result = $this->collectorServiceClient->addWeb($requireData);
- if ($result['code'] != ErrorCode::SUCCESS) {
- return Result::error($result['message'],0,[]);
- }
- return Result::success($result['data']);
- }
- /**
- * 获取网站
- * @return array
- */
- public function getWeb()
- {
- $requireData = $this->request->all();
- $validator = $this->validationFactory->make(
- $requireData,
- [
- 'page'=> 'required',
- 'pageSize'=> 'required',
- ],
- [
- 'page.required' => '第几页不能为空',
- 'pageSize.required' => '每页显示条数不能为空',
- ]
- );
- if(isset($requireData['keyWord'])){
- $validator = $this->validationFactory->make(
- $requireData,
- [
- 'keyWord'=> 'required'
- ],
- [
- 'keyWord.required' => '搜索词不能为空'
- ]
- );
- }
-
- if ($validator->fails()) {
- $errorMessage = $validator->errors()->first();
- return Result::error($errorMessage);
- }
- $result = $this->collectorServiceClient->getWeb($requireData);
- if ($result['code'] != ErrorCode::SUCCESS) {
- return Result::error($result['message'],0,[]);
- }
- return Result::success($result['data']);
- }
- /**
- * 采集动作
- * @return array
- */
- public function sendCrawler()
- {
- $data = [];
- $data['admin_user_id'] = Context::get("UserId");
- $data['time'] = microtime();
- $requireData = $this->request->all();
- $validator = $this->validationFactory->make(
- $requireData,
- [
- 'id'=> 'required',
- ],
- [
- 'id.required' => '规则id不能为空',
- ]
- );
- if ($validator->fails()) {
- $errorMessage = $validator->errors()->first();
- return Result::error($errorMessage);
- }
- // $b = $this->Iservice->push($data,1);
- $data['id'] = $requireData['id'];
- $result = $this->Gservice->push($data,3);
- return Result::success($result);
- }
- /**
- * 发送请求
- * @param $data
- * @return array
- */
- public function goSendCrawler($data=[])
- {
- var_dump("我要开始采集了!!!");
- $this->collectorServiceClient->sendCrawler($data);
- return Result::success([]);
- }
- }
|