tools/python/tornado_handler.py - master - githubmirror/omim

tools/python/tornado_handler.py
99 строк3 KB
Arsentiy Milchakov
[core] freenow taxi is added
20 мар 2020, 18:16
20 мар 2020, 18:1699c405f
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)