From 415956da23db34eacf2b27f27c997d3ca4e0a648 Mon Sep 17 00:00:00 2001
From: Randall Leeds <randall@bleeds.info>
Date: Thu, 9 Oct 2014 10:49:57 +0100
Subject: [PATCH 1/2] Fix eventlet shutdown to actively shut down the workers.

---
 gunicorn/workers/geventlet.py | 57 ++++++++++++++++++++++++++++++-----
 1 file changed, 49 insertions(+), 8 deletions(-)

diff --git a/gunicorn/workers/geventlet.py b/gunicorn/workers/geventlet.py
index c842bb55..280b2c31 100644
--- a/gunicorn/workers/geventlet.py
+++ b/gunicorn/workers/geventlet.py
@@ -5,6 +5,7 @@
 
 from functools import partial
 import errno
+import sys
 
 try:
     import eventlet
@@ -16,9 +17,10 @@ if eventlet.version_info < (0, 9, 7):
     raise RuntimeError("You need eventlet >= 0.9.7")
 
 
-from eventlet import hubs
+from eventlet import hubs, greenthread
 from eventlet.greenio import GreenSocket
 from eventlet.hubs import trampoline
+import greenlet
 
 from gunicorn.http.wsgi import sendfile as o_sendfile
 from gunicorn.workers.async import AsyncWorker
@@ -33,6 +35,47 @@ def _eventlet_sendfile(fdout, fdin, offset, nbytes):
             else:
                 raise
 
+
+def _eventlet_serve(sock, handle, concurrency):
+    """
+    Serve requests forever.
+
+    This code is nearly identical to ``eventlet.convenience.serve`` except
+    that it attempts to join the pool at the end, which allows for gunicorn
+    graceful shutdowns.
+    """
+    pool = eventlet.greenpool.GreenPool(concurrency)
+    server_gt = eventlet.greenthread.getcurrent()
+
+    while True:
+        try:
+            conn, addr = sock.accept()
+            gt = pool.spawn(handle, conn, addr)
+            gt.link(_eventlet_stop, server_gt, conn)
+            conn, addr, gt = None, None, None
+        except eventlet.StopServe:
+            pool.waitall()
+            return
+
+
+def _eventlet_stop(client, server, conn):
+    """
+    Stop a greenlet handling a request and close its connection.
+
+    This code is lifted from eventlet so as not to depend on undocumented
+    functions in the library.
+    """
+    try:
+        try:
+            client.wait()
+        finally:
+            conn.close()
+    except greenlet.GreenletExit:
+        pass
+    except Exception:
+        greenthread.kill(server, *sys.exc_info())
+
+
 def patch_sendfile():
     from gunicorn.http import wsgi
 
@@ -60,16 +103,13 @@ class EventletWorker(AsyncWorker):
 
         super(EventletWorker, self).handle(listener, client, addr)
 
-        if not self.alive:
-            raise eventlet.StopServe()
-
     def run(self):
         acceptors = []
         for sock in self.sockets:
-            sock = GreenSocket(sock)
-            sock.setblocking(1)
-            hfun = partial(self.handle, sock)
-            acceptor = eventlet.spawn(eventlet.serve, sock, hfun,
+            gsock = GreenSocket(sock)
+            gsock.setblocking(1)
+            hfun = partial(self.handle, gsock)
+            acceptor = eventlet.spawn(_eventlet_serve, gsock, hfun,
                     self.worker_connections)
 
             acceptors.append(acceptor)
@@ -82,6 +122,7 @@ class EventletWorker(AsyncWorker):
         self.notify()
         try:
             with eventlet.Timeout(self.cfg.graceful_timeout) as t:
+                [a.kill(eventlet.StopServe()) for a in acceptors]
                 [a.wait() for a in acceptors]
         except eventlet.Timeout as te:
             if te != t:

From 2aabf48317793b20176fae3ee67e4f53872ff62f Mon Sep 17 00:00:00 2001
From: Randall Leeds <randall@bleeds.info>
Date: Tue, 18 Nov 2014 15:03:43 -0800
Subject: [PATCH 2/2] PEP8

---
 gunicorn/workers/geventlet.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/gunicorn/workers/geventlet.py b/gunicorn/workers/geventlet.py
index 280b2c31..48b63fb2 100644
--- a/gunicorn/workers/geventlet.py
+++ b/gunicorn/workers/geventlet.py
@@ -25,6 +25,7 @@ import greenlet
 from gunicorn.http.wsgi import sendfile as o_sendfile
 from gunicorn.workers.async import AsyncWorker
 
+
 def _eventlet_sendfile(fdout, fdin, offset, nbytes):
     while True:
         try:
@@ -82,6 +83,7 @@ def patch_sendfile():
     if o_sendfile is not None:
         setattr(wsgi, "sendfile", _eventlet_sendfile)
 
+
 class EventletWorker(AsyncWorker):
 
     def patch(self):
@@ -99,7 +101,7 @@ class EventletWorker(AsyncWorker):
     def handle(self, listener, client, addr):
         if self.cfg.is_ssl:
             client = eventlet.wrap_ssl(client, server_side=True,
-                **self.cfg.ssl_options)
+                                       **self.cfg.ssl_options)
 
         super(EventletWorker, self).handle(listener, client, addr)
 
@@ -110,7 +112,7 @@ class EventletWorker(AsyncWorker):
             gsock.setblocking(1)
             hfun = partial(self.handle, gsock)
             acceptor = eventlet.spawn(_eventlet_serve, gsock, hfun,
-                    self.worker_connections)
+                                      self.worker_connections)
 
             acceptors.append(acceptor)
             eventlet.sleep(0.0)