VirusAdd.php 4.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161
  1. <?php
  2. /**
  3. * Created by PhpStorm.
  4. * User: Administrator
  5. * Date: 2019/6/24
  6. * Time: 9:52
  7. */
  8. namespace App\Console\Commands;
  9. use GuzzleHttp\Exception\RequestException;
  10. use Illuminate\Console\Command;
  11. use Illuminate\Support\Facades\Log;
  12. use PhpAmqpLib\Connection\AMQPStreamConnection;
  13. use GuzzleHttp\Client;
  14. class VirusAdd extends Command
  15. {
  16. /**
  17. * The name and signature of the console command.
  18. *
  19. * @var string
  20. */
  21. protected $signature = 'virus:add';
  22. /**
  23. * The console command description.
  24. *
  25. * @var string
  26. */
  27. protected $description = 'virus添加数据';
  28. protected $channel;
  29. protected $connection;
  30. protected $queue = 'virus_add';
  31. /**
  32. * Create a new command instance.
  33. *
  34. * @return void
  35. */
  36. public function __construct()
  37. {
  38. parent::__construct();
  39. $this->connection = $this->getConnection();
  40. $this->channel = $this->connection->channel();
  41. $this->channel->exchange_declare($this->queue, 'fanout', false, true, false);
  42. // 初始化virus域名
  43. $this->client = new Client([
  44. 'base_uri' => config('customer.VIRUS_URL'),
  45. 'timeout' => 3
  46. ]);
  47. }
  48. public function getConnection()
  49. {
  50. $conn = false;
  51. if ($this->connection) {
  52. return $this->connection;
  53. }
  54. for ($i = 0; $i < 3; $i++) {
  55. $connection = $this->createConn();
  56. if ($connection) {
  57. $conn = $connection;
  58. break;
  59. }
  60. Log::info("create amqp conn retry=" . $i);
  61. }
  62. return $conn;
  63. }
  64. public function createConn()
  65. {
  66. try {
  67. $connection = new AMQPStreamConnection(env('MQ_HOST'), env('MQ_PORT'), env('MQ_USERNAME'), env('MQ_PWD'), env('MQ_VHOST'));
  68. } catch (\Exception $exception) {
  69. Log::info("AMQP connection Error" . $exception->getMessage());
  70. $connection = false;
  71. }
  72. return $connection;
  73. }
  74. /**
  75. * Execute the console command.
  76. *
  77. * @return mixed
  78. */
  79. public function handle()
  80. {
  81. Log::info('接收一条消息');
  82. $queue_name = $this->queue;
  83. $this->channel->queue_declare($queue_name, false, true, false, false);
  84. $this->channel->queue_bind($queue_name, $this->queue);
  85. $callback = function ($msg) {
  86. $this->line(date('Y-m-d H:i:S').$msg->body);
  87. $data = \GuzzleHttp\json_decode($msg->body, true);
  88. $this->line('接收一条消息收到数据' . $msg->body);
  89. $res = $this->send($data);
  90. if ($res) {
  91. $msg->delivery_info['channel']->basic_ack($msg->delivery_info['delivery_tag']);
  92. }
  93. };
  94. $this->channel->basic_consume($queue_name, '', false, false, false, false, $callback);
  95. while (count($this->channel->callbacks)) {
  96. $this->channel->wait();
  97. }
  98. $this->channel->close();
  99. $this->connection->close();
  100. }
  101. private function send($data)
  102. {
  103. try {
  104. // 签名设置
  105. $signKey = [
  106. 'app_id' => config('customer.VIRUS_APP_ID'),
  107. 'behavior_id' => $data['behavior_id'],
  108. 'target_id' => (string) $data['target_id'],
  109. 'action_id' => (string) $data['action_id'],
  110. 'behavior_flag' => $data['behavior_flag']
  111. ];
  112. ksort($signKey);
  113. $signKey = urldecode(http_build_query($signKey));
  114. $sign = md5(config('customer.VIRUS_APP_SECRET') . $signKey);
  115. Log::debug('签名:'.$sign);
  116. $json = [];
  117. if($data['behavior_flag'] == 'publish'){
  118. $json = [
  119. 'app_id' => config('customer.VIRUS_APP_ID'),
  120. 'behavior_id' => $data['behavior_id'],
  121. 'behavior_flag' => $data['behavior_flag'],
  122. 'post_id' => $data['post_id'],
  123. 'post_desc' => $data['post_desc'],
  124. 'post_cover' => $data['post_cover'],
  125. 'target_id' => (string) $data['target_id'],
  126. 'action_id' => (string) $data['action_id'],
  127. 'extra' => [],
  128. ];
  129. }else{
  130. Log::debug('行为类型错误'.json_encode($data));
  131. }
  132. $response = $this->client->request('POST', 'v2/record/add', [
  133. 'headers' => [
  134. 'Content-Type' => 'application/json',
  135. 'sign' => $sign
  136. ],
  137. 'json' => $json
  138. ]);
  139. // $result = \GuzzleHttp\json_decode($response->getBody()->getContents(), true);
  140. Log::debug('virus添加数据成功:'.$response->getBody()->getContents());
  141. return true;
  142. } catch (RequestException $e) {
  143. Log::debug('virus添加数据失败:'.$e->getMessage());
  144. }
  145. }
  146. }