diff options
Diffstat (limited to 'vendor/guzzle/guzzle/src/Guzzle/Batch/BatchRequestTransfer.php')
-rw-r--r-- | vendor/guzzle/guzzle/src/Guzzle/Batch/BatchRequestTransfer.php | 65 |
1 files changed, 65 insertions, 0 deletions
diff --git a/vendor/guzzle/guzzle/src/Guzzle/Batch/BatchRequestTransfer.php b/vendor/guzzle/guzzle/src/Guzzle/Batch/BatchRequestTransfer.php new file mode 100644 index 0000000..4d8489c --- /dev/null +++ b/vendor/guzzle/guzzle/src/Guzzle/Batch/BatchRequestTransfer.php @@ -0,0 +1,65 @@ +<?php + +namespace Guzzle\Batch; + +use Guzzle\Batch\BatchTransferInterface; +use Guzzle\Batch\BatchDivisorInterface; +use Guzzle\Common\Exception\InvalidArgumentException; +use Guzzle\Http\Message\RequestInterface; + +/** + * Batch transfer strategy used to efficiently transfer a batch of requests. + * This class is to be used with {@see Guzzle\Batch\BatchInterface} + */ +class BatchRequestTransfer implements BatchTransferInterface, BatchDivisorInterface +{ + /** @var int Size of each command batch */ + protected $batchSize; + + /** + * Constructor used to specify how large each batch should be + * + * @param int $batchSize Size of each batch + */ + public function __construct($batchSize = 50) + { + $this->batchSize = $batchSize; + } + + /** + * Creates batches of requests by grouping requests by their associated curl multi object. + * {@inheritdoc} + */ + public function createBatches(\SplQueue $queue) + { + // Create batches by client objects + $groups = new \SplObjectStorage(); + foreach ($queue as $item) { + if (!$item instanceof RequestInterface) { + throw new InvalidArgumentException('All items must implement Guzzle\Http\Message\RequestInterface'); + } + $client = $item->getClient(); + if (!$groups->contains($client)) { + $groups->attach($client, array($item)); + } else { + $current = $groups[$client]; + $current[] = $item; + $groups[$client] = $current; + } + } + + $batches = array(); + foreach ($groups as $batch) { + $batches = array_merge($batches, array_chunk($groups[$batch], $this->batchSize)); + } + + return $batches; + } + + public function transfer(array $batch) + { + if ($batch) { + reset($batch)->getClient()->send($batch); + } + } +} |