media-cloud-sync / includes / sdk / google / guzzlehttp / guzzle / src / Handler / CurlMultiHandler.php

CurlMultiHandler.php in Media Cloud Sync 1.2.9, at includes/sdk/google/guzzlehttp/guzzle/src/Handler/CurlMultiHandler.php

236 lines 8.4 KB

Raw Download Zip

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