gevent: Make sure synchronous calls are blocking

kombu/utils/__init__.py

@@ -12,23 +12,44 @@ except ImportError:
     ctypes = None
 
 
-blocking = None
+def __how_to_block():
-try:
+    ## -eventlet-
-    from eventlet.patches import is_monkey_patched as is_eventlet
+    try:
-    import socket
+        from eventlet.patches import is_monkey_patched as is_eventlet
+        import socket
 
-    if is_eventlet(socket):
+        if is_eventlet(socket):
-        from eventlet import spawn
+            from eventlet import spawn
 
-        def blocking(fun, *args, **kwargs):
+            def __blocking__(fun, *args, **kwargs):
-            return spawn(fun, *args, **kwargs).wait()
+                return spawn(fun, *args, **kwargs).wait()
-except ImportError:
-    pass
 
-if blocking is None:
+            return __blocking__
-    def blocking(fun, *args, **kwargs):
+    except ImportError:
+        pass
+
+    # -gevent-
+    try:
+        from gevent import socket as _gsocket
+        import socket
+
+        if socket.socket is _gsocket.socket:
+            from gevent import Greenlet
+
+            def __blocking__(fun, *args, **kwargs):
+                return Greenlet.spawn(fun, *args, **kwargs).get()
+
+            return __blocking__
+    except ImportError:
+        pass
+
+    def __blocking__(fun, *args, **kwargs):
         return fun(*args, **kwargs)
 
+    return __blocking__
+
+blocking = __how_to_block()
+
 
 def say(m, *s):
     sys.stderr.write(str(m) % s + "\n")