-
Notifications
You must be signed in to change notification settings - Fork 0
/
master.py
44 lines (37 loc) · 1.29 KB
/
master.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
38
39
40
41
42
43
44
import os
import sys
from twisted.internet.protocol import Protocol
from armada.encoding import Reader, packArray, packUTF8, packPickle, ReadException, packUInt, packByte
import inspect
SUCCESS, FAILURE = range(2)
CONTINUE, KILL = range(2)
class ArmadaMasterProtocol(Protocol):
def __init__(self):
self.finished = None
self.reader = Reader()
def queueWork(self, finished, function, *args):
self.finished = finished
if hasattr(function, '__call__'):
if function.__module__ == '__main__':
function = '.'.join([os.path.splitext(os.path.basename(sys.modules['__main__'].__file__))[0], function.__name__])
else:
function = '.'.join([function.__module__, function.__name__])
self.transport.write(packByte(CONTINUE) + packUTF8(function) + packArray(packUInt, packPickle, args))
return self.finished
def kill(self):
self.transport.write(packByte(KILL))
def dataReceived(self, data):
self.reader.addData(data)
try:
status = self.reader.readByte()
result = self.reader.readPickle()
if status == SUCCESS:
self.finished.callback(result)
elif status == FAILURE:
self.finished.errback(result)
self.reader.commit()
except ReadException:
self.reader.revert()
@property
def working(self):
return self.finished is not None