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()
34 def collect_garbage(self):
35 # Deep copy entire sessions list and blank it
36 # This is done to minimise lock contention
38 sessions = self.sessions[:]
40 for session in sessions:
41 if not session.stopped():
42 # If session is still alive then re-add it back
43 # to our internal register
44 self.add_session(session)
47 if self.shared is None:
48 raise TypeError("self.shared not set in Processor")
49 while not self.shared.stopped():
50 self.collect_garbage()
51 session, request = self.pop_request()
52 self.process(session, request)
59 def process(self, session, request):
60 print "New request", request
63 # When ready, you call
64 # self.push_response(session,response)
66 def update_from_blocknum(self,block_number):
67 for session in self.sessions:
68 if not session.stopped():
69 if session.numblocks_sub is not None:
70 response = { 'id':session.numblocks_sub, 'result':block_number }
71 self.push_response(session,response)
73 def update_from_address(self,addr):
74 for session in self.sessions:
75 if not session.stopped():
76 m = session.addresses_sub.get(addr)
78 status = self.get_status( addr )
79 message_id, last_status = m
80 if status != last_status:
81 session.subscribe_to_address(addr,message_id, status)
82 response = { 'id':message_id, 'result':status }
83 self.push_response(session,response)
85 def get_status(self,addr):
86 # return status of an address
87 # return store.get_status(addr)
93 def __init__(self, connection, address):
94 self._connection = connection
95 self.address = address
97 self.lock = threading.Lock()
98 self.numblocks_sub = None
99 self.addresses_sub = {}
100 print "new session", address
103 self._connection.close()
104 print "Terminating connection:", self.address[0]
112 def connection(self):
114 raise Exception("Session was stopped")
116 return self._connection
118 def subscribe_to_numblocks(self,message_id):
120 self.numblocks_sub = message_id
122 def subscribe_to_address(self,address,message_id,status):
124 self.addresses_sub[address] = message_id,status
127 class TcpResponder(threading.Thread):
129 def __init__(self, shared, processor):
131 self.processor = processor
132 threading.Thread.__init__(self)
135 while not self.shared.stopped():
136 session,response = self.processor.pop_response()
137 raw_response = json.dumps(response)
138 # Possible race condition here by having session
140 # I assume Python connections are thread safe interfaces
142 connection = session.connection()
143 connection.send(raw_response + "\n")
147 class TcpClientRequestor(threading.Thread):
149 def __init__(self, shared, processor, session):
151 self.processor = processor
153 self.session = session
154 threading.Thread.__init__(self)
157 while not self.shared.stopped():
158 if not self.update():
165 data = self.receive()
176 return self.session.connection().recv(1024)
181 raw_buffer = self.message.find('\n')
185 raw_command = self.message[0:raw_buffer].strip()
186 self.message = self.message[raw_buffer + 1:]
187 if raw_command == 'quit':
192 command = json.loads(raw_command)
194 self.processor.push_response(self.session,
195 {"error": "bad JSON", "request": raw_command})
199 # Try to load vital fields, and return an error if
201 message_id = command['id']
202 method = command['method']
204 # Return an error JSON in response.
205 self.processor.push_response(self.session,
206 {"error": "syntax error", "request": raw_command})
208 self.processor.push_request(self.session,command)
212 class TcpServer(threading.Thread):
214 def __init__(self, shared, processor, host, port):
216 self.processor = processor
218 threading.Thread.__init__(self)
224 print "TCP server started."
225 sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
226 sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
227 sock.bind((self.host, self.port))
229 responder = TcpResponder(self.shared, self.processor)
231 while not self.shared.stopped():
232 session = Session(*sock.accept())
233 client_req = TcpClientRequestor(self.shared, self.processor, session)
235 self.processor.add_session(session)
240 self.lock = threading.Lock()
241 self._stopped = False
244 print "Stopping Stratum"
254 def start(self, processor):
256 # Bind shared to processor since constructor is user defined
257 processor.shared = shared
259 # Create various transports we need
260 transports = TcpServer(shared, processor, "176.31.24.241", 50001),
261 for server in transports:
263 while not shared.stopped():
264 if raw_input() == "quit":
268 if __name__ == "__main__":
269 processor = Processor()