Async_simple项目中自定义协程Promise内存分配机制的探讨

背景介绍

在C++20协程编程中，协程的Promise对象内存分配是一个值得关注的技术点。标准规定协程框架会自动为每个协程分配Promise对象，但开发者可以通过自定义operator new来控制内存分配策略。async_simple作为阿里巴巴开源的C++协程库，其Lazy协程类型也面临着如何支持用户自定义内存分配的需求。

技术挑战

在async_simple的LazyPromise实现中，目前缺乏对自定义内存分配器的支持。这导致开发者无法灵活控制协程Promise对象的内存分配位置，特别是在需要高性能内存管理或特殊内存池的场景下。

标准C++提供了几种内存控制方案：

1. 全局替换operator new/delete
2. 类特定的operator new/delete重载
3. 使用pmr(Polymorphic Memory Resources)内存资源

解决方案探讨

方案一：弱符号全局函数

通过定义弱符号全局分配函数，允许用户在链接时覆盖默认实现：

__attribute__((weak, noinline))
void* async_simple_allocate(std::size_t n) noexcept {
    return nullptr;
}

class LazyPromise {
    void* operator new(std::size_t n) noexcept {
        if (void* p = async_simple_allocate(n))
            return p;
        return ::operator new(n);
    }
};

这种方案的优点是不侵入现有API，但灵活性较低，无法针对不同协程使用不同分配策略。

方案二：pmr内存资源集成

利用C++17的pmr机制，通过协程参数传递内存资源：

class LazyPromise {
    void* operator new(std::size_t size, std::pmr::memory_resource* resource) {
        void* p = resource->allocate(size + sizeof(void*));
        // 存储resource指针用于后续释放
        return p;
    }
    
    void operator delete(void* ptr, std::size_t size) {
        // 取出resource指针并释放
    }
};

使用方式：

Lazy<int> foo(std::pmr::memory_resource* resource) {
    co_return 42;
}

这种方案符合现代C++内存管理习惯，但需要用户代码显式传递memory_resource。

方案三：PromiseAllocator基类

async_simple已提供了PromiseAllocator模板，可作为基类继承：

template <typename T>
class LazyPromiseBase : public PromiseAllocator<T> {
    // ...
};

但当前实现存在一个问题：PromiseAllocator的operator new未标记noexcept，而协程要求分配失败时调用get_return_object_on_allocation_failure，需要noexcept保证。

技术权衡

1. 性能考量：全局替换方案性能最佳，但灵活性最差；pmr方案有轻微虚函数开销，但灵活性最高。

2. API设计：pmr方案需要修改函数签名，可能影响现有代码；弱符号方案API兼容性最好。

3. 标准兼容：pmr是标准方案，未来兼容性有保障；自定义方案可能面临维护成本。

实现建议

综合来看，推荐采用pmr方案，原因如下：

1. 标准合规，长期可维护
2. 灵活性高，支持不同协程使用不同分配器
3. 与现代C++内存管理实践一致
4. 可通过默认参数保持API兼容性

实现时需要注意：

1. 分配额外空间存储memory_resource指针
2. 确保异常安全，正确处理分配失败
3. 提供合理的默认内存资源

结论

在async_simple中支持自定义Promise内存分配是提升库灵活性的重要改进。基于pmr的方案虽然需要一定的实现工作，但从长期来看是最符合C++发展趋势的选择。开发者可以根据具体场景选择全局替换或细粒度的pmr控制，平衡性能与灵活性需求。