forked from pavlokomarov/roach-php-core
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathDownloader.php
More file actions
106 lines (84 loc) · 2.54 KB
/
Downloader.php
File metadata and controls
106 lines (84 loc) · 2.54 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
<?php
declare(strict_types=1);
/**
* Copyright (c) 2021 Kai Sassnowski
*
* For the full copyright and license information, please view
* the LICENSE file that was distributed with this source code.
*
* @see https://github.com/roach-php/roach
*/
namespace RoachPHP\Downloader;
use RoachPHP\Events\RequestDropped;
use RoachPHP\Events\RequestSending;
use RoachPHP\Http\ClientInterface;
use RoachPHP\Http\Request;
use RoachPHP\Http\Response;
use Symfony\Component\EventDispatcher\EventDispatcherInterface;
final class Downloader
{
/**
* @var DownloaderMiddlewareInterface[]
*/
private array $middleware = [];
/**
* @var Request[]
*/
private array $requests = [];
public function __construct(
private ClientInterface $client,
private EventDispatcherInterface $eventDispatcher,
) {
}
public function withMiddleware(DownloaderMiddlewareInterface ...$middleware): self
{
$this->middleware = $middleware;
return $this;
}
public function prepare(Request $request): void
{
foreach ($this->middleware as $middleware) {
$request = $middleware->handleRequest($request);
if ($request->wasDropped()) {
$this->eventDispatcher->dispatch(
new RequestDropped($request),
RequestDropped::NAME,
);
return;
}
}
/** @var RequestSending $event */
$event = $this->eventDispatcher->dispatch(
new RequestSending($request),
RequestSending::NAME,
);
if ($event->request->wasDropped()) {
$this->eventDispatcher->dispatch(
new RequestDropped($event->request),
RequestDropped::NAME,
);
return;
}
$this->requests[] = $event->request;
}
public function flush(?callable $callback = null): void
{
$requests = $this->requests;
$this->requests = [];
$this->client->pool($requests, function (Response $response) use ($callback): void {
$this->onResponseReceived($response, $callback);
});
}
private function onResponseReceived(Response $response, ?callable $callback): void
{
foreach ($this->middleware as $middleware) {
$response = $middleware->handleResponse($response);
if ($response->wasDropped()) {
return;
}
}
if (null !== $callback) {
$callback($response);
}
}
}