|
2 | 2 |
|
3 | 3 | namespace PhpMiddleware\RequestId; |
4 | 4 |
|
| 5 | +use Interop\Http\ServerMiddleware\DelegateInterface; |
| 6 | +use Interop\Http\ServerMiddleware\MiddlewareInterface; |
| 7 | +use PhpMiddleware\DoublePassCompatibilityTrait; |
5 | 8 | use PhpMiddleware\RequestId\Exception\NotGenerated; |
6 | 9 | use PhpMiddleware\RequestId\RequestIdProviderFactoryInterface; |
7 | 10 | use Psr\Http\Message\ResponseInterface; |
8 | 11 | use Psr\Http\Message\ServerRequestInterface; |
9 | 12 |
|
10 | | -final class RequestIdMiddleware implements RequestIdProviderInterface |
| 13 | +final class RequestIdMiddleware implements RequestIdProviderInterface, MiddlewareInterface |
11 | 14 | { |
| 15 | + use DoublePassCompatibilityTrait; |
| 16 | + |
12 | 17 | const DEFAULT_RESPONSE_HEADER = 'X-Request-Id'; |
13 | 18 | const ATTRIBUTE_NAME = 'request-id'; |
14 | 19 |
|
@@ -40,26 +45,45 @@ public function __construct( |
40 | 45 | } |
41 | 46 |
|
42 | 47 | /** |
43 | | - * @param ServerRequestInterface $request |
44 | | - * @param ResponseInterface $response |
45 | | - * @param callable $next |
46 | | - * |
47 | 48 | * @return ResponseInterface |
48 | 49 | */ |
49 | | - public function __invoke(ServerRequestInterface $request, ResponseInterface $response, callable $next) |
| 50 | + public function process(ServerRequestInterface $request, DelegateInterface $delegate) |
50 | 51 | { |
51 | | - $requestIdProvider = $this->requestIdProviderFactory->create($request); |
| 52 | + $requestWithAttribute = $this->attachRequestIdToAttribute($request); |
52 | 53 |
|
| 54 | + $response = $delegate->process($requestWithAttribute); |
| 55 | + |
| 56 | + if ($this->canAttachToResponse()) { |
| 57 | + return $this->attachRequestIdToResponse($response); |
| 58 | + } |
| 59 | + return $response; |
| 60 | + } |
| 61 | + |
| 62 | + /** |
| 63 | + * @return ResponseInterface |
| 64 | + */ |
| 65 | + private function attachRequestIdToAttribute(ServerRequestInterface $request) |
| 66 | + { |
| 67 | + $requestIdProvider = $this->requestIdProviderFactory->create($request); |
53 | 68 | $this->requestId = $requestIdProvider->getRequestId(); |
54 | 69 |
|
55 | | - $requestWithAttribute = $request->withAttribute(self::ATTRIBUTE_NAME, $this->requestId); |
| 70 | + return $request->withAttribute(self::ATTRIBUTE_NAME, $this->requestId); |
| 71 | + } |
56 | 72 |
|
57 | | - $nextResponse = $next($requestWithAttribute, $response); |
| 73 | + /** |
| 74 | + * @return ResponseInterface |
| 75 | + */ |
| 76 | + private function attachRequestIdToResponse(ResponseInterface $response) |
| 77 | + { |
| 78 | + return $response->withHeader($this->responseHeader, $this->requestId); |
| 79 | + } |
58 | 80 |
|
59 | | - if (is_string($this->responseHeader)) { |
60 | | - return $nextResponse->withHeader($this->responseHeader, $this->requestId); |
61 | | - } |
62 | | - return $nextResponse; |
| 81 | + /** |
| 82 | + * @return bool |
| 83 | + */ |
| 84 | + private function canAttachToResponse() |
| 85 | + { |
| 86 | + return is_string($this->responseHeader) && !empty($this->responseHeader); |
63 | 87 | } |
64 | 88 |
|
65 | 89 | /** |
|
0 commit comments