|
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,20 @@ 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 | 52 | $requestIdProvider = $this->requestIdProviderFactory->create($request);
|
52 |
| - |
53 | 53 | $this->requestId = $requestIdProvider->getRequestId();
|
54 |
| - |
55 | 54 | $requestWithAttribute = $request->withAttribute(self::ATTRIBUTE_NAME, $this->requestId);
|
56 | 55 |
|
57 |
| - $nextResponse = $next($requestWithAttribute, $response); |
| 56 | + $response = $delegate->process($requestWithAttribute); |
58 | 57 |
|
59 | 58 | if (is_string($this->responseHeader)) {
|
60 |
| - return $nextResponse->withHeader($this->responseHeader, $this->requestId); |
| 59 | + return $response->withHeader($this->responseHeader, $this->requestId); |
61 | 60 | }
|
62 |
| - return $nextResponse; |
| 61 | + return $response; |
63 | 62 | }
|
64 | 63 |
|
65 | 64 | /**
|
|
0 commit comments