tbox的协程实现,是stackfull模式的,需要指定独立堆栈和协程函数,目前暂时还不能像golang那样实现堆栈的动态增长,之后会对其进行支持。

目前提供下面一些功能特性:

  1. 1. 提供yield切换调度支持,这个是必须的哈
  2. 2. 提供suspend(挂起)/resume(恢复)协程接口,不同于yield的是,被suspend后,如果不显示调用resume恢复它,是永远不会被调度到的
  3. 3. 提供sleep等待接口支持
  4. 4. 提供io调度支持,支持socketio等待(内部使用epoll, poll, kqueue, select, poll等接口调度)
  5. 5. 原生支持streamsockethttp等模块的协程支持,可与线程进行无缝切换
  6. 6. 提供channel同道,进行协程间交互通信
  7. 7. 提供locksemaphore等接口

这里主要讲讲,基础的切换调度如何使用。。

yield切换

这个的使用非常简单,直接上代码吧,嘿嘿。。

  1. static tb_void_t switchfunc(tb_cpointer_t priv)
  2. {
  3. // 获取传入的参数
  4. tb_size_t count = (tb_size_t)priv;
  5. while (count--)
  6. {
  7. // 打印当前协程id
  8. tb_trace_i("[coroutine: %p]: %lu", tb_coroutine_self(), count);
  9.  
  10. // 让出当前协程,进行切换
  11. tb_coroutine_yield();
  12. }
  13. }
  14. tb_int_t main(tb_int_t argc, tb_char_t** argv)
  15. {
  16. // 初始化tbox
  17. if (!tb_init(tb_null, tb_null)) return -1;
  18.  
  19. // 初始化一个调度器实例
  20. tb_co_scheduler_ref_t scheduler = tb_co_scheduler_init();
  21. if (scheduler)
  22. {
  23. // 开启一个协程,传递switchfunc切换函数和参数10,最后一个参数指定栈大小,传0使用默认值
  24. tb_coroutine_start(scheduler, switchfunc, (tb_cpointer_t)10, 0);
  25.  
  26. // 开启协程,使用指定的栈大小:8192
  27. tb_coroutine_start(scheduler, switchfunc, (tb_cpointer_t)10, 8192);
  28.  
  29. /* 运行调用,开始运行里面的所有协程
  30. *
  31. * 第二个参数指定是否启用独占模式,这个稍后细讲
  32. */
  33. tb_co_scheduler_loop(scheduler, tb_true);
  34.  
  35. // 退出调度器
  36. tb_co_scheduler_exit(scheduler);
  37. }
  38.  
  39. // 退出tbox
  40. tb_exit();
  41. }

所有协程执行完后,就会从loop调用处返回,当然在协程函数内部也是可以嵌套开启新协程的

这个时候第一个参数就不需要显示指定scheduler了,传tb_null表示在当前调度器环境中开新协程,例如:

  1. static tb_void_t switchfunc2(tb_cpointer_t priv)
  2. {
  3. // ..
  4. }
  5. static tb_void_t switchfunc(tb_cpointer_t priv)
  6. {
  7. // 在当前协程函数内,开启一个新协程
  8. tb_coroutine_start(tb_null, switchfunc2, tb_null, 0);
  9.  
  10. // 获取传入的参数
  11. tb_size_t count = (tb_size_t)priv;
  12. while (count--)
  13. {
  14. // 让出当前协程,进行切换
  15. tb_coroutine_yield();
  16. }
  17. }

独占模式

上面的代码中提到的独占模式,需要特殊说明下,一般情况下,传tb_false到loop()中,不启用此模式是最为稳妥的。

因为这样每个scheduler对应的线程都会在Tls中维护自己的scheduler引用,使得协程中所有操作,都回去访问当前线程tls对应的独立scheduler,互不干涉。

这个在存在多个scheduler的情况,尤为重要,当时这样多少会有些tls操作上的性能损耗,像server端一般只有一个scheduler的情况下,就没必要放到独立tls中去了

可以传入tb_true启用独占模式,告诉tb_co_scheduler_loop(),我当前不需要tls维护,只需要一个全局scheduler变量来维护就行了

这样的话,性能会提升一些,因此在只有一个scheduler存在的情况下,启用独占效率会高些。。

sleep等待

tbox的tb_sleeptb_msleep()接口,是可以原生支持协程的,在协程外就是线程等待,在协程内就是协程等待。

当然也可以直接使用协程的接口:tb_coroutine_sleep()

例如:

  1. static tb_void_t sleepfunc(tb_cpointer_t priv)
  2. {
  3. while (1)
  4. {
  5. // 等待10ms,切换到其他协程,直到10ms后才会切换回来继续执行
  6. tb_msleep(10);
  7. }
  8. }

resume/suspend接口

挂起域恢复,跟yield的区别就是,被suspend挂起的协程,默认是不会被切换调度回来的,除非执行resume恢复它。

因此这两个接口是成对使用的,像sleep,lock和semaphore的内部实现,也是基于此套接口。

这两个接口还有个功能,就是可以在两个协程间,更加快速方便的传递一些参数数据,进行交互,而不需要channel支持。。

例如:

  1. static tb_void_t resumefunc(tb_cpointer_t priv)
  2. {
  3. // 获取suspend协程引用
  4. tb_coroutine_ref_t coroutine = (tb_coroutine_ref_t)priv;
  5.  
  6. /* 恢复suspend协程,传递参数"hello suspend!"给suspend()作为其返回值
  7. *
  8. * retval为suspend()中传入的参数:"hello resume!"
  9. */
  10. tb_char_t const* retval = tb_coroutine_resume(coroutine, "hello suspend!");
  11. }
  12. static tb_void_t suspendfunc(tb_cpointer_t priv)
  13. {
  14. // 开启一个恢复协程,传入当前协程引用
  15. tb_coroutine_start(tb_null, resumefunc, tb_coroutine_self(), 0);
  16.  
  17. /* 挂起当前协程,传递参数"hello resume!"给resume()作为其返回值
  18. *
  19. * retval为resume()中传入的参数:"hello suspend!"
  20. */
  21. tb_char_t const* retval = tb_coroutine_suspend("hello resume!");
  22. }

当然,如果不需要交互数据,那么只需要传tb_null就行了。。