updated to match middleware's interface

Lawouach · Lawouach · commit 95ef8ff4c935 · 2012-02-19T20:41:53.000+01:00
diff --git a/example/echo_gevent_server.py b/example/echo_gevent_server.py
@@ -1,9 +1,11 @@
 # -*- coding: utf-8 -*-
+from gevent import monkey; monkey.patch_all()
+
 import argparse
 import random
 import os
 
-from gevent import monkey; monkey.patch_all()
+import gevent
 import gevent.pywsgi
 
 from ws4py.server.geventserver import UpgradableWSGIHandler
@@ -24,7 +26,8 @@ def __init__(self, host, port):
         # let's use wrap the websocket handler with
         # a middleware that'll perform the websocket
         # handshake
-        self.ws = WebSocketUpgradeMiddleware(websocket_class=EchoWebSocket)
+        self.ws = WebSocketUpgradeMiddleware(app=self.ws_app,
+                                             websocket_class=EchoWebSocket)
 
         # keep track of connected websocket clients
         # so that we can brodcasts messages sent by one
@@ -47,6 +50,11 @@ def __call__(self, environ, start_response):
         
         return self.webapp(environ, start_response)
 
+    def ws_app(self, websocket):
+        g = gevent.spawn(websocket.run)
+        g.start()
+        g.join()
+
     def favicon(self, environ, start_response):
         """
         Don't care about favicon, let's send nothing.
