version 1.5
This commit is contained in:
@@ -0,0 +1,39 @@
|
||||
<?php
|
||||
|
||||
declare(strict_types=1);
|
||||
|
||||
namespace Http\Client\Common\Plugin;
|
||||
|
||||
use Http\Client\Common\Plugin;
|
||||
use Http\Message\Encoding\ChunkStream;
|
||||
use Http\Promise\Promise;
|
||||
use Psr\Http\Message\RequestInterface;
|
||||
|
||||
/**
|
||||
* Allow to set the correct content length header on the request or to transfer it as a chunk if not possible.
|
||||
*
|
||||
* @author Joel Wurtz <joel.wurtz@gmail.com>
|
||||
*/
|
||||
final class ContentLengthPlugin implements Plugin
|
||||
{
|
||||
/**
|
||||
* {@inheritdoc}
|
||||
*/
|
||||
public function handleRequest(RequestInterface $request, callable $next, callable $first): Promise
|
||||
{
|
||||
if (!$request->hasHeader('Content-Length')) {
|
||||
$stream = $request->getBody();
|
||||
|
||||
// Cannot determine the size so we use a chunk stream
|
||||
if (null === $stream->getSize()) {
|
||||
$stream = new ChunkStream($stream);
|
||||
$request = $request->withBody($stream);
|
||||
$request = $request->withAddedHeader('Transfer-Encoding', 'chunked');
|
||||
} else {
|
||||
$request = $request->withHeader('Content-Length', (string) $stream->getSize());
|
||||
}
|
||||
}
|
||||
|
||||
return $next($request);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user