GuzzleHandler.php 2.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788
  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 ($e) {
  41. $error = [
  42. 'exception' => $e,
  43. 'connection_error' => $e instanceof ConnectException,
  44. 'response' => null,
  45. ];
  46. if (
  47. ($e instanceof RequestException)
  48. && $e->getResponse()
  49. ) {
  50. $error['response'] = $e->getResponse();
  51. }
  52. return new Promise\RejectedPromise($error);
  53. }
  54. );
  55. }
  56. private function parseOptions(array $options)
  57. {
  58. if (isset($options['http_stats_receiver'])) {
  59. $fn = $options['http_stats_receiver'];
  60. unset($options['http_stats_receiver']);
  61. $prev = isset($options['on_stats'])
  62. ? $options['on_stats']
  63. : null;
  64. $options['on_stats'] = static function (
  65. TransferStats $stats
  66. ) use ($fn, $prev) {
  67. if (is_callable($prev)) {
  68. $prev($stats);
  69. }
  70. $transferStats = ['total_time' => $stats->getTransferTime()];
  71. $transferStats += $stats->getHandlerStats();
  72. $fn($transferStats);
  73. };
  74. }
  75. return $options;
  76. }
  77. }