|
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