HummingBirdAnimeClient/src/AnimeClient/API/ParallelAPIRequest.php

111 lines
2.2 KiB
PHP
Raw Normal View History

2017-02-17 08:25:19 -05:00
<?php declare(strict_types=1);
/**
* Hummingbird Anime List Client
*
2018-08-22 13:48:27 -04:00
* An API client for Kitsu to manage anime and manga watch lists
2017-02-17 08:25:19 -05:00
*
* PHP version 7.4
2017-02-17 08:25:19 -05:00
*
* @package HummingbirdAnimeClient
* @author Timothy J. Warren <tim@timshomepage.net>
2020-01-08 15:39:49 -05:00
* @copyright 2015 - 2020 Timothy J. Warren
2017-02-17 08:25:19 -05:00
* @license http://www.opensource.org/licenses/mit-license.html MIT License
* @version 5
* @link https://git.timshomepage.net/timw4mail/HummingBirdAnimeClient
2017-02-17 08:25:19 -05:00
*/
namespace Aviat\AnimeClient\API;
2020-03-11 16:26:17 -04:00
use Amp\Http\Client\Request;
use function Amp\call;
use function Amp\Promise\{all, wait};
use function Aviat\AnimeClient\getApiClient;
2018-11-29 11:00:50 -05:00
2019-12-09 14:34:23 -05:00
use Throwable;
2017-02-17 08:25:19 -05:00
/**
* Class to simplify making and validating simultaneous requests
*/
final class ParallelAPIRequest {
2017-02-17 08:25:19 -05:00
/**
* Set of requests to make in parallel
*
* @var array
*/
2020-04-10 20:01:46 -04:00
private array $requests = [];
2017-02-17 08:25:19 -05:00
/**
* Add a request
*
2019-12-06 09:15:49 -05:00
* @param string|Request $request
2017-02-17 08:25:19 -05:00
* @param string|number $key
* @return self
*/
2017-02-17 10:55:17 -05:00
public function addRequest($request, $key = NULL): self
2017-02-17 08:25:19 -05:00
{
2018-02-02 09:50:58 -05:00
if ($key !== NULL)
2017-02-17 08:25:19 -05:00
{
$this->requests[$key] = $request;
return $this;
}
2017-02-17 08:25:19 -05:00
$this->requests[] = $request;
return $this;
}
2017-02-17 08:25:19 -05:00
/**
* Add multiple requests
*
2019-12-06 09:15:49 -05:00
* @param string[]|Request[] $requests
2017-02-17 08:25:19 -05:00
* @return self
*/
public function addRequests(array $requests): self
{
array_walk($requests, [$this, 'addRequest']);
return $this;
}
2017-02-17 08:25:19 -05:00
/**
2018-10-01 10:50:22 -04:00
* Make the requests, and return the body for each
2017-02-17 08:25:19 -05:00
*
* @return array
2019-12-09 14:34:23 -05:00
* @throws Throwable
2017-02-17 08:25:19 -05:00
*/
public function makeRequests(): array
2017-02-17 08:25:19 -05:00
{
$client = getApiClient();
$promises = [];
foreach ($this->requests as $key => $url)
{
2019-12-06 09:15:49 -05:00
$promises[$key] = call(static function () use ($client, $url) {
$response = yield $client->request($url);
2020-03-11 16:26:17 -04:00
return yield $response->getBody()->buffer();
});
}
return wait(all($promises));
2017-02-17 08:25:19 -05:00
}
2018-10-01 10:50:22 -04:00
/**
* Make the requests and return the response objects
*
* @return array
2019-12-09 14:34:23 -05:00
* @throws Throwable
2018-10-01 10:50:22 -04:00
*/
public function getResponses(): array
{
$client = getApiClient();
2018-10-01 10:50:22 -04:00
$promises = [];
foreach ($this->requests as $key => $url)
{
2020-05-11 09:17:11 -04:00
$promises[$key] = call(fn () => yield $client->request($url));
2018-10-01 10:50:22 -04:00
}
return wait(all($promises));
}
2017-02-17 08:25:19 -05:00
}