Concurrence系列-Tasklet使用

 

Concurrence框架中，最小的执行单位就是tasklet了。

 

 

创建一个Tasklet

 

 

from concurrence import dispatch, Tasklet def greeting(msg): print msg def start(): Tasklet.new(greeting)("Hello") Tasklet.new(greeting)("World") print 'start done.' if __name__ == '__main__': dispatch(start) 

 

以上代码会输出：

 

 

start done.

Hello

World

 

 

 

我们共同看一下上面代码的含义：

main函数里，dispatcher创建了一个task, 就是执行start方法。

在start方法里，又以Tasklet.new()的方式创建了两个tasklet，都是要执行greeting方法，参数不同。

Tasklet.new()创建任务，是non-blocking的。

所以start方式会首先执行完，print出start done.

 

然后dispatcher会scheduled两个task，并且执行了。

 

 

 

为了让这个过程更清楚显示出来，我们来改一下代码

 

from concurrence import dispatch, Tasklet def greeting(msg): while True: print msg Tasklet.sleep(1) def start(): Tasklet.new(greeting)("Hello") Tasklet.new(greeting)("World") if __name__ == '__main__': dispatch(start) 

 

 

这段代码会每1秒钟，打印出字符：

Hello

World

Hello

World

Hello

World

Hello

World

Hello

World

......

注意代码里的：Tasklet .sleep( 1) 方法。

 

 

等待Tasklet执行完

有时候我们需要挂起当前的任务，等待一个，多个自任务执行完毕后，再返回。可以用join的方法来做到这个。

from concurrence import dispatch, Tasklet def sum(a, b): #... potentially long calculation, involving calls to databases etc... return a + b def start(): t1 = Tasklet.new(sum)(10, 20) t2 = Tasklet.new(sum)(30, 40) res1 = Tasklet.join(t1) res2 = Tasklet.join(t2) print res1 print res2 if __name__ == '__main__': dispatch(start) 

 

在这个例子中，start方法创建了两个子任务。

在然后，start任务会block住，等待t1, t2返回。

注意，这里用的是，join方法。

还有两个方法可以用：

 

join_all() which takes a lists of tasks to join

 

join_children() which joins all the children of the current task.

 

 

 

 

Loop, Interval, Later

 

• loop()
• interval()
• later()

 

提起任务，就必须要有这几个方法。

 

示例：

from concurrence import dispatch, Tasklet def hello(msg): print msg tasklet.sleep(0.5) def greeting(msg): print msg def start(): Tasklet.loop(hello)("Hello World") Tasklet.interval(1.0, greeting)("Hi There!") Tasklet.later(5.0, greeting)("Nice to see You!") if __name__ == '__main__': dispatch(start) 

 

当前任务

找到当前正在执行的任务，用current()

from concurrence import dispatch, Tasklet def greeting(msg): print Tasklet.current(), msg def start(): Tasklet.interval(1.0, greeting)("Task 1") Tasklet.interval(1.0, greeting)("Task 2") if __name__ == '__main__': dispatch(start) 

 

任务树，任务名

每个task都有一个引用，指向创建它的任务（父任务），可以通过parent()获得。

每个task可以有子任务，children()可以访问到。当一个child执行完，推出后，会自动被从父任务中删除。

 

这样的一个task tree，可以通过tree()方法traverse。

 

任务是可以用个名字的。

在new()的时候传入

 

from concurrence import dispatch, Tasklet def greeting(msg): print msg Tasklet.sleep(2) print 'done' def start(): Tasklet.new(greeting, name = 'task1')("Hello") Tasklet.new(greeting, name = 'task2')("World") Tasklet.new(greeting, name = 'task3')("Hi There") while True: Tasklet.sleep(1.0) #print a nice tree of tasks and their subtasks for task, level in Tasklet.current().tree(): print "/t" * level, task.name if __name__ == '__main__': dispatch(start) 

 

 

下一篇，会介绍Messages.