|
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