1 |
efrain |
1 |
<?php
|
|
|
2 |
|
|
|
3 |
namespace GuzzleHttp\Handler;
|
|
|
4 |
|
|
|
5 |
use GuzzleHttp\Promise as P;
|
|
|
6 |
use GuzzleHttp\Promise\Promise;
|
|
|
7 |
use GuzzleHttp\Promise\PromiseInterface;
|
|
|
8 |
use GuzzleHttp\Utils;
|
|
|
9 |
use Psr\Http\Message\RequestInterface;
|
|
|
10 |
|
|
|
11 |
/**
|
|
|
12 |
* Returns an asynchronous response using curl_multi_* functions.
|
|
|
13 |
*
|
|
|
14 |
* When using the CurlMultiHandler, custom curl options can be specified as an
|
|
|
15 |
* associative array of curl option constants mapping to values in the
|
|
|
16 |
* **curl** key of the provided request options.
|
|
|
17 |
*
|
|
|
18 |
* @property resource|\CurlMultiHandle $_mh Internal use only. Lazy loaded multi-handle.
|
|
|
19 |
*
|
|
|
20 |
* @final
|
|
|
21 |
*/
|
|
|
22 |
#[\AllowDynamicProperties]
|
|
|
23 |
class CurlMultiHandler
|
|
|
24 |
{
|
|
|
25 |
/**
|
|
|
26 |
* @var CurlFactoryInterface
|
|
|
27 |
*/
|
|
|
28 |
private $factory;
|
|
|
29 |
|
|
|
30 |
/**
|
|
|
31 |
* @var int
|
|
|
32 |
*/
|
|
|
33 |
private $selectTimeout;
|
|
|
34 |
|
|
|
35 |
/**
|
|
|
36 |
* @var int Will be higher than 0 when `curl_multi_exec` is still running.
|
|
|
37 |
*/
|
|
|
38 |
private $active = 0;
|
|
|
39 |
|
|
|
40 |
/**
|
|
|
41 |
* @var array Request entry handles, indexed by handle id in `addRequest`.
|
|
|
42 |
*
|
|
|
43 |
* @see CurlMultiHandler::addRequest
|
|
|
44 |
*/
|
|
|
45 |
private $handles = [];
|
|
|
46 |
|
|
|
47 |
/**
|
|
|
48 |
* @var array<int, float> An array of delay times, indexed by handle id in `addRequest`.
|
|
|
49 |
*
|
|
|
50 |
* @see CurlMultiHandler::addRequest
|
|
|
51 |
*/
|
|
|
52 |
private $delays = [];
|
|
|
53 |
|
|
|
54 |
/**
|
|
|
55 |
* @var array<mixed> An associative array of CURLMOPT_* options and corresponding values for curl_multi_setopt()
|
|
|
56 |
*/
|
|
|
57 |
private $options = [];
|
|
|
58 |
|
|
|
59 |
/**
|
|
|
60 |
* This handler accepts the following options:
|
|
|
61 |
*
|
|
|
62 |
* - handle_factory: An optional factory used to create curl handles
|
|
|
63 |
* - select_timeout: Optional timeout (in seconds) to block before timing
|
|
|
64 |
* out while selecting curl handles. Defaults to 1 second.
|
|
|
65 |
* - options: An associative array of CURLMOPT_* options and
|
|
|
66 |
* corresponding values for curl_multi_setopt()
|
|
|
67 |
*/
|
|
|
68 |
public function __construct(array $options = [])
|
|
|
69 |
{
|
|
|
70 |
$this->factory = $options['handle_factory'] ?? new CurlFactory(50);
|
|
|
71 |
|
|
|
72 |
if (isset($options['select_timeout'])) {
|
|
|
73 |
$this->selectTimeout = $options['select_timeout'];
|
|
|
74 |
} elseif ($selectTimeout = Utils::getenv('GUZZLE_CURL_SELECT_TIMEOUT')) {
|
|
|
75 |
@trigger_error('Since guzzlehttp/guzzle 7.2.0: Using environment variable GUZZLE_CURL_SELECT_TIMEOUT is deprecated. Use option "select_timeout" instead.', \E_USER_DEPRECATED);
|
|
|
76 |
$this->selectTimeout = (int) $selectTimeout;
|
|
|
77 |
} else {
|
|
|
78 |
$this->selectTimeout = 1;
|
|
|
79 |
}
|
|
|
80 |
|
|
|
81 |
$this->options = $options['options'] ?? [];
|
|
|
82 |
}
|
|
|
83 |
|
|
|
84 |
/**
|
|
|
85 |
* @param string $name
|
|
|
86 |
*
|
|
|
87 |
* @return resource|\CurlMultiHandle
|
|
|
88 |
*
|
|
|
89 |
* @throws \BadMethodCallException when another field as `_mh` will be gotten
|
|
|
90 |
* @throws \RuntimeException when curl can not initialize a multi handle
|
|
|
91 |
*/
|
|
|
92 |
public function __get($name)
|
|
|
93 |
{
|
|
|
94 |
if ($name !== '_mh') {
|
|
|
95 |
throw new \BadMethodCallException("Can not get other property as '_mh'.");
|
|
|
96 |
}
|
|
|
97 |
|
|
|
98 |
$multiHandle = \curl_multi_init();
|
|
|
99 |
|
|
|
100 |
if (false === $multiHandle) {
|
|
|
101 |
throw new \RuntimeException('Can not initialize curl multi handle.');
|
|
|
102 |
}
|
|
|
103 |
|
|
|
104 |
$this->_mh = $multiHandle;
|
|
|
105 |
|
|
|
106 |
foreach ($this->options as $option => $value) {
|
|
|
107 |
// A warning is raised in case of a wrong option.
|
|
|
108 |
curl_multi_setopt($this->_mh, $option, $value);
|
|
|
109 |
}
|
|
|
110 |
|
|
|
111 |
return $this->_mh;
|
|
|
112 |
}
|
|
|
113 |
|
|
|
114 |
public function __destruct()
|
|
|
115 |
{
|
|
|
116 |
if (isset($this->_mh)) {
|
|
|
117 |
\curl_multi_close($this->_mh);
|
|
|
118 |
unset($this->_mh);
|
|
|
119 |
}
|
|
|
120 |
}
|
|
|
121 |
|
|
|
122 |
public function __invoke(RequestInterface $request, array $options): PromiseInterface
|
|
|
123 |
{
|
|
|
124 |
$easy = $this->factory->create($request, $options);
|
|
|
125 |
$id = (int) $easy->handle;
|
|
|
126 |
|
|
|
127 |
$promise = new Promise(
|
|
|
128 |
[$this, 'execute'],
|
|
|
129 |
function () use ($id) {
|
|
|
130 |
return $this->cancel($id);
|
|
|
131 |
}
|
|
|
132 |
);
|
|
|
133 |
|
|
|
134 |
$this->addRequest(['easy' => $easy, 'deferred' => $promise]);
|
|
|
135 |
|
|
|
136 |
return $promise;
|
|
|
137 |
}
|
|
|
138 |
|
|
|
139 |
/**
|
|
|
140 |
* Ticks the curl event loop.
|
|
|
141 |
*/
|
|
|
142 |
public function tick(): void
|
|
|
143 |
{
|
|
|
144 |
// Add any delayed handles if needed.
|
|
|
145 |
if ($this->delays) {
|
|
|
146 |
$currentTime = Utils::currentTime();
|
|
|
147 |
foreach ($this->delays as $id => $delay) {
|
|
|
148 |
if ($currentTime >= $delay) {
|
|
|
149 |
unset($this->delays[$id]);
|
|
|
150 |
\curl_multi_add_handle(
|
|
|
151 |
$this->_mh,
|
|
|
152 |
$this->handles[$id]['easy']->handle
|
|
|
153 |
);
|
|
|
154 |
}
|
|
|
155 |
}
|
|
|
156 |
}
|
|
|
157 |
|
|
|
158 |
// Step through the task queue which may add additional requests.
|
|
|
159 |
P\Utils::queue()->run();
|
|
|
160 |
|
|
|
161 |
if ($this->active && \curl_multi_select($this->_mh, $this->selectTimeout) === -1) {
|
|
|
162 |
// Perform a usleep if a select returns -1.
|
|
|
163 |
// See: https://bugs.php.net/bug.php?id=61141
|
|
|
164 |
\usleep(250);
|
|
|
165 |
}
|
|
|
166 |
|
|
|
167 |
while (\curl_multi_exec($this->_mh, $this->active) === \CURLM_CALL_MULTI_PERFORM);
|
|
|
168 |
|
|
|
169 |
$this->processMessages();
|
|
|
170 |
}
|
|
|
171 |
|
|
|
172 |
/**
|
|
|
173 |
* Runs until all outstanding connections have completed.
|
|
|
174 |
*/
|
|
|
175 |
public function execute(): void
|
|
|
176 |
{
|
|
|
177 |
$queue = P\Utils::queue();
|
|
|
178 |
|
|
|
179 |
while ($this->handles || !$queue->isEmpty()) {
|
|
|
180 |
// If there are no transfers, then sleep for the next delay
|
|
|
181 |
if (!$this->active && $this->delays) {
|
|
|
182 |
\usleep($this->timeToNext());
|
|
|
183 |
}
|
|
|
184 |
$this->tick();
|
|
|
185 |
}
|
|
|
186 |
}
|
|
|
187 |
|
|
|
188 |
private function addRequest(array $entry): void
|
|
|
189 |
{
|
|
|
190 |
$easy = $entry['easy'];
|
|
|
191 |
$id = (int) $easy->handle;
|
|
|
192 |
$this->handles[$id] = $entry;
|
|
|
193 |
if (empty($easy->options['delay'])) {
|
|
|
194 |
\curl_multi_add_handle($this->_mh, $easy->handle);
|
|
|
195 |
} else {
|
|
|
196 |
$this->delays[$id] = Utils::currentTime() + ($easy->options['delay'] / 1000);
|
|
|
197 |
}
|
|
|
198 |
}
|
|
|
199 |
|
|
|
200 |
/**
|
|
|
201 |
* Cancels a handle from sending and removes references to it.
|
|
|
202 |
*
|
|
|
203 |
* @param int $id Handle ID to cancel and remove.
|
|
|
204 |
*
|
|
|
205 |
* @return bool True on success, false on failure.
|
|
|
206 |
*/
|
|
|
207 |
private function cancel($id): bool
|
|
|
208 |
{
|
|
|
209 |
if (!is_int($id)) {
|
|
|
210 |
trigger_deprecation('guzzlehttp/guzzle', '7.4', 'Not passing an integer to %s::%s() is deprecated and will cause an error in 8.0.', __CLASS__, __FUNCTION__);
|
|
|
211 |
}
|
|
|
212 |
|
|
|
213 |
// Cannot cancel if it has been processed.
|
|
|
214 |
if (!isset($this->handles[$id])) {
|
|
|
215 |
return false;
|
|
|
216 |
}
|
|
|
217 |
|
|
|
218 |
$handle = $this->handles[$id]['easy']->handle;
|
|
|
219 |
unset($this->delays[$id], $this->handles[$id]);
|
|
|
220 |
\curl_multi_remove_handle($this->_mh, $handle);
|
|
|
221 |
\curl_close($handle);
|
|
|
222 |
|
|
|
223 |
return true;
|
|
|
224 |
}
|
|
|
225 |
|
|
|
226 |
private function processMessages(): void
|
|
|
227 |
{
|
|
|
228 |
while ($done = \curl_multi_info_read($this->_mh)) {
|
|
|
229 |
if ($done['msg'] !== \CURLMSG_DONE) {
|
|
|
230 |
// if it's not done, then it would be premature to remove the handle. ref https://github.com/guzzle/guzzle/pull/2892#issuecomment-945150216
|
|
|
231 |
continue;
|
|
|
232 |
}
|
|
|
233 |
$id = (int) $done['handle'];
|
|
|
234 |
\curl_multi_remove_handle($this->_mh, $done['handle']);
|
|
|
235 |
|
|
|
236 |
if (!isset($this->handles[$id])) {
|
|
|
237 |
// Probably was cancelled.
|
|
|
238 |
continue;
|
|
|
239 |
}
|
|
|
240 |
|
|
|
241 |
$entry = $this->handles[$id];
|
|
|
242 |
unset($this->handles[$id], $this->delays[$id]);
|
|
|
243 |
$entry['easy']->errno = $done['result'];
|
|
|
244 |
$entry['deferred']->resolve(
|
|
|
245 |
CurlFactory::finish($this, $entry['easy'], $this->factory)
|
|
|
246 |
);
|
|
|
247 |
}
|
|
|
248 |
}
|
|
|
249 |
|
|
|
250 |
private function timeToNext(): int
|
|
|
251 |
{
|
|
|
252 |
$currentTime = Utils::currentTime();
|
|
|
253 |
$nextTime = \PHP_INT_MAX;
|
|
|
254 |
foreach ($this->delays as $time) {
|
|
|
255 |
if ($time < $nextTime) {
|
|
|
256 |
$nextTime = $time;
|
|
|
257 |
}
|
|
|
258 |
}
|
|
|
259 |
|
|
|
260 |
return ((int) \max(0, $nextTime - $currentTime)) * 1000000;
|
|
|
261 |
}
|
|
|
262 |
}
|