本文首發於 PHP 多工協程處理,轉載請註明出處!
上週 有幸和同事一起在 SilverStripe 分享最近的工作事宜。今天我計劃分享 PHP 非同步程式設計,不過由於上週我聊過 ReactPHP;我決定討論一些不一樣的內容。所以本文將探討多工協程這方面的內容。
另外我還計劃把這個主題加入到我正在籌備的一本 PHP 非同步程式設計的圖書中。雖然這本書相比本文來說會涉及更多細節,但我覺得本文依然具有實際意義!
那麼,開始吧!
這就是本文我們要討論的問題。不過我們會從更簡單更熟悉的示例開始。
一切從陣列開始
我們可以通過簡單的遍歷來使用陣列:
$array = ["foo", "bar", "baz"];
foreach ($array as $key => $value) {
print "item: " . $key . "|" . $value . "\n";
}
for ($i = 0; $i < count($array); $i++) {
print "item: " . $i . "|" . $array[$i] . "\n";
}
這是我們日常編碼所依賴的基本實現。可以通過遍歷陣列獲取每個元素的鍵名和鍵值。
當然,如果我們希望能夠知道在何時可以使用陣列。PHP 提供了一個方便的內建函式:
print is_array($array) ? "yes" : "no"; // yes
類陣列處理
有時,我們需要對一些資料使用相同的方式進行遍歷處理,但它們並非陣列型別。比如對 DOMDocument 類進行處理:
$document = new DOMDocument();
$document->loadXML("<div></div>");
$elements = $document->getElementsByTagName("div");
print_r($elements); // DOMNodeList Object ( [length] => 1 )
這顯然不是一個陣列,但是它有一個 length 屬性。我們能像遍歷陣列一樣,對其進行遍歷麼?我們可以判斷它是否實現了下面這個特殊的介面:
print ($elements instanceof Traversable) ? "yes" : "no"; // yes
這真的太有用了。它不會導致我們在遍歷非可遍歷資料時觸發錯誤。我們僅需在處理前進行檢測即可。
不過,這會引發另外一個問題:我們能否讓自定義類也擁有這個功能呢?回答是肯定的!第一個實現方法類似如下:
class MyTraversable implements Traversable
{
// 在這裡編碼...
}
如果我們執行這個類,我們將看到一個錯誤資訊:
PHP Fatal error: Class MyTraversable must implement interface Traversable as part of either Iterator or IteratorAggregate
Iterator(迭代器)
我們無法直接實現 Traversable,但是我們可以嘗試第二種方案:
class MyTraversable implements Iterator
{
// 在這裡編碼...
}
這個介面需要我們實現 5 個方法。讓我們完善我們的迭代器:
class MyTraversable implements Iterator
{
protected $data;
protected $index = 0;
public function __construct($data)
{
$this->data = $data;
}
public function current()
{
return $this->data[$this->index];
}
public function next()
{
return $this->data[$this->index++];
}
public function key()
{
return $this->index;
}
public function rewind()
{
$this->index = 0;
}
public function valid()
{
return $this->index < count($this->data);
}
}
這邊我們需要注意幾個事項:
-
我們需要儲存構造器方法傳入的 $data 陣列,以便後續我們可以從中獲取它的元素。
-
還需要一個內部索引(或指標)來跟蹤 current 或 next 元素。
-
rewind() 僅僅重置 index 屬性,這樣 current() 和 next() 才能正常工作。
- 鍵名並非只能是數字型別!這裡使用陣列索引是為了保證示例足夠簡單。
我們可以向下面這樣執行這段程式碼:
$iterator = new MyIterator(["foo", "bar", "baz"]);
foreach ($iterator as $key => $value) {
print "item: " . $key . "|" . $value . "\n";
}
這看起來需要處理太多工作,但是這是能夠像陣列一樣使用 foreach/for 功能的一個簡潔實現。
IteratorAggregate(聚合迭代器)
還記得第二個介面丟擲的 Traversable 異常麼?下面看一個比實現 Iterator 介面更快的實現吧:
class MyIteratorAggregate implements IteratorAggregate
{
protected $data;
public function __construct($data)
{
$this->data = $data;
}
public function getIterator()
{
return new ArrayIterator($this->data);
}
}
這裡我們作弊了。相比於實現一個完整的 Iterator,我們通過 ArrayIterator() 裝飾。不過,這相比於通過實現完整的 Iterator 簡化了不少程式碼。
兄弟莫急!先讓我們比較一些程式碼。首先,我們在不使用生成器的情況下從檔案中讀取每一行資料:
$content = file_get_contents(__FILE__);
$lines = explode("\n", $content);
foreach ($lines as $i => $line) {
print $i . ". " . $line . "\n";
}
這段程式碼讀取檔案自身,然後會列印出每行的行號和程式碼。那麼為什麼我們不使用生成器呢!
function lines($file) {
$handle = fopen($file, 'r');
while (!feof($handle)) {
yield trim(fgets($handle));
}
fclose($handle);
}
foreach (lines(__FILE__) as $i => $line) {
print $i . ". " . $line . "\n";
}
我知道這看起來更加複雜。不錯,不過這是因為我們沒有使用 file_get_contents() 函式。一個生成器看起來就像是一個函式,但是它會在每次獲取到 yield 關鍵詞是停止執行。
生成器看起來有點像迭代器:
print_r(lines(__FILE__)); // Generator Object ( )
儘管它不是迭代器,它是一個 Generator。它的內部定義了什麼方法呢?
print_r(get_class_methods(lines(__FILE__)));
// Array
// (
// [0] => rewind
// [1] => valid
// [2] => current
// [3] => key
// [4] => next
// [5] => send
// [6] => throw
// [7] => __wakeup
// )
如果你讀取一個大檔案,然後使用 memory_get_peak_usage(),你會注意到生成器的程式碼會使用固定的記憶體,無論這個檔案有多大。它每次進度去一行。而是用 file_get_contents() 函式讀取整個檔案,會使用更大的記憶體。這就是在迭代處理這類事物時,生成器的能給我們帶來的優勢!
Send(傳送資料)
可以將資料傳送到生成器中。看下下面這個生成器:
<?php
$generator = call_user_func(function() {
yield "foo";
});
print $generator->current() . "\n"; // foo
注意這裡我們如何在 call_user_func() 函式中封裝生成器函式的?這裡僅僅是一個簡單的函式定義,然後立即呼叫它獲取一個新的生成器例項...
我們已經見過 yield 的用法。我們可以通過擴充套件這個生成器來接收資料:
$generator = call_user_func(function() {
$input = (yield "foo");
print "inside: " . $input . "\n";
});
print $generator->current() . "\n";
$generator->send("bar");
資料通過 yield 關鍵字傳入和返回。首先,執行 current() 程式碼直到遇到 yield,返回 foo。send() 將輸出傳入到生成器列印輸入的位置。你需要習慣這種用法。
丟擲異常(Throw)
由於我們需要同這些函式進行互動,可能希望將異常推送到生成器中。這樣這些函式就可以自行處理異常。
看看下面這個示例:
$multiply = function($x, $y) {
yield $x * $y;
};
print $multiply(5, 6)->current(); // 30
現在讓我們將它封裝到另一個函式中:
$calculate = function ($op, $x, $y) use ($multiply) {
if ($op === 'multiply') {
$generator = $multiply($x, $y);
return $generator->current();
}
};
print $calculate("multiply", 5, 6); // 30
這裡我們通過一個普通閉包將乘法生成器封裝起來。現在讓我們驗證無效引數:
$calculate = function ($op, $x, $y) use ($multiply) {
if ($op === "multiply") {
$generator = $multiply($x, $y);
if (!is_numeric($x) || !is_numeric($y)) {
throw new InvalidArgumentException();
}
return $generator->current();
}
};
print $calculate('multiply', 5, 'foo'); // PHP Fatal error...
如果我們希望能夠通過生成器處理異常?我們怎樣才能將異常傳入生成器呢!
$multiply = function ($x, $y) {
try {
yield $x * $y;
} catch (InvalidArgumentException $exception) {
print "ERRORS!";
}
};
$calculate = function ($op, $x, $y) use ($multiply) {
if ($op === "multiply") {
$generator = $multiply($x, $y);
if (!is_numeric($x) || !is_numeric($y)) {
$generator->throw(new InvalidArgumentException());
}
return $generator->current();
}
};
print $calculate('multiply', 5, 'foo'); // PHP Fatal error...
棒呆了!我們不僅可以像迭代器一樣使用生成器。還可以通過它們傳送資料並丟擲異常。它們是可中斷和可恢復的函式。有些語言把這些函式叫做……
我們可以使用協程(coroutines)來構建非同步程式碼。讓我們來建立一個簡單的任務排程程式。首先我們需要一個 Task 類:
class Task
{
protected $generator;
public function __construct(Generator $generator)
{
$this->generator = $generator;
}
public function run()
{
$this->generator->next();
}
public function finished()
{
return !$this->generator->valid();
}
}
Task 是普通生成器的裝飾器。我們將生成器賦值給它的成員變數以供後續使用,然後實現一個簡單的 run() 和 finished() 方法。run() 方法用於執行任務,finished() 方法用於讓排程程式知道何時終止執行。
然後我們需要一個 Scheduler 類:
class Scheduler
{
protected $queue;
public function __construct()
{
$this->queue = new SplQueue();
}
public function enqueue(Task $task)
{
$this->queue->enqueue($task);
}
pulic function run()
{
while (!$this->queue->isEmpty()) {
$task = $this->queue->dequeue();
$task->run();
if (!$task->finished()) {
$this->queue->enqueue($task);
}
}
}
}
Scheduler 用於維護一個待執行的任務佇列。run() 會彈出佇列中的所有任務並執行它,直到執行完整個佇列任務。如果某個任務沒有執行完畢,當這個任務本次執行完成後,我們將再次入列。
SplQueue 對於這個示例來講再合適不過了。它是一種 FIFO(先進先出:fist in first out) 資料結構,能夠確保每個任務都能夠獲取足夠的處理時間。
我們可以像這樣執行這段程式碼:
$scheduler = new Scheduler();
$task1 = new Task(call_user_func(function() {
for ($i = 0; $i < 3; $i++) {
print "task1: " . $i . "\n";
yield;
}
}));
$task2 = new Task(call_user_func(function() {
for ($i = 0; $i < 6; $i++) {
print "task2: " . $i . "\n";
yield;
}
}));
$scheduler->enqueue($task1);
$scheduler->enqueue($task2);
$scheduler->run();
執行時,我們將看到如下執行結果:
task 1: 0
task 1: 1
task 2: 0
task 2: 1
task 1: 2
task 2: 2
task 2: 3
task 2: 4
task 2: 5
這幾乎就是我們想要的執行結果。不過有個問題發生在首次執行每個任務時,它們都執行了兩次。我們可以對 Task 類稍作修改來修復這個問題:
class Task
{
protected $generator;
protected $run = false;
public function __construct(Generator $generator)
{
$this->generator = $generator;
}
public function run()
{
if ($this->run) {
$this->generator->next();
} else {
$this->generator->current();
}
$this->run = true;
}
public function finished()
{
return !$this->generator->valid();
}
}
我們需要調整首次 run() 方法呼叫,從生成器當前有效的指標讀取執行。後續呼叫可以從下一個指標讀取執行...
有些人基於這個思路實現了一些超讚的類庫。我們來看看其中的兩個...
RecoilPHP
RecoilPHP 是一套基於協程的類庫,它最令人印象深刻的是用於 ReactPHP 核心。可以將事件迴圈在 RecoilPHP 和 RecoilPHP 之間進行交換,而你的程式無需架構上的調整。
我們來看一下 ReactPHP 非同步 DNS 解決方案:
function resolve($domain, $resolver) {
$resolver
->resolve($domain)
->then(function ($ip) use ($domain) {
print "domain: " . $domain . "\n";
print "ip: " . $ip . "\n";
}, function ($error) {
print $error . "\n";
})
}
function run()
{
$loop = React\EventLoop\Factory::create();
$factory = new React\Dns\Resolver\Factory();
$resolver = $factory->create("8.8.8.8", $loop);
resolve("silverstripe.org", $resolver);
resolve("wordpress.org", $resolver);
resolve("wardrobecms.com", $resolver);
resolve("pagekit.com", $resolver);
$loop->run();
}
run();
resolve() 接收域名和 DNS 解析器,並使用 ReactPHP 執行標準的 DNS 查詢。不用太過糾結與 resolve() 函式內部。重要的是這個函式不是生成器,而是一個函式!
run() 建立一個 ReactPHP 事件迴圈,DNS 解析器(這裡是個工廠例項)解析若干域名。同樣,這個也不是一個生成器。
想知道 RecoilPHP 到底有何不同?還希望掌握更多細節!
use Recoil\Recoil;
function resolve($domain, $resolver)
{
try {
$ip = (yield $resolver->resolve($domain));
print "domain: " . $domain . "\n";
print "ip: " . $ip . "\n";
} catch (Exception $exception) {
print $exception->getMessage() . "\n";
}
}
function run()
{
$loop = (yield Recoil::eventLoop());
$factory = new React\Dns\Resolver\Factory();
$resolver = $factory->create("8.8.8.8", $loop);
yield [
resolve("silverstripe.org", $resolver),
resolve("wordpress.org", $resolver),
resolve("wardrobecms.com", $resolver),
resolve("pagekit.com", $resolver),
];
}
Recoil::run("run");
通過將它整合到 ReactPHP 來完成一些令人稱奇的工作。每次執行 resolve() 時,RecoilPHP 會管理由 $resoler->resolve() 返回的 promise 物件,然後將資料傳送給生成器。此時我們就像在編寫同步程式碼一樣。與我們在其他一步模型中使用回撥程式碼不同,這裡只有一個指令列表。
RecoilPHP 知道它應該管理一個有執行 run() 函式時返回的 yield 陣列。RoceilPHP 還支援基於協程的資料庫(PDO)和日誌庫。
IcicleIO
IcicleIO 為了一全新的方案實現 ReactPHP 一樣的目標,而僅僅使用協程功能。相比 ReactPHP 它僅包含極少的元件。但是,核心的非同步流、伺服器、Socket、事件迴圈特性一個不落。
讓我們看一個 socket 伺服器示例:
use Icicle\Coroutine\Coroutine;
use Icicle\Loop\Loop;
use Icicle\Socket\Client\ClientInterface;
use Icicle\Socket\Server\ServerInterface;
use Icicle\Socket\Server\ServerFactory;
$factory = new ServerFactory();
$coroutine = Coroutine::call(function (ServerInterface $server) {
$clients = new SplObjectStorage();
$handler = Coroutine::async(
function (ClientInterface $client) use (&$clients) {
$clients->attach($client);
$host = $client->getRemoteAddress();
$port = $client->getRemotePort();
$name = $host . ":" . $port;
try {
foreach ($clients as $stream) {
if ($client !== $stream) {
$stream->write($name . "connected.\n");
}
}
yield $client->write("Welcome " . $name . "!\n");
while ($client->isReadable()) {
$data = trim(yield $client->read());
if ("/exit" === $data) {
yield $client->end("Goodbye!\n");
} else {
$message = $name . ":" . $data . "\n";
foreach ($clients as $stream) {
if ($client !== $stream) {
$stream->write($message);
}
}
}
}
} catch (Exception $exception) {
$client->close($exception);
} finally {
$clients->detach($client);
foreach ($clients as $stream) {
$stream->write($name . "disconnected.\n");
}
}
}
);
while ($server->isOpen()) {
$handler(yield $server->accept());
}
}, $factory->create("127.0.0.1", 6000));
Loop::run();
據我所知,這段程式碼所做的事情如下:
-
在 127.0.0.1 和 6000 埠建立一個伺服器例項,然後將其傳入外部生成器.
-
外部生成器執行,同時伺服器等待新連線。當伺服器接收一個連線它將其傳入內部生成器。
-
內部生成器寫入訊息到 socket。當 socket 可讀時執行。
- 每次 socket 向伺服器傳送訊息時,內部生成器檢測訊息是否是退出標識。如果是,通知其他 socket。否則,其它 socket 傳送這個相同的訊息。
開啟命令列終端輸入 nc localhost 6000 檢視執行結果!
該示例使用 SplObjectStorage 跟蹤 socket 連線。這樣我們就可以向所有 socket 傳送訊息。
這個話題可以包含很多內容。希望您能看到生成器是如何建立的,以及它們如何幫助編寫迭代程式和非同步程式碼。
如果你有問題,可以隨時問我。
感謝 Nikita Popov(還有它的啟蒙教程 Cooperative multitasking using coroutines (in PHP!) ),Anthony Ferrara 和 Joe Watkins。這些研究工作澤被蒼生,給我以寫作此篇文章的靈感。關注他們吧,好麼?