def fork_server():
pid = os.fork()
if not pid:
server = TCPServer()
server.commands = {'add': TestTCPHandlerSimple,
'cpu': TestTCPHandlerCPU}
server.listen()
else:
try:
time.sleep(2)
run_tests()
except Exception:
print "TESTS FAILED"
raise
os.kill(pid, signal.SIGKILL)
# Get memory usage until the parent tells us to stop or we timeout.
max_mem = self.start_mem
self.conn.sock.settimeout(1.0 / 10000)
while True:
curr_mem = get_memory_usage(self.proc_to_watch)
max_mem = max(curr_mem, max_mem)
# Poll for stop signal
try:
self.recv()
break
except:
continue
# Send the results back to the parent.
return self.success(peak_usage=max_mem - self.start_mem)
except Exception as e:
msg = e.message
if not msg:
msg = "Memory Profiler process died watching %d!" % self.proc_to_watch
logger.error(msg)
return self.error(msg)
if __name__ == "__main__":
server = TCPServer(PROFILER_HOST, PROFILER_PORT)
server.commands = {
'profile': MemoryProfiler
}
server.listen()
count += i + i
return self.success(result=count)
class TestTCPHandlerDone(TCPHandler):
def execute(self):
global REQUESTS
reqs = REQUESTS
REQUESTS = 0
return self.success(requests_served=reqs)
class TestTCPHandlerLargeResponse(TCPHandler):
def execute(self):
global REQUESTS
with LOCK:
REQUESTS += 1
junk = [0] * 100000
return self.success(junk=junk)
server = TCPServer()
server.commands = {
'add': TestTCPHandlerSimple,
'cpu': TestTCPHandlerCPU,
'large': TestTCPHandlerLargeResponse,
'finish': TestTCPHandlerDone
}
if __name__ == "__main__":
server.listen()
from tcpy import TCPServer, TCPHandler
class HelloWorldHandler(TCPHandler):
def execute(self):
return self.success(message="Hello world!")
if __name__ == "__main__":
server = TCPServer()
server.commands = {"hello_world": HelloWorldHandler}
server.listen()
