Repo created

tools/python/test_server/server/tornado_handler.py

@@ -0,0 +1,99 @@
+from __future__ import print_function
+
+from ResponseProvider import Payload
+from ResponseProvider import ResponseProvider
+from ResponseProvider import ResponseProviderMixin
+from threading import Timer
+import os
+import threading
+import tornado.ioloop
+import tornado.web
+
+import logging
+
+
+class MainHandler(tornado.web.RequestHandler, ResponseProviderMixin):
+    
+    ping_count = 1
+    self_destruct_timer = None
+    
+    def got_pinged(self):
+        MainHandler.ping_count += 1
+
+        
+    def kill(self):
+        MainHandler.ping_count -= 1
+        if MainHandler.ping_count <= 0: #so that if we decrease the value from several threads we still kill it.
+            MainHandler.suicide()
+            if MainHandler.self_destruct_timer:
+                MainHandler.self_destruct_timer.cancel()
+
+    def dispatch_response(self, payload):
+        self.set_status(payload.response_code())
+        for h in payload.headers():
+            self.add_header(h, payload.headers()[h])
+        self.add_header("Content-Length", payload.length())
+        self.write(payload.message())
+    
+    
+    def prepare_headers(self):
+        ret = dict()
+        for h in self.request.headers:
+            ret[h.lower()] = self.request.headers.get(h)
+            
+        return ret
+    
+    
+    def init_vars(self):
+        self.response_provider = ResponseProvider(self)
+        self.headers = self.prepare_headers()
+        
+    
+    def prepare(self):
+        MainHandler.reset_self_destruct_timer()
+        self.init_vars()
+
+
+    def get(self, param):
+        self.dispatch_response(self.response_provider.response_for_url_and_headers(self.request.uri, self.headers))
+        
+
+    def post(self, param):
+        payload = self.response_provider.response_for_url_and_headers(self.request.uri, self.headers)
+        if payload.response_code() >= 300:
+            self.dispatch_response(Payload(self.request.body))
+        else:
+            self.dispatch_response(payload)
+
+
+    @staticmethod
+    def suicide():
+        tornado.ioloop.IOLoop.current().stop()
+        logging.info("The server's life has come to an end, pid: {}".format(os.getpid()))
+
+
+    @staticmethod
+    def reset_self_destruct_timer():
+        if MainHandler.self_destruct_timer:
+            logging.debug("Canceling the kill timer")
+            MainHandler.self_destruct_timer.cancel()
+        MainHandler.self_destruct_timer = Timer(MainHandler.lifespan, MainHandler.suicide)
+        logging.debug("Starting the kill timer")
+        MainHandler.self_destruct_timer.start()
+    
+    
+    @staticmethod
+    def start_serving():
+        thread = threading.Thread(target=tornado.ioloop.IOLoop.current().start)
+        thread.deamon = True
+        thread.start()
+        
+    
+    @staticmethod
+    def init_server(port, lifespan):
+        MainHandler.lifespan = lifespan
+        MainHandler.reset_self_destruct_timer()
+        application = tornado.web.Application([
+            (r"/(.*)", MainHandler),
+        ])
+        application.listen(port)