7 class Processor(threading.Thread):
11 self.lock = threading.Lock()
13 threading.Thread.__init__(self)
16 def add_session(self, session):
18 self.sessions.append(session)
21 if self.shared is None:
22 raise TypeError("self.shared not set in Processor")
23 while not self.shared.stopped():
24 # Deep copy entire sessions list and blank it
25 # This is done to minimise lock contention
27 sessions = self.sessions[:]
29 for session in sessions:
30 if not session.stopped():
31 # If session is still alive then re-add it back
32 # to our internal register
33 self.add_session(session)
36 def process(self, session):
37 request = session.pop_request()
38 print "New request", request
39 # Execute and when ready, you call
40 # session.push_response(response)
44 def __init__(self, connection, address):
45 self._connection = connection
46 self.address = address
48 self.lock = threading.Lock()
50 self.request_queue = queue.Queue()
51 self.response_queue = queue.Queue()
54 self._connection.close()
55 print "Terminating connection:", self.address[0]
65 raise Exception("Session was stopped")
67 return self._connection
69 def push_request(self, item):
70 self.request_queue.put(item)
72 def pop_request(self):
73 return self.request_queue.get()
75 def push_response(self):
76 self.response_queue.put(item)
78 def pop_response(self):
79 return self.response_queue.get()
81 class TcpClientResponder(threading.Thread):
83 def __init__(self, shared, session):
85 self.session = session
86 threading.Thread.__init__(self)
89 while not self.shared.stopped() or self.session.stopped():
90 response = self.session.pop_response()
91 # Possible race condition here by having session
93 # I assume Python connections are thread safe interfaces
94 connection = self.session.connection()
96 connection.send(response + "\n")
100 class TcpClientRequestor(threading.Thread):
102 def __init__(self, shared, session):
105 self.session = session
106 threading.Thread.__init__(self)
109 while not self.shared.stopped():
110 if not self.update():
115 data = self.receive()
128 return self.session.connection().recv(1024)
134 raw_buffer = self.message.find('\n')
138 command = self.message[0:raw_buffer].strip()
139 self.message = self.message[raw_buffer + 1:]
140 if command == 'quit':
144 command = json.loads(command)
146 print "json error", repr(command)
150 message_id = command.get('id')
151 method = command.get('method')
152 params = command.get('params')
154 print "syntax error", repr(command), self.session.address[0]
157 self.session.push_request((message_id, method, params))
158 print message_id, method, params
160 class TcpServer(threading.Thread):
162 def __init__(self, shared, processor):
164 self.processor = processor
166 threading.Thread.__init__(self)
170 print "TCP server started."
171 sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
172 sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
173 sock.bind(("localhost", 50001))
175 while not self.shared.stopped():
176 session = Session(*sock.accept())
177 client_req = TcpClientRequestor(self.shared, session)
179 client_res = TcpClientResponder(self.shared, session)
181 self.processor.add_session(session)
186 self.lock = threading.Lock()
187 self._stopped = False
190 print "Stopping Stratum"
200 def start(self, processor):
202 # Bind shared to processor since constructor is user defined
203 processor.shared = shared
205 # Create various transports we need
206 transports = TcpServer(shared, processor),
207 for server in transports:
209 while not shared.stopped():
210 if raw_input() == "quit":
214 if __name__ == "__main__":
215 processor = Processor()