C++協(xié)程實(shí)現(xiàn)序列生成器的案例分享
目標(biāo)
序列生成器通常的實(shí)現(xiàn)是在一個(gè)協(xié)程內(nèi)部通過(guò)某種方式向外部傳一個(gè)值出去，并且將自己掛起，外部調(diào)用者則可以獲取到這個(gè)值，并且在后續(xù)繼續(xù)恢復(fù)執(zhí)行序列生成器來(lái)獲取下一個(gè)值。
掛起和向外部傳值的任務(wù)就需要通過(guò)co_await來(lái)完成，外部獲取值的任務(wù)就要通過(guò)協(xié)程的返回值來(lái)完成。
程序大概可以寫(xiě)成下面這樣：
Generator sequence() {
int i = 0;
while (true) {
co_await i++;
}
}
int main() {
auto generator = sequence();
for (int i = 0; i < 10; ++i) {
std::cout << generator.next() << std::endl;
}
}
注意到 generator 有個(gè) next 函數(shù)，調(diào)用它時(shí)我們需要想辦法讓協(xié)程恢復(fù)執(zhí)行，并將下一個(gè)值傳出來(lái)。
好了，接下來(lái)我們就帶著這兩個(gè)問(wèn)題去尋找解決辦法，順便把剩下的一點(diǎn)點(diǎn) C++ 協(xié)程的知識(shí)補(bǔ)齊。
調(diào)用者獲取協(xié)程產(chǎn)生的值
我們觀察main函數(shù)中的這段代碼：
int main() {
auto generator = sequence();
for (int i = 0; i < 10; ++i) {
std::cout << generator.next() << std::endl;
}
}
generator 的類(lèi)型就是我們即將實(shí)現(xiàn)的序列生成器類(lèi)型 Generator，結(jié)合上一篇文章當(dāng)中對(duì)于協(xié)程返回值類(lèi)型的介紹，我們先大致給出它的定義：
struct Generator {
struct promise_type {
// 開(kāi)始執(zhí)行時(shí)直接掛起等待外部調(diào)用 resume 獲取下一個(gè)值
std::suspend_always initial_suspend() { return {}; };
// 執(zhí)行結(jié)束后不需要掛起
std::suspend_never final_suspend() noexcept { return {}; }
// 為了簡(jiǎn)單，我們認(rèn)為序列生成器當(dāng)中不會(huì)拋出異常，這里不做任何處理
void unhandled_exception() { }
// 構(gòu)造協(xié)程的返回值類(lèi)型
Generator get_return_object() {
return Generator{};
}
// 沒(méi)有返回值
void return_void() { }
};
int next() {
???.resume();
return ???;
}
};
代碼當(dāng)中有兩處我們標(biāo)注為 ???，表示暫時(shí)還不知道怎么處理。
如果我們想要在Generator中resume協(xié)程的話，需要拿到coroutine_handle，要如何做到這一點(diǎn)呢？
這個(gè)時(shí)候可以記住一點(diǎn)，promise_type是鏈接協(xié)程內(nèi)外的橋梁，想要拿到什么，找promise_type要。標(biāo)準(zhǔn)庫(kù)提供了一個(gè)通過(guò)promise_type的對(duì)象的地址獲取coroutine_handle的函數(shù)，它實(shí)際上是coroutine_handle的一個(gè)靜態(tài)函數(shù)：
template <class _Promise>
struct coroutine_handle {
static coroutine_handle from_promise(_Promise& _Prom) noexcept {
...
}
...
}
這樣看來(lái)的話，我們只需要在get_return_object函數(shù)調(diào)用時(shí)，先獲取coroutine_handle，然后再傳給即將構(gòu)造出來(lái)的Generator就行，因此我們稍微修改一下前面的代碼：
struct Generator {
struct promise_type {
...
// 構(gòu)造協(xié)程的返回值類(lèi)型
Generator get_return_object() {
return Generator{ std::coroutine_handle<promise_type>::from_promise(*this) };
}
...
};
std::coroutine_handle<promise_type> handle;
int next() {
handle.resume();
return ???;
}
};
接下來(lái)就是如何獲取協(xié)程內(nèi)部傳出來(lái)的值的問(wèn)題了。同樣，本著有事找promise_type的原則，我們可以直接給它定義一個(gè)value成員：
struct Generator {
struct promise_type {
int value;
...
};
std::coroutine_handle<promise_type> handle;
int next() {
handle.resume();
// 通過(guò) handle 獲取 promise，然后再取到 value
return handle.promise().value;
}
};
協(xié)程內(nèi)部掛起并傳值
現(xiàn)在的問(wèn)題就是如何從協(xié)程內(nèi)部傳值給promise_type了。
我們?cè)儆^察一下最終實(shí)現(xiàn)的效果：
Generator sequence() {
int i = 0;
while (true) {
co_await i++;
}
}
特別需要注意的是co_await i++;這一句，我們發(fā)現(xiàn)co_await后面的是一個(gè)整型值，而不是我們?cè)谇懊嫖恼轮刑岬降臐M(mǎn)足等待體(awaiter)條件的類(lèi)型，這種情況下該怎么辦呢？
實(shí)際上，對(duì)于co_await <expr>表達(dá)式中expr的處理，C++有一套完善的流程：
- 如果
promise_type當(dāng)中定義了await_transform函數(shù)，那么先通過(guò)promise.await_transform(expr)來(lái)對(duì)expr做一次轉(zhuǎn)換，得到的對(duì)象則為awaitable;否則awaitable就是expr本身。 - 接下來(lái)使用
awaitable對(duì)象獲取等待體(awaiter)。如果awaitable對(duì)象有operator co_await運(yùn)算符重載，那么等待體就是operator co_await(awaitable),否則等待體就是awaitable對(duì)象本身。
按照上面的說(shuō)法，我們要么給promise_type實(shí)現(xiàn)一個(gè)await_transform(int)函數(shù)，要么就為整型實(shí)現(xiàn)一個(gè)operator co_await的運(yùn)算符重載，二者選一個(gè)就可以了。
方案 1：實(shí)現(xiàn) operator co_await
這個(gè)方案就是給 int 定義 operator co_await 的重載：
auto operator co_await(int value) {
struct IntAwaiter {
int value;
bool await_ready() const noexcept {
return false;
}
void await_suspend(std::coroutine_handle<Generator::promise_type> handle) const {
handle.promise().value = value;
}
void await_resume() { }
};
return IntAwaiter{.value = value};
}
當(dāng)然，這個(gè)方案對(duì)于我們這個(gè)特定的場(chǎng)景下是行不通的，因?yàn)樵?C++ 當(dāng)中我們是無(wú)法給基本類(lèi)型定義運(yùn)算符重載的。
不過(guò)，如果我們遇到的情況不是基本類(lèi)型，那么運(yùn)算符重載的思路就可以行得通。operator co_await 的重載我們將會(huì)在后面給出例子。
方案 2：await_transform
運(yùn)算符重載行不通，那就只能通過(guò) await_tranform 來(lái)做轉(zhuǎn)換了。
代碼比較簡(jiǎn)單：
struct Generator {
struct promise_type {
int value;
// 傳值的同時(shí)要掛起，值存入 value 當(dāng)中
std::suspend_always await_transform(int value) {
this->value = value;
return {};
}
...
};
std::coroutine_handle<promise_type> handle;
int next() {
handle.resume();
// 外部調(diào)用者或者恢復(fù)者可以通過(guò)讀取 value
return handle.promise().value;
}
};
定義了 await_transform 函數(shù)之后，co_await expr 就相當(dāng)于 co_await promise.await_transform(expr) 了。
至此，我們的例子就可以運(yùn)行了：
Generator sequence() {
int i = 0;
while (true) {
co_await i++;
}
}
int main() {
auto gen = sequence();
for (int i = 0; i < 5; ++i) {
std::cout << gen.next() << std::endl;
}
}
運(yùn)行結(jié)果如下：
0
1
2
3
4
協(xié)程的銷(xiāo)毀
雖然我們的協(xié)程已經(jīng)能夠正常工作，但它仍然存在一些問(wèn)題。
問(wèn)題1： 無(wú)法確定是否存在下一個(gè)元素
當(dāng)外部調(diào)用者或者恢復(fù)者試圖調(diào)用 next 來(lái)獲取下一個(gè)元素的時(shí)候，它其實(shí)并不知道能不能真的得到一個(gè)結(jié)果。程序也可能拋出異常：
如下例：
Generator sequence() {
int i = 0;
// 只傳出 5 個(gè)值
while (i < 5) {
co_await i++;
}
}
int main() {
auto gen = sequence();
for (int i = 0; i < 15; ++i) {
// 試圖讀取 15 個(gè)值
std::cout << gen.next() << std::endl;
}
return 0;
}
程序的結(jié)果是什么呢？
0
1
2
3
4
4Process finished with exit code 139 (interrupted by signal 11: SIGSEGV)
最后一個(gè)輸出的 4 實(shí)際上是恰好遇到協(xié)程銷(xiāo)毀之前的狀態(tài)，此時(shí) promise 當(dāng)中的 value 值還是之前的 4。而當(dāng)我們?cè)噲D不斷的去讀取協(xié)程的值，程序就拋出 SIGSEGV 的錯(cuò)誤。錯(cuò)誤的原因你可能已經(jīng)想到了，當(dāng)協(xié)程體執(zhí)行完之后，協(xié)程的狀態(tài)就會(huì)被銷(xiāo)毀，如果我們?cè)僭L問(wèn)協(xié)程的話，就相當(dāng)于訪問(wèn)了一個(gè)野指針。
為了解決這個(gè)問(wèn)題，我們需要增加一個(gè) has_next 函數(shù)，用來(lái)判斷是否還有新的值傳出來(lái)，has_next 函數(shù)調(diào)用的時(shí)候有兩種情況：
- 已經(jīng)有一個(gè)值傳出來(lái)了，還沒(méi)有被外部消費(fèi)
- 還沒(méi)有現(xiàn)成的值可以用，需要嘗試恢復(fù)執(zhí)行協(xié)程來(lái)看看還有沒(méi)有下一個(gè)值傳出來(lái)
這里我們需要有一種有效的辦法來(lái)判斷 value 是不是有效的，單憑 value 本身我們其實(shí)是無(wú)法確定它的值是不是被消費(fèi)了，因此我們需要加一個(gè)值來(lái)存儲(chǔ)這個(gè)狀態(tài)：
struct Generator {
// 協(xié)程執(zhí)行完成之后，外部讀取值時(shí)拋出的異常
class ExhaustedException: std::exception { };
struct promise_type {
int value;
bool is_ready = false;
...
}
...
}
我們定義一個(gè)成員 state 來(lái)記錄協(xié)程執(zhí)行的狀態(tài)，狀態(tài)的類(lèi)型一共三種，只有 READY 的時(shí)候我們才能拿到值。
接下來(lái)改造 next 函數(shù)，同時(shí)增加 has_next 函數(shù)來(lái)描述協(xié)程是否仍然可以有值傳出：
struct Generator {
...
bool has_next() {
// 協(xié)程已經(jīng)執(zhí)行完成
if (handle.done()) {
return false;
}
// 協(xié)程還沒(méi)有執(zhí)行完成，并且下一個(gè)值還沒(méi)有準(zhǔn)備好
if (!handle.promise().is_ready) {
handle.resume();
}
if (handle.done()) {
// 恢復(fù)執(zhí)行之后協(xié)程執(zhí)行完，這時(shí)候必然沒(méi)有通過(guò) co_await 傳出值來(lái)
return false;
} else {
return true;
}
}
int next() {
if (has_next()) {
// 此時(shí)一定有值，is_ready 為 true
// 消費(fèi)當(dāng)前的值，重置 is_ready 為 false
handle.promise().is_ready = false;
return handle.promise().value;
}
throw ExhaustedException();
}
};
注意，我們?cè)?code>promise_type中產(chǎn)生新值后需要將is_ready置為true如下：
struct Generator {
...
struct promise_type {
...
std::suspend_always await_transform(T value) {
this->value = value;
is_ready = true;
return {};
}
};
};
這樣外部使用時(shí)就需要先通過(guò) has_next 來(lái)判斷是否有下一個(gè)值，然后再去讀取了：
...
int main() {
auto generator = sequence();
for (int i = 0; i < 15; ++i) {
if (generator.has_next()) {
std::cout << generator.next() << std::endl;
} else {
break;
}
}
return 0;
}
問(wèn)題2：協(xié)程狀態(tài)的銷(xiāo)毀比Generator對(duì)象的銷(xiāo)毀更早
我們前面提到過(guò)，協(xié)程的狀態(tài)在協(xié)程體執(zhí)行完之后就會(huì)銷(xiāo)毀，除非協(xié)程掛起在 final_suspend 調(diào)用時(shí)。
我們的例子當(dāng)中 final_suspend 返回了 std::suspend_never，因此協(xié)程的銷(xiāo)毀時(shí)機(jī)其實(shí)比 Generator 更早：
auto generator = sequence();
for (int i = 0; i < 15; ++i) {
if (generator.has_next()) {
std::cout << generator.next() << std::endl;
} else {
// 協(xié)程已經(jīng)執(zhí)行完，協(xié)程的狀態(tài)已經(jīng)銷(xiāo)毀
break;
}
}
// generator 對(duì)象在此仍然有效
這看上去似乎問(wèn)題不大，因?yàn)槲覀冊(cè)谇懊嫱ㄟ^(guò) has_next 的判斷保證了讀取值的安全性。
但實(shí)際上情況并非如此。我們?cè)?nbsp;has_next 當(dāng)中調(diào)用了 coroutine_handle::done 來(lái)判斷協(xié)程體是否執(zhí)行完成，判斷之前很可能協(xié)程已經(jīng)銷(xiāo)毀，coroutine_handle 這時(shí)候都已經(jīng)是無(wú)效的了：
bool has_next() {
// 如果協(xié)程已經(jīng)執(zhí)行完成，理論上協(xié)程的狀態(tài)已經(jīng)銷(xiāo)毀，handle 指向的是一個(gè)無(wú)效的協(xié)程
// 如果 handle 本身已經(jīng)無(wú)效，因此 done 函數(shù)的調(diào)用此時(shí)也是無(wú)效的
if (handle.done()) {
return false;
}
...
}
因此為了讓協(xié)程的狀態(tài)的生成周期與 Generator 一致，我們必須將協(xié)程的銷(xiāo)毀交給 Generator 來(lái)處理：
struct Generator {
class ExhaustedException: std::exception { };
struct promise_type {
...
// 總是掛起，讓 Generator 來(lái)銷(xiāo)毀
std::suspend_always final_suspend() noexcept { return {}; }
...
};
...
~Generator() {
// 銷(xiāo)毀協(xié)程
handle.destroy();
}
};
問(wèn)題3：復(fù)制對(duì)象導(dǎo)致協(xié)程被銷(xiāo)毀
這個(gè)問(wèn)題確切地說(shuō)是問(wèn)題 2的解決方案不完善引起的。
我們?cè)?Generator 的析構(gòu)函數(shù)當(dāng)中銷(xiāo)毀協(xié)程，這本身沒(méi)有什么問(wèn)題，但如果我們把 Generator 對(duì)象做一下復(fù)制，例如從一個(gè)函數(shù)當(dāng)中返回，情況可能就會(huì)變得復(fù)雜。例如：
Generator returns_generator() {
auto g = sequence();
if (g.has_next()) {
std::cout << g.next() << std::endl;
}
return g;
}
int main() {
auto generator = returns_generator();
for (int i = 0; i < 15; ++i) {
if (generator.has_next()) {
std::cout << generator.next() << std::endl;
} else {
break;
}
}
return 0;
}
這段代碼乍一看似乎沒(méi)什么問(wèn)題，但由于我們把 g 當(dāng)做返回值返回了，這時(shí)候 g 這個(gè)對(duì)象就發(fā)生了一次復(fù)制，然后臨時(shí)對(duì)象被銷(xiāo)毀。接下來(lái)的事兒大家就很容易想到了，運(yùn)行結(jié)果如下：
0
-572662307Process finished with exit code -1073741819 (0xC0000005)
為了解決這個(gè)問(wèn)題，我們需要妥善地處理 Generator 的復(fù)制構(gòu)造器：
struct Generator {
...
explicit Generator(std::coroutine_handle<promise_type> handle) noexcept
: handle(handle) {}
Generator(Generator &&generator) noexcept
: handle(std::exchange(generator.handle, {})) {}
Generator(Generator &) = delete;
Generator &operator=(Generator &) = delete;
~Generator() {
if (handle) handle.destroy();
}
}
我們只提供了右值復(fù)制構(gòu)造器，對(duì)于左值復(fù)制構(gòu)造器，我們直接刪除掉以禁止使用。原因也很簡(jiǎn)單，對(duì)于每一個(gè)協(xié)程實(shí)例，都有且僅能有一個(gè) Generator 實(shí)例與之對(duì)應(yīng)，因此我們只支持移動(dòng)對(duì)象，而不支持復(fù)制對(duì)象。
使用co_yield
序列生成器這個(gè)需求的實(shí)現(xiàn)其實(shí)有個(gè)更好的選擇，那就是使用 co_yield。co_yield 就是專(zhuān)門(mén)為向外傳值來(lái)設(shè)計(jì)的，如果大家對(duì)其他語(yǔ)言的協(xié)程有了解，也一定見(jiàn)到過(guò)各種 yield 的實(shí)現(xiàn)。
C++ 當(dāng)中的 co_yield expr 等價(jià)于 co_await promise.yield_value(expr)，我們只需要將前面例子當(dāng)中的 await_transform 函數(shù)替換成 yield_value 就可以使用 co_yield 來(lái)傳值了：
struct Generator {
class ExhaustedException: std::exception { };
struct promise_type {
...
// 將 await_transform 替換為 yield_value
std::suspend_always yield_value(int value) {
this->value = value;
is_ready = true;
return {};
}
...
};
...
};
Generator sequence() {
int i = 0;
while (i < 5) {
// 使用 co_yield 來(lái)替換 co_await
co_yield i++;
}
}
可以看到改動(dòng)點(diǎn)非常少，運(yùn)行效果與前面的例子一致。
盡管可以實(shí)現(xiàn)相同的效果，但通常情況下我們使用 co_await 更多的關(guān)注點(diǎn)在掛起自己，等待別人上，而使用 co_yield 則是掛起自己傳值出去。因此我們應(yīng)該針對(duì)合適的場(chǎng)景做出合適的選擇。
使用序列生成器生成fibonacci數(shù)列
接下來(lái)我們要使用序列生成器來(lái)實(shí)現(xiàn)一個(gè)更有意義的例子，即斐波那契數(shù)列。
Generator fibonacci() {
co_yield 0; // fib(0)
co_yield 1; // fib(1)
int a = 0;
int b = 1;
while(true) {
co_yield a + b; // fib(N), N > 1
b = a + b;
a = b - a;
}
}
我們看到這個(gè)實(shí)現(xiàn)非常的直接，完全不需要考慮 fib(N - 1) 和 fib(N - 2) 的存儲(chǔ)問(wèn)題。
如果沒(méi)有協(xié)程，我們的實(shí)現(xiàn)可能是這樣的：
class Fibonacci {
public:
int next() {
// 初值不符合整體的規(guī)律，需要單獨(dú)處理
if (a == -1){
a = 0;
b = 1;
return 0;
}
int next = b;
b = a + b;
a = b - a;
return next;
}
private:
int a = -1;
int b = 0;
};
使用時(shí)先構(gòu)造一個(gè) Fibonacci 對(duì)象，然后調(diào)用 next 函數(shù)來(lái)獲取下一個(gè)值。對(duì)比之下，協(xié)程的實(shí)現(xiàn)帶來(lái)的好處是顯而易見(jiàn)的。協(xié)程生成的這個(gè)序列是個(gè)懶序列（Lazy Sequence），沒(méi)用到的項(xiàng)就不會(huì)被生成。
總結(jié)
本文圍繞序列生成器這個(gè)經(jīng)典的協(xié)程案例介紹了協(xié)程的銷(xiāo)毀、co_await 運(yùn)算符、await_transform 以及 yield_value 的用法。
以上就是C++協(xié)程實(shí)現(xiàn)序列生成器的案例分享的詳細(xì)內(nèi)容，更多關(guān)于C++序列生成器的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！
相關(guān)文章
Qt實(shí)現(xiàn)將qsqlite數(shù)據(jù)庫(kù)中的數(shù)據(jù)導(dǎo)出為Excel表格
這篇文章主要為大家詳細(xì)介紹了如何通過(guò)Qt實(shí)現(xiàn)將qsqlite數(shù)據(jù)庫(kù)中的數(shù)據(jù)導(dǎo)出為Excel表格,文中的示例代碼簡(jiǎn)潔易懂,有需要的小伙伴可以了解一下2024-12-12
本文主要介紹了C/C++實(shí)現(xiàn)蛇形矩陣的示例代碼，文中通過(guò)示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2022-01-01
這篇文章主要介紹了C/C++ 左移<<, 右移>>的作用及說(shuō)明，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教2022-07-07
C語(yǔ)言作為大學(xué)編程或者計(jì)算機(jī)專(zhuān)業(yè)的一門(mén)必修課，把很多初學(xué)編程的小伙伴都難住了，感覺(jué)無(wú)從下手，今天呢，我們來(lái)簡(jiǎn)單介紹下，如何快速入門(mén)C語(yǔ)言2019-02-02
這篇文章主要介紹了C++ seekg函數(shù)用法案例詳解,本篇文章通過(guò)簡(jiǎn)要的案例,講解了該項(xiàng)技術(shù)的了解與使用,以下就是詳細(xì)內(nèi)容,需要的朋友可以參考下2021-08-08
C++中成員函數(shù)和友元函數(shù)的使用及區(qū)別詳解
大家好，本篇文章主要講的是C++中成員函數(shù)和友元函數(shù)的使用及區(qū)別詳解，感興趣的同學(xué)趕快來(lái)看一看吧，對(duì)你有幫助的話記得收藏一下2022-01-01
C++中std::tuple和std::pair的高級(jí)用法
本文主要介紹了C++標(biāo)準(zhǔn)庫(kù)中std::pair和std::tuple的使用,包括它們的基本概念、使用場(chǎng)景、區(qū)別以及高級(jí)用法,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2024-11-11
這篇文章主要介紹了C++ 中 socket編程實(shí)例詳解的相關(guān)資料,需要的朋友可以參考下2017-06-06
C語(yǔ)言入門(mén)篇--關(guān)鍵字static詳解
本篇文章是C語(yǔ)言系列基礎(chǔ)篇，C語(yǔ)言中,static是用來(lái)修飾變量和函數(shù):1.修飾局部變量–>靜態(tài)局部變量2.修飾全局變量–>靜態(tài)全局變量3.修飾函數(shù)–>靜態(tài)函數(shù)2021-08-08
C++實(shí)現(xiàn)LeetCode(118.楊輝三角)
這篇文章主要介紹了C++實(shí)現(xiàn)LeetCode(118.楊輝三角),本篇文章通過(guò)簡(jiǎn)要的案例,講解了該項(xiàng)技術(shù)的了解與使用,以下就是詳細(xì)內(nèi)容,需要的朋友可以參考下2021-07-07