今生今世

--nothreading

 
class GeventWorker(AsyncWorker):
 
    server_class = None
    wsgi_handler = None
 
    def patch(self):
        from gevent import monkey
        monkey.noisy = False
 
        # if the new version is used make sure to patch subprocess
        if gevent.version_info[0] == 0:
            monkey.patch_all()
        else:
            monkey.patch_all(subprocess=True)
 
        # monkey patch sendfile to make it none blocking
        patch_sendfile()
 
        # patch sockets
        sockets = []
        for s in self.sockets:
            if sys.version_info[0] == 3:
                sockets.append(socket(s.FAMILY, _socket.SOCK_STREAM,
                    fileno=s.sock.fileno()))
            else:
                sockets.append(socket(s.FAMILY, _socket.SOCK_STREAM,
                    _sock=s))
        self.sockets = sockets
 
    def notify(self):
        super(GeventWorker, self).notify()
        if self.ppid != os.getppid():
            self.log.info("Parent changed, shutting down: %s", self)
            sys.exit(0)
 
    def timeout_ctx(self):
        return gevent.Timeout(self.cfg.keepalive, False)
 
    def run(self):
        servers = []
        ssl_args = {}
 
        if self.cfg.is_ssl:
            ssl_args = dict(server_side=True, **self.cfg.ssl_options)
 
        for s in self.sockets:
            s.setblocking(1)
            pool = Pool(self.worker_connections)
            if self.server_class is not None:
                environ = base_environ(self.cfg)
                environ.update({
                    "wsgi.multithread": True,
                    "SERVER_SOFTWARE": VERSION,
                })
                server = self.server_class(
                    s, application=self.wsgi, spawn=pool, log=self.log,
                    handler_class=self.wsgi_handler, environ=environ,
                    **ssl_args)
            else:
                hfun = partial(self.handle, s)
                server = StreamServer(s, handle=hfun, spawn=pool, **ssl_args)
 
            server.start()
            servers.append(server)
 
        while self.alive:
            self.notify()
            gevent.sleep(1.0)
 
        try:
            # Stop accepting requests
            for server in servers:
                if hasattr(server, 'close'):  # gevent 1.0
                    server.close()
                if hasattr(server, 'kill'):  # gevent < 1.0
                    server.kill()
 
            # Handle current requests until graceful_timeout
            ts = time.time()
            while time.time() - ts <= self.cfg.graceful_timeout:
                accepting = 0
                for server in servers:
                    if server.pool.free_count() != server.pool.size:
                        accepting += 1
 
                # if no server is accepting a connection, we can exit
                if not accepting:
                    return
 
                self.notify()
                gevent.sleep(1.0)
 
            # Force kill all active the handlers
            self.log.warning("Worker graceful timeout (pid:%s)" % self.pid)
            [server.stop(timeout=1) for server in servers]
        except:
            pass
 
    def handle_request(self, *args):
        try:
            super(GeventWorker, self).handle_request(*args)
        except gevent.GreenletExit:
            pass
        except SystemExit:
            pass
 
    def handle_quit(self, sig, frame):
        # Move this out of the signal handler so we can use
        # blocking calls. See #1126
        gevent.spawn(super(GeventWorker, self).handle_quit, sig, frame)
 
    if gevent.version_info[0] == 0:
 
        def init_process(self):
            # monkey patch here
            self.patch()
 
            # reinit the hub
            import gevent.core
            gevent.core.reinit()
 
            #gevent 0.13 and older doesn't reinitialize dns for us after forking
            #here's the workaround
            gevent.core.dns_shutdown(fail_requests=1)
            gevent.core.dns_init()
            super(GeventWorker, self).init_process()
 
    else:
 
        def init_process(self):
            # monkey patch here
            self.patch()
 
            # reinit the hub
            from gevent import hub
            hub.reinit()
 
            # then initialize the process
            super(GeventWorker, self).init_process()