mirror of
https://gitee.com/acl-dev/acl.git
synced 2024-12-15 01:10:52 +08:00
181 lines
3.4 KiB
C++
181 lines
3.4 KiB
C++
#pragma once
|
||
#include <stddef.h>
|
||
|
||
struct ACL_FIBER;
|
||
|
||
namespace acl {
|
||
|
||
/**
|
||
* 协程类定义，纯虚类，需要子类继承并实现纯虚方法
|
||
*/
|
||
class fiber
|
||
{
|
||
public:
|
||
fiber(void);
|
||
virtual ~fiber(void);
|
||
|
||
/**
|
||
* 在创建一个协程类后，需要本函数启动协程
|
||
* @param stack_size {size_t} 创建的协程对象的栈大小
|
||
*/
|
||
void start(size_t stack_size = 320000);
|
||
|
||
/**
|
||
* 在本协程运行时调用此函数通知该协程退出
|
||
* @return {bool} 返回 false 表示本协程未启动或已经退出
|
||
*/
|
||
bool kill(void);
|
||
|
||
/**
|
||
* 判断当前协程是否被通知退出
|
||
* @return {bool} 本协程是否被通知退出
|
||
*/
|
||
bool killed(void) const;
|
||
|
||
/**
|
||
* 获得本协程对象的 ID 号
|
||
* @return {int}
|
||
*/
|
||
int get_id(void) const;
|
||
|
||
/**
|
||
* 获得当前运行的协程对象的 ID 号
|
||
*/
|
||
static int self(void);
|
||
|
||
/**
|
||
* 获得当前协程在执行某个系统 API 出错时的错误号
|
||
* return {int}
|
||
*/
|
||
int get_errno(void) const;
|
||
|
||
/**
|
||
* 设置当前协程的错误号
|
||
* @param errnum {int}
|
||
*/
|
||
void set_errno(int errnum);
|
||
|
||
/**
|
||
* 启动协程运行的调度过程
|
||
*/
|
||
static void schedule(void);
|
||
|
||
/**
|
||
* 停止协程调度过程
|
||
*/
|
||
static void schedule_stop(void);
|
||
|
||
public:
|
||
/**
|
||
* 将当前正在运行的协程(即本协程) 挂起
|
||
*/
|
||
static void yield(void);
|
||
|
||
/**
|
||
* 挂起当前协程，执行等待队列中的下一个协程
|
||
*/
|
||
static void switch_to_next(void);
|
||
|
||
/**
|
||
* 将指定协程对象置入待运行队列中
|
||
* @param f {fiber&}
|
||
*/
|
||
static void ready(fiber& f);
|
||
|
||
/**
|
||
* 线程启动后调用此函数设置当前线程是否需要 hook 系统 API，内部缺省
|
||
* 会 hook 系统 API
|
||
* @param on {bool}
|
||
*/
|
||
static void hook_api(bool on);
|
||
|
||
public:
|
||
/**
|
||
* 返回本协程对象对应的 C 语言的协程对象
|
||
* @return {ACL_FIBER *}
|
||
*/
|
||
ACL_FIBER *get_fiber(void) const;
|
||
|
||
protected:
|
||
/**
|
||
* 纯虚函数，子类必须实现本函数，当通过调用 start 方法启动协程后，本
|
||
* 虚函数将会被调用，从而通知子类
|
||
*/
|
||
virtual void run(void) = 0;
|
||
|
||
private:
|
||
ACL_FIBER *f_;
|
||
|
||
static void fiber_callback(ACL_FIBER *f, void *ctx);
|
||
};
|
||
|
||
} // namespace acl
|
||
|
||
#if defined(__GNUC__) && (__GNUC__ > 6 ||(__GNUC__ == 6 && __GNUC_MINOR__ >= 0))
|
||
# ifndef ACL_USE_CPP11
|
||
# define ACL_USE_CPP11
|
||
# endif
|
||
#endif
|
||
|
||
#ifdef ACL_USE_CPP11
|
||
|
||
#include <functional>
|
||
|
||
namespace acl
|
||
{
|
||
|
||
class go_fiber
|
||
{
|
||
public:
|
||
go_fiber(void) {}
|
||
go_fiber(size_t stack_size) : stack_size_(stack_size) {}
|
||
|
||
void operator=(std::function<void()> fn);
|
||
|
||
private:
|
||
size_t stack_size_ = 320000;
|
||
};
|
||
|
||
} // namespace acl
|
||
|
||
#define go acl::go_fiber()=
|
||
#define go_stack(size) acl::go_fiber(size)=
|
||
|
||
/**
|
||
* static void fiber1(void)
|
||
* {
|
||
* printf("fiber: %d\r\n", acl::fiber::self());
|
||
* }
|
||
*
|
||
* static void fiber2(acl::string& buf)
|
||
* {
|
||
* printf("in fiber: %d, buf: %s\r\n", acl::fiber::self(), buf.c_str());
|
||
* buf = "world";
|
||
* }
|
||
*
|
||
* static void fiber3(const acl::string& buf)
|
||
* {
|
||
* printf("in fiber: %d, buf: %s\r\n", acl::fiber::self(), buf.c_str());
|
||
* }
|
||
*
|
||
* static test(void)
|
||
* {
|
||
* go fiber1;
|
||
*
|
||
* acl::string buf("hello");
|
||
*
|
||
* go[&] {
|
||
* fiber2(buf);
|
||
* };
|
||
*
|
||
* go[=] {
|
||
* fiber3(buf);
|
||
* };
|
||
*
|
||
* go[&] {
|
||
* fiber3(buf);
|
||
* };
|
||
* }
|
||
*/
|
||
#endif
|