CollectorController.php 3.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116
  1. <?php
  2. declare(strict_types=1);
  3. namespace App\Controller;
  4. use App\JsonRpc\CollectorServiceInterface;
  5. use App\Tools\PublicData;
  6. use Hyperf\Di\Annotation\Inject;
  7. use Hyperf\HttpServer\Annotation\AutoController;
  8. use App\Tools\Result;
  9. use Hyperf\Validation\Contract\ValidatorFactoryInterface;
  10. use App\Constants\ErrorCode;
  11. use Hyperf\Context\Context;
  12. /**
  13. * ClassCollectorController
  14. * @package App\Controller
  15. */
  16. class CollectorController extends AbstractController
  17. {
  18. #[Inject]
  19. protected ValidatorFactoryInterface $validationFactory;
  20. /**
  21. * @var CollectorServiceInterface
  22. */
  23. #[Inject]
  24. private $collectorServiceClient;
  25. /**
  26. * 添加网站
  27. * @return array
  28. */
  29. public function addWeb()
  30. {
  31. $requireData = $this->request->all();
  32. $validator = $this->validationFactory->make(
  33. $requireData,
  34. [
  35. 'name'=> 'required',
  36. 'url'=> 'required',
  37. ],
  38. [
  39. 'name.required' => '网站名称不能为空',
  40. 'url.required' => '网站地址不能为空',
  41. ]
  42. );
  43. if ($validator->fails()) {
  44. $errorMessage = $validator->errors()->first();
  45. return Result::error($errorMessage);
  46. }
  47. $result = $this->collectorServiceClient->addWeb($requireData);
  48. if ($result['code'] != ErrorCode::SUCCESS) {
  49. return Result::error($result['message'],0,[]);
  50. }
  51. return Result::success($result['data']);
  52. }
  53. /**
  54. * 获取网站
  55. * @return array
  56. */
  57. public function getWeb()
  58. {
  59. $requireData = $this->request->all();
  60. $validator = $this->validationFactory->make(
  61. $requireData,
  62. [
  63. 'page'=> 'required',
  64. 'pageSize'=> 'required',
  65. ],
  66. [
  67. 'page.required' => '第几页不能为空',
  68. 'pageSize.required' => '每页显示条数不能为空',
  69. ]
  70. );
  71. if(isset($requireData['keyWord'])){
  72. $validator = $this->validationFactory->make(
  73. $requireData,
  74. [
  75. 'keyWord'=> 'required'
  76. ],
  77. [
  78. 'keyWord.required' => '搜索词不能为空'
  79. ]
  80. );
  81. }
  82. if ($validator->fails()) {
  83. $errorMessage = $validator->errors()->first();
  84. return Result::error($errorMessage);
  85. }
  86. $result = $this->collectorServiceClient->getWeb($requireData);
  87. if ($result['code'] != ErrorCode::SUCCESS) {
  88. return Result::error($result['message'],0,[]);
  89. }
  90. return Result::success($result['data']);
  91. }
  92. /**
  93. * @return array
  94. */
  95. public function sendCrawler()
  96. {
  97. $requireData = $this->request->all();
  98. $result = $this->collectorServiceClient->sendCrawler($requireData);
  99. return $result?Result::success($result['data']):Result::error($result['message']);
  100. }
  101. }