mirror of
https://github.com/danog/amp.git
synced 2025-01-22 05:11:42 +01:00
Add EvReactor
This commit is contained in:
Daniel Lowrey
parent
962699263c
commit
288530d296
@ -13,6 +13,7 @@ matrix:
|
||||
|
||||
before_script:
|
||||
- yes "" | pecl install "channel://pecl.php.net/libevent-0.1.0"
|
||||
- yes "" | pecl install ev
|
||||
- composer self-update
|
||||
- composer install
|
||||
|
||||
|
||||
359
lib/EvReactor.php
Normal file
359
lib/EvReactor.php
Normal file
@ -0,0 +1,359 @@
|
||||
<?php
|
||||
|
||||
namespace Amp;
|
||||
|
||||
use Ev;
|
||||
use EvIo;
|
||||
use EvLoop;
|
||||
use EvTimer;
|
||||
use EvSignal;
|
||||
|
||||
class EvReactor implements ExtensionReactor {
|
||||
private static $instanceCount = 0;
|
||||
|
||||
private $loop;
|
||||
private $lastWatcherId = "a";
|
||||
private $enabledWatchers = [];
|
||||
private $disabledWatchers = [];
|
||||
private $enabledImmediates = [];
|
||||
private $disabledImmediates = [];
|
||||
private $isRunning = false;
|
||||
private $stopException;
|
||||
private $onError;
|
||||
private $onCoroutineResolution;
|
||||
|
||||
public function __construct($flags = null) {
|
||||
if (!extension_loaded("ev")) {
|
||||
throw new \RuntimeException(
|
||||
"The ev extension is required to use the EvReactor."
|
||||
);
|
||||
}
|
||||
$flags = $flags ?: Ev::FLAG_AUTO;
|
||||
$this->loop = new EvLoop($flags);
|
||||
$this->onCoroutineResolution = function ($e = null, $r = null) {
|
||||
if ($e) {
|
||||
$this->onCallbackError($e);
|
||||
}
|
||||
};
|
||||
self::$instanceCount++;
|
||||
}
|
||||
|
||||
public function __destruct() {
|
||||
self::$instanceCount--;
|
||||
foreach (array_keys($this->enabledWatchers) as $watcherId) {
|
||||
$this->cancel($watcherId);
|
||||
}
|
||||
foreach (array_keys($this->disabledWatchers) as $watcherId) {
|
||||
$this->cancel($watcherId);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* {@inheritdoc}
|
||||
*/
|
||||
public function run(callable $onStart = null) {
|
||||
if ($this->isRunning) {
|
||||
return;
|
||||
}
|
||||
|
||||
$this->isRunning = true;
|
||||
if ($onStart) {
|
||||
$this->immediately($onStart);
|
||||
}
|
||||
|
||||
while ($this->isRunning) {
|
||||
$immediates = $this->enabledImmediates;
|
||||
foreach ($immediates as $watcherId => $immediateArr) {
|
||||
list($callback, $cbData) = $immediateArr;
|
||||
if (!$this->tryImmediate($watcherId, $callback, $cbData)) {
|
||||
break;
|
||||
}
|
||||
}
|
||||
if (!($this->enabledWatchers || $this->enabledImmediates)) {
|
||||
break;
|
||||
}
|
||||
$flags = $this->enabledImmediates ? (Ev::RUN_ONCE | Ev::RUN_NOWAIT) : Ev::RUN_ONCE;
|
||||
$this->loop->run($flags);
|
||||
}
|
||||
|
||||
if ($this->stopException) {
|
||||
$e = $this->stopException;
|
||||
$this->stopException = null;
|
||||
throw $e;
|
||||
}
|
||||
}
|
||||
|
||||
private function tryImmediate($watcherId, $callback, $cbData) {
|
||||
try {
|
||||
unset($this->enabledImmediates[$watcherId]);
|
||||
$out = \call_user_func($callback, $this, $watcherId, $cbData);
|
||||
if ($out instanceof \Generator) {
|
||||
resolve($out, $this)->when($this->onCoroutineResolution);
|
||||
}
|
||||
} catch (\Exception $e) {
|
||||
$this->onCallbackError($e);
|
||||
}
|
||||
|
||||
return $this->isRunning;
|
||||
}
|
||||
|
||||
private function onCallbackError(\Exception $e) {
|
||||
if (empty($this->onError)) {
|
||||
$this->stopException = $e;
|
||||
$this->stop();
|
||||
} else {
|
||||
$this->tryUserErrorCallback($e);
|
||||
}
|
||||
}
|
||||
|
||||
private function tryUserErrorCallback(\Exception $e) {
|
||||
try {
|
||||
\call_user_func($this->onError, $e);
|
||||
} catch (\Exception $e) {
|
||||
$this->stopException = $e;
|
||||
$this->stop();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* {@inheritdoc}
|
||||
*/
|
||||
public function tick($noWait = false) {
|
||||
if ($this->isRunning) {
|
||||
// If we're already running/ticking this method is superfluous
|
||||
return;
|
||||
}
|
||||
$noWait = (bool) $noWait;
|
||||
$this->isRunning = true;
|
||||
|
||||
$immediates = $this->enabledImmediates;
|
||||
foreach ($immediates as $watcherId => $immediateArr) {
|
||||
list($callback, $cbData) = $immediateArr;
|
||||
if (!$this->tryImmediate($watcherId, $callback, $cbData)) {
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if ($this->isRunning) {
|
||||
$flags = $noWait || $this->enabledImmediates ? Ev::RUN_NOWAIT | Ev::RUN_ONCE : Ev::RUN_ONCE;
|
||||
$this->loop->run($flags);
|
||||
$this->isRunning = false;
|
||||
}
|
||||
|
||||
if ($this->stopException) {
|
||||
$e = $this->stopException;
|
||||
$this->stopException = null;
|
||||
throw $e;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* {@inheritDoc}
|
||||
*/
|
||||
public function stop() {
|
||||
$this->loop->stop();
|
||||
$this->isRunning = false;
|
||||
}
|
||||
|
||||
/**
|
||||
* {@inheritdoc}
|
||||
*/
|
||||
public function immediately(callable $callback, array $options = []) {
|
||||
$watcherId = $this->lastWatcherId++;
|
||||
$cbData = isset($options["cb_data"]) ? $options["cb_data"] : null;
|
||||
$isEnabled = isset($options["enable"]) ? (bool) $options["enable"] : true;
|
||||
if ($isEnabled) {
|
||||
$this->enabledImmediates[$watcherId] = [$callback, $cbData];
|
||||
} else {
|
||||
$this->disabledImmediates[$watcherId] = [$callback, $cbData];
|
||||
}
|
||||
|
||||
return $watcherId;
|
||||
}
|
||||
|
||||
/**
|
||||
* {@inheritdoc}
|
||||
*/
|
||||
public function once(callable $callback, $msDelay, array $options = []) {
|
||||
$type = Watcher::TIMER_ONCE;
|
||||
$msInterval = 0.0;
|
||||
// @TODO assert($msDelay > 0)
|
||||
|
||||
return $this->registerTimer($type, $callback, $msDelay, $msInterval, $options);
|
||||
}
|
||||
|
||||
/**
|
||||
* {@inheritdoc}
|
||||
*/
|
||||
public function repeat(callable $callback, $msInterval, array $options = []) {
|
||||
$type = Watcher::TIMER_REPEAT;
|
||||
$msDelay = isset($options["ms_delay"]) ? $options["ms_delay"] : $msInterval;
|
||||
if ($msInterval == 0) {
|
||||
$msInterval = 1;
|
||||
}
|
||||
|
||||
return $this->registerTimer($type, $callback, $msDelay, $msInterval, $options);
|
||||
}
|
||||
|
||||
private function registerTimer($type, $callback, $msDelay, $msInterval, $options = []) {
|
||||
// ev uses one second resolution
|
||||
$msDelay = $msDelay/1000;
|
||||
$msInterval = $msInterval/1000;
|
||||
$watcherId = $this->lastWatcherId++;
|
||||
$wrappedCallback = $this->wrapWatcherCallback($type, $watcherId, $callback);
|
||||
$cbData = isset($options["cb_data"]) ? $options["cb_data"] : null;
|
||||
$isEnabled = isset($options["enable"]) ? (bool) $options["enable"] : true;
|
||||
$watcher = $this->loop->timer($msDelay, $msInterval, $wrappedCallback, $cbData);
|
||||
if ($isEnabled) {
|
||||
$this->enabledWatchers[$watcherId] = $watcher;
|
||||
} else {
|
||||
$watcher->stop();
|
||||
$this->disabledWatchers[$watcherId] = $watcher;
|
||||
}
|
||||
|
||||
return $watcherId;
|
||||
}
|
||||
|
||||
private function wrapWatcherCallback($type, $watcherId, $callback, $stream = null) {
|
||||
return function($evHandle) use ($type, $watcherId, $callback, $stream) {
|
||||
try {
|
||||
if ($type === Watcher::TIMER_ONCE) {
|
||||
unset($this->enabledWatchers[$watcherId]);
|
||||
$out = \call_user_func($callback, $this, $watcherId, $evHandle->data);
|
||||
} elseif ($type & Watcher::IO) {
|
||||
$out = \call_user_func($callback, $this, $watcherId, $stream, $evHandle->data);
|
||||
} else {
|
||||
$out = \call_user_func($callback, $this, $watcherId, $evHandle->data);
|
||||
}
|
||||
if ($out instanceof \Generator) {
|
||||
resolve($out, $this)->when($this->onCoroutineResolution);
|
||||
}
|
||||
} catch (\Exception $e) {
|
||||
$this->onCallbackError($e);
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* {@inheritdoc}
|
||||
*/
|
||||
public function onReadable($stream, callable $callback, array $options = []) {
|
||||
return $this->registerIo($events = Ev::READ, $stream, $callback, $options);
|
||||
}
|
||||
|
||||
/**
|
||||
* {@inheritdoc}
|
||||
*/
|
||||
public function onWritable($stream, callable $callback, array $options = []) {
|
||||
return $this->registerIo($events = Ev::WRITE, $stream, $callback, $options);
|
||||
}
|
||||
|
||||
private function registerIo($events, $stream, $callback, $options) {
|
||||
$watcherId = $this->lastWatcherId++;
|
||||
$type = Watcher::IO;
|
||||
$wrappedCallback = $this->wrapWatcherCallback($type, $watcherId, $callback, $stream);
|
||||
$cbData = isset($options["cb_data"]) ? $options["cb_data"] : null;
|
||||
$isEnabled = isset($options["enable"]) ? (bool) $options["enable"] : true;
|
||||
|
||||
$watcher = $this->loop->io($stream, $events, $wrappedCallback, $cbData);
|
||||
if ($isEnabled) {
|
||||
$this->enabledWatchers[$watcherId] = $watcher;
|
||||
} else {
|
||||
$watcher->stop();
|
||||
$this->disabledWatchers[$watcherId] = $watcher;
|
||||
}
|
||||
|
||||
return $watcherId;
|
||||
}
|
||||
|
||||
/**
|
||||
* {@inheritDoc}
|
||||
*/
|
||||
public function onSignal($signo, callable $callback, array $options = []) {
|
||||
$watcherId = $this->lastWatcherId++;
|
||||
$type = Watcher::SIGNAL;
|
||||
$wrappedCallback = $this->wrapWatcherCallback($type, $watcherId, $callback);
|
||||
$cbData = isset($options["cb_data"]) ? $options["cb_data"] : null;
|
||||
$isEnabled = isset($options["enable"]) ? (bool) $options["enable"] : true;
|
||||
|
||||
$watcher = $this->loop->signal($signo, $wrappedCallback, $cbData);
|
||||
if ($isEnabled) {
|
||||
$this->enabledWatchers[$watcherId] = $watcher;
|
||||
} else {
|
||||
$timer->stop();
|
||||
$this->disabledWatchers[$watcherId] = $watcher;
|
||||
}
|
||||
|
||||
return $watcherId;
|
||||
}
|
||||
|
||||
/**
|
||||
* {@inheritdoc}
|
||||
*/
|
||||
public function cancel($watcherId) {
|
||||
if (isset($this->enabledWatchers[$watcherId])) {
|
||||
$watcher = $this->enabledWatchers[$watcherId];
|
||||
$watcher->clear();
|
||||
$watcher->stop();
|
||||
unset($this->enabledWatchers[$watcherId]);
|
||||
} elseif (isset($this->disabledWatchers[$watcherId])) {
|
||||
$watcher = $this->disabledWatchers[$watcherId];
|
||||
$watcher->clear();
|
||||
unset($this->disabledWatchers[$watcherId]);
|
||||
} elseif (isset($this->enabledImmediates[$watcherId])) {
|
||||
$watcher = $this->enabledImmediates[$watcherId];
|
||||
unset($this->enabledImmediates[$watcherId]);
|
||||
} elseif (isset($this->disabledImmediates[$watcherId])) {
|
||||
$watcher = $this->disabledImmediates[$watcherId];
|
||||
unset($this->disabledImmediates[$watcherId]);
|
||||
} else {
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* {@inheritdoc}
|
||||
*/
|
||||
public function disable($watcherId) {
|
||||
if (isset($this->enabledWatchers[$watcherId])) {
|
||||
$evHandle = $this->enabledWatchers[$watcherId];
|
||||
unset($this->enabledWatchers[$watcherId]);
|
||||
$this->disabledWatchers[$watcherId] = $evHandle;
|
||||
$evHandle->stop();
|
||||
} elseif (isset($this->enabledImmediates[$watcherId])) {
|
||||
$immediateArr = $this->enabledImmediates[$watcherId];
|
||||
$this->disabledImmediates[$watcherId] = $immediateArr;
|
||||
unset($this->enabledImmediates[$watcherId]);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* {@inheritdoc}
|
||||
*/
|
||||
public function enable($watcherId) {
|
||||
if (isset($this->disabledWatchers[$watcherId])) {
|
||||
$evHandle = $this->disabledWatchers[$watcherId];
|
||||
unset($this->disabledWatchers[$watcherId]);
|
||||
$this->enabledWatchers[$watcherId] = $evHandle;
|
||||
$evHandle->start();
|
||||
} elseif (isset($this->disabledImmediates[$watcherId])) {
|
||||
$immediateArr = $this->disabledImmediates[$watcherId];
|
||||
$this->enabledImmediates[$watcherId] = $immediateArr;
|
||||
unset($this->disabledImmediates[$watcherId]);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* {@inheritdoc}
|
||||
*/
|
||||
public function onError(callable $callback) {
|
||||
$this->onError = $callback;
|
||||
}
|
||||
|
||||
/**
|
||||
* {@inheritdoc}
|
||||
*/
|
||||
public function getUnderlyingLoop() {
|
||||
return $this->loop;
|
||||
}
|
||||
}
|
||||
@ -2,7 +2,7 @@
|
||||
|
||||
namespace Amp;
|
||||
|
||||
interface SignalReactor extends Reactor {
|
||||
interface ExtensionReactor extends Reactor {
|
||||
/**
|
||||
* React to process control signals
|
||||
*
|
||||
@ -12,4 +12,11 @@ interface SignalReactor extends Reactor {
|
||||
* @return string Returns unique (to the process) string watcher ID
|
||||
*/
|
||||
public function onSignal($signo, callable $func, array $options = []);
|
||||
|
||||
/**
|
||||
* Retrieve the underlying event loop representation
|
||||
*
|
||||
* @return mixed
|
||||
*/
|
||||
public function getUnderlyingLoop();
|
||||
}
|
||||
@ -14,8 +14,12 @@ function reactor(Reactor $assignReactor = null) {
|
||||
return ($reactor = $assignReactor);
|
||||
} elseif ($reactor) {
|
||||
return $reactor;
|
||||
} elseif (\extension_loaded('uv')) {
|
||||
} elseif (\extension_loaded("uv")) {
|
||||
return ($reactor = new UvReactor);
|
||||
} elseif (\extension_loaded("ev")) {
|
||||
return ($reactor = new EvReactor);
|
||||
} elseif (\extension_loaded("libevent")) {
|
||||
return ($reactor = new LibeventReactor);
|
||||
} else {
|
||||
return ($reactor = new NativeReactor);
|
||||
}
|
||||
|
||||
17
test/EvReactorTest.php
Normal file
17
test/EvReactorTest.php
Normal file
@ -0,0 +1,17 @@
|
||||
<?php
|
||||
|
||||
namespace Amp\Test;
|
||||
|
||||
use Amp\EvReactor;
|
||||
|
||||
class EvReactorTest extends ReactorTest {
|
||||
protected function getReactor() {
|
||||
if (extension_loaded("ev")) {
|
||||
return new EvReactor();
|
||||
} else {
|
||||
$this->markTestSkipped(
|
||||
"ev extension not loaded"
|
||||
);
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
x
Reference in New Issue
Block a user