std::shared_ptr的巧妙应用

std::shared_ptr

一次创建,多处共享,通过引用计数控制生命周期。

问题:

假设有一项任务Task,其分为N个小模块,每个小模块都是并发执行。

现在要求,任务开始,每个小模块并发执行,在全部小模块执行完后,将所有小模块结果进行下一步处理,任务结束。

方案一:

任务开始,先创建一个成员变量std::set<Span*> n_span;(这里Span代表一个小模块,异步执行模块的功能);

将每个小模块都记录在n_span中;

当小模块执行完后,从n_span中移除;

当n_span为空时,意味着所有小模块都执行完毕,可以继续下一步处理了。

方案一看起来简单清晰,那么其存在什么问题呢?

假如某个小模块因为某种原因被销毁了,那么,其模块便不会返回,Task就会永远lost,永远不会继续下一步处理了。

*****std::shared_ptr闪亮登场*****

在使用std::shared_ptr时,我们很少注意到,其构造函数有两个参数,

第一个是对象指针,

第二个是deleter,可以让我们自己定义其析构函数。

auto next = [](){};
std::shared_ptr<void> shared_count((void*)0, [next](void*){ next() });
for(int i = 0; i < N; i++){
    Span* s = new Span(shared_count);
    s->Run();  // Span在执行结束后delete this
}
无论Span是正常还是非正常结束,保存在Span中的std::shared_ptr<void>销毁后都会导致shared_count引用计数-1,当引用计数为0时,其执行了我们定义的deleter,在此可以执行我们的下一步处理了。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章