-
Notifications
You must be signed in to change notification settings - Fork 0
/
my_taskworker.py
37 lines (32 loc) · 1015 Bytes
/
my_taskworker.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
#-*- coding=utf-8 -*-
#taskworker.py
import time,sys,Queue
from multiprocessing.managers import BaseManager
#创建类似的QueueManager
class QueueManager(BaseManager):
pass
#由于这个QueueManager 只从网络上获取Queue,所以注册时挚爱提供名字
QueueManager.register('get_task_queue')
QueueManager.register('get_result_queue')
#连接到服务器,也就是运行taskmanager的机器
server_addr = '127.0.0.1'
print 'Connect to server %s...' %server_addr
#端口和验证码注意保持与taskmanager.py设置的完全一致
m = QueueManager(address=(server_addr,5000),authkey='abc')
#从网络连接
m.connect()
#获取Queue的对象
task = m.get_task_queue()
result = m.get_result_queue()
#从task队列取任务,并把结果写入result队列
for i in range(10):
try:
n = task.get(timeout=1)
print 'run task %d * %d...' %(n,n)
r = '%d * %d = %d' % (n,n,n*n)
time.sleep(1)
result.put(r)
except Queue.Empty:
print 'task queue is empty...'
#处理结束
print 'worker exit..'