1
0
mirror of https://github.com/danog/file.git synced 2024-11-26 20:04:51 +01:00
file/lib/EioDriver.php
2015-08-10 12:23:58 -04:00

481 lines
13 KiB
PHP

<?php
namespace Amp\File;
use Amp\Promise;
use Amp\Success;
use Amp\Failure;
use Amp\Deferred;
class EioDriver implements Driver {
private $watcher;
private $callableDelReq;
private $pending = 0;
private static $stream;
/**
* We have to keep a static reference of eio event streams
* because if we don't garbage collection can unload eio's
* underlying pipe via a system close() call before it's
* finished and generate a SIGPIPE.
*/
public function __construct() {
if (empty(self::$stream)) {
\eio_init();
self::$stream = \eio_get_event_stream();
}
$this->callableDelReq = function() {
$this->decrementPending();
};
$this->watcher = \Amp\onReadable(self::$stream, function() {
while (\eio_npending()) {
\eio_poll();
}
}, $options = ["enable" => false]);
}
private function incrementPending() {
if ($this->pending++ === 0) {
\Amp\enable($this->watcher);
}
}
private function decrementPending() {
if ($this->pending-- === 1) {
\Amp\disable($this->watcher);
}
}
/**
* {@inheritdoc}
*/
public function stat($path) {
if ($stat = StatCache::get($path)) {
return new Success($stat);
}
$this->incrementPending();
$promisor = new Deferred;
$priority = \EIO_PRI_DEFAULT;
$data = [$promisor, $path];
\eio_stat($path, $priority, [$this, "onStat"], $data);
return $promisor->promise();
}
private function onStat($data, $result, $req) {
list($promisor, $path) = $data;
$this->decrementPending();
if ($result === -1) {
$promisor->succeed(null);
} else {
StatCache::set($path, $result);
$promisor->succeed($result);
}
}
/**
* {@inheritdoc}
*/
public function exists($path) {
$promisor = new Deferred;
$this->stat($path)->when(function ($error, $result) use ($promisor) {
$promisor->succeed((bool) $result);
});
return $promisor->promise();
}
/**
* {@inheritdoc}
*/
public function isdir($path) {
$promisor = new Deferred;
$this->stat($path)->when(function ($error, $result) use ($promisor) {
if ($result) {
$promisor->succeed(!($result["mode"] & \EIO_S_IFREG));
} else {
$promisor->succeed(false);
}
});
return $promisor->promise();
}
/**
* {@inheritdoc}
*/
public function isfile($path) {
$promisor = new Deferred;
$this->stat($path)->when(function ($error, $result) use ($promisor) {
if ($result) {
$promisor->succeed((bool) ($result["mode"] & \EIO_S_IFREG));
} else {
$promisor->succeed(false);
}
});
return $promisor->promise();
}
/**
* {@inheritdoc}
*/
public function size($path) {
$promisor = new Deferred;
$this->stat($path)->when(function ($error, $result) use ($promisor) {
if (empty($result)) {
$promisor->fail(new FilesystemException(
"Specified path does not exist"
));
} elseif (($result["mode"] & \EIO_S_IFREG)) {
$promisor->succeed($result["size"]);
} else {
$promisor->fail(new FilesystemException(
"Specified path is not a regular file"
));
}
});
return $promisor->promise();
}
/**
* {@inheritdoc}
*/
public function mtime($path) {
$promisor = new Deferred;
$this->stat($path)->when(function ($error, $result) use ($promisor) {
if ($result) {
$promisor->succeed($result["mtime"]);
} else {
$promisor->fail(new FilesystemException(
"Specified path does not exist"
));
}
});
return $promisor->promise();
}
/**
* {@inheritdoc}
*/
public function atime($path) {
$promisor = new Deferred;
$this->stat($path)->when(function ($error, $result) use ($promisor) {
if ($result) {
$promisor->succeed($result["atime"]);
} else {
$promisor->fail(new FilesystemException(
"Specified path does not exist"
));
}
});
return $promisor->promise();
}
/**
* {@inheritdoc}
*/
public function ctime($path) {
$promisor = new Deferred;
$this->stat($path)->when(function ($error, $result) use ($promisor) {
if ($result) {
$promisor->succeed($result["ctime"]);
} else {
$promisor->fail(new FilesystemException(
"Specified path does not exist"
));
}
});
return $promisor->promise();
}
/**
* {@inheritdoc}
*/
public function lstat($path) {
$this->incrementPending();
$promisor = new Deferred;
$priority = \EIO_PRI_DEFAULT;
\eio_lstat($path, $priority, [$this, "onLstat"], $promisor);
return $promisor->promise();
}
private function onLstat($promisor, $result, $req) {
$this->decrementPending();
if ($result === -1) {
$promisor->succeed(null);
} else {
$promisor->succeed($result);
}
}
/**
* {@inheritdoc}
*/
public function symlink($target, $link) {
$this->incrementPending();
$promisor = new Deferred;
$priority = \EIO_PRI_DEFAULT;
\eio_symlink($target, $link, $priority, [$this, "onGenericResult"], $promisor);
return $promisor->promise();
}
private function onGenericResult($promisor, $result, $req) {
$this->decrementPending();
if ($result === -1) {
$promisor->fail(new FilesystemException(
\eio_get_last_error($req)
));
} else {
$promisor->succeed(true);
}
}
/**
* {@inheritdoc}
*/
public function rename($from, $to) {
$this->incrementPending();
$promisor = new Deferred;
$priority = \EIO_PRI_DEFAULT;
\eio_rename($from, $to, $priority, [$this, "onGenericResult"], $promisor);
return $promisor->promise();
}
/**
* {@inheritdoc}
*/
public function unlink($path) {
$this->incrementPending();
$promisor = new Deferred;
$priority = \EIO_PRI_DEFAULT;
$data = [$promisor, $path];
\eio_unlink($path, $priority, [$this, "onUnlink"], $data);
return $promisor->promise();
}
private function onUnlink($data, $result, $req) {
list($promisor, $path) = $data;
$this->decrementPending();
if ($result === -1) {
$promisor->fail(new FilesystemException(
\eio_get_last_error($req)
));
} else {
StatCache::clear($path);
$promisor->succeed(true);
}
}
/**
* {@inheritdoc}
*/
public function mkdir($path, $mode = 0644) {
$this->incrementPending();
$promisor = new Deferred;
$priority = \EIO_PRI_DEFAULT;
\eio_mkdir($path, $mode, $priority, [$this, "onGenericResult"], $promisor);
return $promisor->promise();
}
/**
* {@inheritdoc}
*/
public function rmdir($path) {
$this->incrementPending();
$promisor = new Deferred;
$priority = \EIO_PRI_DEFAULT;
$data = [$promisor, $path];
\eio_rmdir($path, $priority, [$this, "onRmdir"], $data);
return $promisor->promise();
}
private function onRmdir($data, $result, $req) {
list($promisor, $path) = $data;
$this->decrementPending();
if ($result === -1) {
$promisor->fail(new FilesystemException(
\eio_get_last_error($req)
));
} else {
StatCache::clear($path);
$promisor->succeed(true);
}
}
/**
* {@inheritdoc}
*/
public function scandir($path) {
$this->incrementPending();
$promisor = new Deferred;
$flags = \EIO_READDIR_STAT_ORDER | \EIO_READDIR_DIRS_FIRST;
$priority = \EIO_PRI_DEFAULT;
\eio_readdir($path, $flags, $priority, [$this, "onScandir"], $promisor);
return $promisor->promise();
}
private function onScandir($promisor, $result, $req) {
$this->decrementPending();
if ($result === -1) {
$promisor->fail(new FilesystemException(
\eio_get_last_error($req)
));
} else {
$promisor->succeed($result["names"]);
}
}
/**
* {@inheritdoc}
*/
public function chmod($path, $mode) {
$this->incrementPending();
$promisor = new Deferred;
$priority = \EIO_PRI_DEFAULT;
\eio_chmod($path, $mode, $priority, [$this, "onGenericResult"], $promisor);
return $promisor->promise();
}
/**
* {@inheritdoc}
*/
public function chown($path, $uid, $gid) {
$this->incrementPending();
$promisor = new Deferred;
$priority = \EIO_PRI_DEFAULT;
\eio_chown($path, $uid, $gid, $priority, [$this, "onGenericResult"], $promisor);
return $promisor->promise();
}
/**
* {@inheritdoc}
*/
public function touch($path) {
$atime = $mtime = time();
$promisor = new Deferred;
$priority = \EIO_PRI_DEFAULT;
\eio_utime($path, $atime, $mtime, $priority, [$this, "onGenericResult"], $promisor);
return $promisor->promise();
}
/**
* {@inheritdoc}
*/
public function get($path) {
$flags = $flags = \EIO_O_RDONLY;
$mode = 0;
$priority = \EIO_PRI_DEFAULT;
$this->incrementPending();
$promisor = new Deferred;
\eio_open($path, $flags, $mode, $priority, [$this, "onGetOpen"], $promisor);
return $promisor->promise();
}
private function onGetOpen($promisor, $result, $req) {
if ($result === -1) {
$promisor->fail(new FilesystemException(
\eio_get_last_error($req)
));
} else {
$priority = \EIO_PRI_DEFAULT;
\eio_fstat($result, $priority, [$this, "onGetFstat"], [$result, $promisor]);
}
}
private function onGetFstat($fhAndPromisor, $result, $req) {
list($fh, $promisor) = $fhAndPromisor;
if ($result === -1) {
$promisor->fail(new FilesystemException(
\eio_get_last_error($req)
));
return;
}
$offset = 0;
$length = $result["size"];
$priority = \EIO_PRI_DEFAULT;
\eio_read($fh, $length, $offset, $priority, [$this, "onGetRead"], $fhAndPromisor);
}
private function onGetRead($fhAndPromisor, $result, $req) {
list($fh, $promisor) = $fhAndPromisor;
$priority = \EIO_PRI_DEFAULT;
\eio_close($fh, $priority, $this->callableDelReq);
if ($result === -1) {
$promisor->fail(new FilesystemException(
\eio_get_last_error($req)
));
} else {
$promisor->succeed($result);
}
}
/**
* {@inheritdoc}
*/
public function put($path, $contents) {
$flags = \EIO_O_RDWR | \EIO_O_CREAT;
$mode = \EIO_S_IRUSR | \EIO_S_IWUSR | \EIO_S_IXUSR;
$priority = \EIO_PRI_DEFAULT;
$this->incrementPending();
$promisor = new Deferred;
$data = [$contents, $promisor];
\eio_open($path, $flags, $mode, $priority, [$this, "onPutOpen"], $data);
return $promisor->promise();
}
private function onPutOpen($data, $result, $req) {
list($contents, $promisor) = $data;
if ($result === -1) {
$promisor->fail(new FilesystemException(
\eio_get_last_error($req)
));
} else {
$length = strlen($contents);
$offset = 0;
$priority = \EIO_PRI_DEFAULT;
$callback = [$this, "onPutWrite"];
$fhAndPromisor = [$result, $promisor];
\eio_write($result, $contents, $length, $offset, $priority, $callback, $fhAndPromisor);
}
}
private function onPutWrite($fhAndPromisor, $result, $req) {
list($fh, $promisor) = $fhAndPromisor;
\eio_close($fh);
$priority = \EIO_PRI_DEFAULT;
\eio_close($fh, $priority, $this->callableDelReq);
if ($result === -1) {
$promisor->fail(new FilesystemException(
\eio_get_last_error($req)
));
} else {
$promisor->succeed($result);
}
}
public function __destruct() {
\Amp\cancel($this->watcher);
}
}