7 class Processor(threading.Thread):
11 self.lock = threading.Lock()
13 threading.Thread.__init__(self)
15 self.request_queue = queue.Queue()
16 self.response_queue = queue.Queue()
18 def add_session(self, session):
20 self.sessions.append(session)
22 def push_response(self, session, item):
23 self.response_queue.put((session,item))
25 def pop_response(self):
26 return self.response_queue.get()
28 def push_request(self, session, item):
29 self.request_queue.put((session,item))
31 def pop_request(self):
32 return self.request_queue.get()
35 if self.shared is None:
36 raise TypeError("self.shared not set in Processor")
37 while not self.shared.stopped():
38 # Deep copy entire sessions list and blank it
39 # This is done to minimise lock contention
41 sessions = self.sessions[:]
43 for session in sessions:
44 if not session.stopped():
45 # If session is still alive then re-add it back
46 # to our internal register
47 self.add_session(session)
49 session, request = self.pop_request()
50 self.process(session, request)
57 def process(self, session, request):
58 print "New request", request
61 # When ready, you call
62 # self.push_response(session,response)
66 def __init__(self, connection, address):
67 self._connection = connection
68 self.address = address
70 self.lock = threading.Lock()
71 self.numblocks_sub = None
72 self.addresses_sub = {}
73 print "new session", address
76 self._connection.close()
77 print "Terminating connection:", self.address[0]
87 raise Exception("Session was stopped")
89 return self._connection
91 def subscribe_to_numblocks(self,message_id):
93 self.numblocks_sub = message_id
95 def subscribe_to_address(self,address,message_id,status):
97 self.addresses_sub[address] = message_id,status
100 class TcpResponder(threading.Thread):
102 def __init__(self, shared, processor):
104 self.processor = processor
105 threading.Thread.__init__(self)
108 while not self.shared.stopped():
109 session,response = self.processor.pop_response()
110 raw_response = json.dumps(response)
111 # Possible race condition here by having session
113 # I assume Python connections are thread safe interfaces
115 connection = session.connection()
116 connection.send(raw_response + "\n")
120 class TcpClientRequestor(threading.Thread):
122 def __init__(self, shared, processor, session):
124 self.processor = processor
126 self.session = session
127 threading.Thread.__init__(self)
130 while not self.shared.stopped():
131 if not self.update():
138 data = self.receive()
149 return self.session.connection().recv(1024)
154 raw_buffer = self.message.find('\n')
158 raw_command = self.message[0:raw_buffer].strip()
159 self.message = self.message[raw_buffer + 1:]
160 if raw_command == 'quit':
165 command = json.loads(raw_command)
167 self.processor.push_response(self.session,
168 {"error": "bad JSON", "request": raw_command})
172 # Try to load vital fields, and return an error if
174 message_id = command['id']
175 method = command['method']
177 # Return an error JSON in response.
178 self.processor.push_response(self.session,
179 {"error": "syntax error", "request": raw_command})
181 self.processor.push_request(self.session,command)
185 class TcpServer(threading.Thread):
187 def __init__(self, shared, processor, host, port):
189 self.processor = processor
191 threading.Thread.__init__(self)
197 print "TCP server started."
198 sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
199 sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
200 sock.bind((self.host, self.port))
202 responder = TcpResponder(self.shared, self.processor)
204 while not self.shared.stopped():
205 session = Session(*sock.accept())
206 client_req = TcpClientRequestor(self.shared, self.processor, session)
208 self.processor.add_session(session)
213 self.lock = threading.Lock()
214 self._stopped = False
217 print "Stopping Stratum"
227 def start(self, processor):
229 # Bind shared to processor since constructor is user defined
230 processor.shared = shared
232 # Create various transports we need
233 transports = TcpServer(shared, processor, "176.31.24.241", 50001),
234 for server in transports:
236 while not shared.stopped():
237 if raw_input() == "quit":
241 if __name__ == "__main__":
242 processor = Processor()