def __init__(self, host, port, handler):
threading.Thread.__init__(self)
self.setDaemon(True)
self.host = host
self.port = port
self.handler = handler
self.handler.server = self
self.running = True
self._pending_rpcs = {}
self._listen_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self._listen_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
self._listen_socket.settimeout(None)
self._listen_socket.bind((host, port))
self._poller = EPollWorker('Poll.%s.%d' % (handler.__class__.__name__, port))
class RPCServer(threading.Thread):
def __init__(self, host, port, handler):
threading.Thread.__init__(self)
self.setDaemon(True)
self.host = host
self.port = port
self.handler = handler
self.handler.server = self
self.running = True
self._pending_rpcs = {}
self._listen_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self._listen_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
self._listen_socket.settimeout(None)
self._listen_socket.bind((host, port))
self._poller = EPollWorker('Poll.%s.%d' % (handler.__class__.__name__, port))
def num_pending_rpcs(self):
return len(self._pending_rpcs)
def run(self):
self.serve_forever()
def stop(self):
self.shutdown()
def shutdown(self):
# self._call_histogram.dump()
self.running = False
logging.debug('Shutting down!')
def serve_forever(self):
self._listen_socket.listen(1000)
while self.running:
#logging.debug('Waiting for connections...')
s = [self._listen_socket]
try:
r, w, _ = select.select(s, s, s, 0.1)
except select.error:
continue
if r or w:
client_socket, addr = self._listen_socket.accept()
conn = ServerConnection(self, client_socket, addr)
self._poller.register(conn)
logging.info('Listen socket shutdown successfully.')
self._poller.stop()
self._listen_socket.close()
def rpc_finished(self, server_rpcid, *result):
# logging.info('RPCFINISHED %d', server_rpcid)
try:
handle = self._pending_rpcs[server_rpcid]
logging.debug('Returning result for sid %d cid %d',
handle.server_rpcid, handle.client_rpcid)
message = pickle((result, handle.elapsed(), None))
handle.connection.push_message(handle.client_rpcid, message)
del self._pending_rpcs[server_rpcid]
except:
logging.fatal('Unexpected error handling rpc finished!', exc_info = 1)
time.sleep(1)
def _dispatch(self, handle):
# logging.info('RPCSTART %d', handle.server_rpcid)
# logging.debug('Dispatching: %s', handle.method)
logging.debug('Dispatching for sid %d cid %d',
handle.server_rpcid, handle.client_rpcid)
self._pending_rpcs[handle.server_rpcid] = handle
handle.done = lambda *args: self.rpc_finished(handle.server_rpcid, *args)
try:
getattr(self.handler, handle.method)(handle, *handle.args)
except:
logging.warn('Exception while handling method: %s', handle.method, exc_info = 1)
handle.done(RemoteException(sys.exc_info()))
