7 from processor import Session, Dispatcher, Shared
9 class TcpSession(Session):
11 def __init__(self, connection, address):
12 self._connection = connection
13 self.address = address
14 Session.__init__(self)
18 raise Exception("Session was stopped")
20 return self._connection
23 self._connection.close()
24 print "Terminating connection:", self.address[0]
28 def send_response(self, response):
29 raw_response = json.dumps(response)
30 # Possible race condition here by having session
32 # I assume Python connections are thread safe interfaces
34 connection = self.connection()
35 connection.send(raw_response + "\n")
41 class TcpClientRequestor(threading.Thread):
43 def __init__(self, shared, processor, session):
45 self.processor = processor
47 self.session = session
48 threading.Thread.__init__(self)
51 while not self.shared.stopped():
70 return self.session.connection().recv(1024)
75 raw_buffer = self.message.find('\n')
79 raw_command = self.message[0:raw_buffer].strip()
80 self.message = self.message[raw_buffer + 1:]
81 if raw_command == 'quit':
86 command = json.loads(raw_command)
88 self.processor.push_response({"error": "bad JSON", "request": raw_command})
92 # Try to load vital fields, and return an error if
94 message_id = command['id']
95 method = command['method']
97 # Return an error JSON in response.
98 self.processor.push_response({"error": "syntax error", "request": raw_command})
100 self.processor.push_request(self.session,command)
104 class TcpServer(threading.Thread):
106 def __init__(self, shared, processor, host, port):
108 self.processor = processor
109 threading.Thread.__init__(self)
113 self.lock = threading.Lock()
116 print "TCP server started."
117 sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
118 sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
119 sock.bind((self.host, self.port))
121 while not self.shared.stopped():
122 session = TcpSession(*sock.accept())
123 client_req = TcpClientRequestor(self.shared, self.processor, session)
125 self.processor.add_session(session)
126 self.processor.collect_garbage()
133 def start(self, processor):
135 # Bind shared to processor since constructor is user defined
136 processor.shared = shared
138 # Create various transports we need
139 transports = TcpServer(shared, processor, "176.31.24.241", 50001),
140 for server in transports:
142 while not shared.stopped():
143 if raw_input() == "quit":
147 if __name__ == "__main__":
148 processor = Processor()