GuzzleHandler.php 2.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485
  1. <?php
  2. namespace Aws\Handler\GuzzleV6;
  3. use Exception;
  4. use GuzzleHttp\Exception\ConnectException;
  5. use GuzzleHttp\Exception\RequestException;
  6. use GuzzleHttp\Promise;
  7. use GuzzleHttp\Client;
  8. use GuzzleHttp\ClientInterface;
  9. use GuzzleHttp\TransferStats;
  10. use Psr\Http\Message\RequestInterface as Psr7Request;
  11. /**
  12. * A request handler that sends PSR-7-compatible requests with Guzzle 6.
  13. */
  14. class GuzzleHandler
  15. {
  16. /** @var ClientInterface */
  17. private $client;
  18. /**
  19. * @param ClientInterface $client
  20. */
  21. public function __construct(ClientInterface $client = null)
  22. {
  23. $this->client = $client ?: new Client();
  24. }
  25. /**
  26. * @param Psr7Request $request
  27. * @param array $options
  28. *
  29. * @return Promise\Promise
  30. */
  31. public function __invoke(Psr7Request $request, array $options = [])
  32. {
  33. $request = $request->withHeader(
  34. 'User-Agent',
  35. $request->getHeaderLine('User-Agent')
  36. . ' ' . \GuzzleHttp\default_user_agent()
  37. );
  38. return $this->client->sendAsync($request, $this->parseOptions($options))
  39. ->otherwise(
  40. static function (\Exception $e) {
  41. $error = [
  42. 'exception' => $e,
  43. 'connection_error' => $e instanceof ConnectException,
  44. 'response' => null,
  45. ];
  46. if ($e instanceof RequestException && $e->getResponse()) {
  47. $error['response'] = $e->getResponse();
  48. }
  49. return new Promise\RejectedPromise($error);
  50. }
  51. );
  52. }
  53. private function parseOptions(array $options)
  54. {
  55. if (isset($options['http_stats_receiver'])) {
  56. $fn = $options['http_stats_receiver'];
  57. unset($options['http_stats_receiver']);
  58. $prev = isset($options['on_stats'])
  59. ? $options['on_stats']
  60. : null;
  61. $options['on_stats'] = static function (
  62. TransferStats $stats
  63. ) use ($fn, $prev) {
  64. if (is_callable($prev)) {
  65. $prev($stats);
  66. }
  67. $transferStats = ['total_time' => $stats->getTransferTime()];
  68. $transferStats += $stats->getHandlerStats();
  69. $fn($transferStats);
  70. };
  71. }
  72. return $options;
  73. }
  74. }