forked from ktuite/twisted-server-cpp-worker
/
workerManager.py
62 lines (49 loc) · 1.91 KB
/
workerManager.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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
from twisted.internet import protocol, defer, threads, utils, error
from twisted.protocols import basic
from twisted.internet.protocol import Factory
class WorkerProtocol(basic.LineReceiver):
def __init__(self):
print "init"
self.queue = []
# Don't need a connectionMade, nothing interesting happens on connection
def connectionMade(self):
print "[ATTENTION] connection made"
self.factory.master.client = self
def connectionLost(self, reason):
#if reason.check("twisted.internet.error.ConnectionDone"):
print "[ATTENTION] connection lost"
def sendCommand(self, cmd, job):
self.enqueue(cmd, job)
def enqueue(self, cmd, job):
print "enqueuing command [%s]\n" % cmd
if not self.queue:
self.sendLine(cmd)
self.queue.insert(0, (cmd, job) )
def lineReceived(self, response_line):
print "Line received:", response_line
if len(self.queue) > 0:
_, job = self.queue.pop()
job.callback(response_line)
if self.queue:
cmd, job = self.queue[-1]
print "dequeuing command [%s]"
self.sendLine(cmd)
class WorkerManager(object):
def __init__(self, reactor):
self.reactor = reactor
self.client = None
print "initializing client manager"
@defer.inlineCallbacks
def launchClient(self):
yield utils.getProcessValue("workerclient", [])
#yield utils.getProcessValue("python", ["workerClient.py", "8005"])
def start(self):
factory = Factory()
factory.protocol = WorkerProtocol
factory.master = self
self.reactor.listenTCP(8005, factory)
self.launchClient()
def sendMessageToClient(self, message):
job = defer.Deferred()
self.client.sendCommand(message, job)
return job