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] 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: