Middleware.php 3.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111
  1. <?php
  2. namespace WeWork\Http;
  3. use GuzzleHttp\Exception\ConnectException;
  4. use GuzzleHttp\Exception\RequestException;
  5. use GuzzleHttp\MessageFormatter;
  6. use GuzzleHttp\Psr7\Request;
  7. use GuzzleHttp\Psr7\Uri;
  8. use Psr\Http\Message\RequestInterface;
  9. use Psr\Http\Message\ResponseInterface;
  10. use Psr\Log\LoggerInterface;
  11. use Psr\Log\LogLevel;
  12. use WeWork\ApiCache\Token;
  13. class Middleware
  14. {
  15. const RETRY_MAX_RETRIES = 1;
  16. /**
  17. * @param Token $token
  18. * @return callable
  19. */
  20. public static function auth(Token $token)
  21. {
  22. return \GuzzleHttp\Middleware::mapRequest(function (RequestInterface $request) use ($token) {
  23. return $request->withUri(Uri::withQueryValue($request->getUri(), 'access_token', $token->get()));
  24. });
  25. }
  26. /**
  27. * @param LoggerInterface $logger
  28. * @return callable
  29. */
  30. public static function log(LoggerInterface $logger)
  31. {
  32. return \GuzzleHttp\Middleware::log($logger, new MessageFormatter(MessageFormatter::DEBUG), LogLevel::DEBUG);
  33. }
  34. /**
  35. * @param LoggerInterface $logger
  36. * @return callable
  37. */
  38. public static function retry(LoggerInterface $logger)
  39. {
  40. return \GuzzleHttp\Middleware::retry(function (
  41. $retries,
  42. Request $request,
  43. Response $response = null,
  44. RequestException $exception = null
  45. ) use ($logger) {
  46. if ($retries >= self::RETRY_MAX_RETRIES) {
  47. return false;
  48. }
  49. if (!(self::isServerError($response) || self::isConnectError($exception))) {
  50. return false;
  51. }
  52. $logger->warning(
  53. sprintf(
  54. 'Retrying %s %s %s/%s, %s',
  55. $request->getMethod(),
  56. $request->getUri(),
  57. $retries + 1,
  58. self::RETRY_MAX_RETRIES,
  59. $response ? 'status code: ' . $response->getStatusCode() : $exception->getMessage()
  60. ),
  61. [$request->getHeader('Host')[0]]
  62. );
  63. return true;
  64. });
  65. }
  66. /**
  67. * @return callable
  68. */
  69. public static function response()
  70. {
  71. return \GuzzleHttp\Middleware::mapResponse(function (ResponseInterface $response) {
  72. return new Response(
  73. $response->getStatusCode(),
  74. $response->getHeaders(),
  75. $response->getBody(),
  76. $response->getProtocolVersion(),
  77. $response->getReasonPhrase()
  78. );
  79. });
  80. }
  81. /**
  82. * @param Response $response
  83. *
  84. * @return bool
  85. */
  86. private static function isServerError(Response $response = null)
  87. {
  88. return $response && $response->getStatusCode() >= 500;
  89. }
  90. /**
  91. * @param RequestException $exception
  92. *
  93. * @return bool
  94. */
  95. private static function isConnectError(RequestException $exception = null)
  96. {
  97. return $exception instanceof ConnectException;
  98. }
  99. }