利用高阶函数来实现协程(Racordon 1812.08278)

论文链接

协程的概念与实现

协程(coroutine)这一概念最早在1963年由Convay提出,虽然在上世纪80年代受到冷落,但在此之后,协程在Lua、Python、Ruby、Kotlin等诸多主流语言中都发挥了重要的作用。然而,包括Java、Swift等在内的很多语言并不能原生支持协程。本文作者提出了一种利用高阶函数来实现协程的方法,这可以应用于几乎所有编程语言。

协程和函数非常相像,通常来说,二者都接受若干参数,然后产生一定的结果。二者的区别在于,协程可以暂时挂起(通常会使用一个yield语句),保留其运行环境,并将控制权转交给另一个协程。在此之后,这一协程会重新获得控制权,继续运行直到再次挂起或终止。由于协程运行在同一个线程内,不存在资源共享的问题,无须使用锁,因而就避免了死锁的发生。

协程有很多不同的实现方法。这些方法间一个重要的分野就在于协程的实现是(全)对称还是半对称。所谓“(全)对称”,是指协程的切换可以发生在任意两个协程之间,而“半对称”则表示协程的切换只能在调用栈中直接的父/子(parent/child)之间发生。对称实现的协程有利于对整个流程进行更加全面的把控,而半对称的协程实现通常更容易理解、使用和调试。

这两种实现方式在最终的效果上并没有差异(参见de Moura & Ierusalimschy, 2009

另一个重要的区别在于协程是否是“栈满”(stackfull)的。栈满的协程可以从调用栈的更深层被挂起,而非栈满的则不行。如果要在一个非栈满的协程实现环境下,编写一个异步非阻塞的程序,就要求所有函数都能作为生成器,并且所有的函数调用都要显式地调用内层嵌套函数的结果。

同样的,这两种实现方式在最终的效果上也没有差异。

本文方法是一个半对称非栈满的协程实现。

协程实现:生成器方法

原文图1 JavaScript利用生成器实现Fibonacci序列的计算

上图给出了一个在JavaScript中利用生成器(generator)来实现协程的例子。函数内部的while(true)并不会一直运行,而是每次在yield处暂时挂起,直到下一次的.next()被调用时才继续进行。

协程实现:高阶函数方法(本文方法)

在这一部分作者给出了一个更加简单的例子:利用协程实现两个元素的加法。

首先是生成器的实现:

function * f(n) {
  const x = yield n
  yield n + x
}

const add2 = f(2)
add2.next() //=>2
add2.next(3) //=>5
add2.next(5) //=>undefined (因为一共只有两次yield)

接下来是高阶函数版本的实现:

function f(n) {
  let inst = 1
  return function(x) {
    switch (inst) {
    case 1 : inst += 1; return n
    case 2 : inst += 1; return n + x
    default: break
    }
  }
}

const add2 = f(2)
add2() //=>2
add2(3) //=>5
add2(5) //=>undefined

利用高阶函数来实现协程的出发点很简单:大部分支持高阶函数的编程语言都实现了闭包(closure),而实现协程的关键就在于保存环境(现场),那么就可以借助于高阶函数的函数闭包,来保存协程挂起时的所在环境。与生成器版本相比,利用高阶函数实现的协程不需要使用额外的语法(function*/yield/next)。

同样的,可以用高阶函数来实现上一节中Fibonacci序列的例子。

function fib() {
  let inst = 1; let a = null; let b = null
  return function() {
    while (true) {
      switch (inst) {
      case 1:
        inst = 2; a = 1; b = 2
      case 2:
        inst = 3; return a
      case 3:
        inst = 2; const c = a; a = b; b = c + a
      }
    }
  }
}

const f = fib()
f() //=>1
f() //=>2
f() //=>3
f() //=>5
f() //=>8

那么如果编程语言不支持闭包怎么办呢?那就需要自行实现类似闭包的保存环境机制,作者在附录中给出了一个C语言的例子,可以参考一下。

结语

这篇文章的内容到这里就介绍完了。用高阶函数的方式实现协程是不是很清晰明了呢?感兴趣的话,就亲自尝试一下吧!


附:Fibonacci序列生成器的C语言实现

// Emulation of first order functions.
typedef struct {
  void* env;
  void* (*fn)(void*);
} function_t;

void* apply(function_t closure) {
  return closure.fn(closure.env);
}

// Rewriting of the fibonacci sequence coroutine.
typedef struct {
  int inst;
  int a;
  int b;
} fib_env;

void* fib_fo(void* e) {
  fib_env* fe = (fib_env*)(e);
  while (1) {
    switch (fe->inst) {
    case 1:
      fe->inst = 2; fe->a = 1; fe->b = 2;
    case 2:
      fe->inst = 3; return &(fe->a);
    case 3:
      fe->inst = 2; int c = fe->a; fe->a = fe->b; fe->b = c + fe->a;
    }
  }
}

function_t fib() {
  fib_env* env = (fib_env*)(malloc(sizeof(fib_env)));
  env->inst = 1;
  function_t closure = { env, &fib_fo };
  return closure;
}

// Example of invocation.
int main() {
  function_t g = fib();
  for (int i = 0; i < 10; ++i) {
    printf("%i\n", *(int*)(apply(g)));
  }
  free(g.env);
  return 0;
}
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 205,033评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 87,725评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,473评论 0 338
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,846评论 1 277
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,848评论 5 368
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,691评论 1 282
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,053评论 3 399
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,700评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 42,856评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,676评论 2 323
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,787评论 1 333
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,430评论 4 321
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,034评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,990评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,218评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,174评论 2 352
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,526评论 2 343

推荐阅读更多精彩内容