Question

我正在玩gevent和websockets。这是一个简单的echo服务器：

from gevent.pywsgi import WSGIServer
from geventwebsocket.handler import WebSocketHandler
from gevent import sleep
from datetime import datetime
def app(environ, start_response):
    ws = environ['wsgi.websocket']
    while True:
        data = ws.receive()
        print('{} got data "{}"'.format(
            datetime.now().strftime('%H:%M:%S'), data))
        sleep(5)
        ws.send(data)

server = WSGIServer(("", 10004), app,
    handler_class=WebSocketHandler)
server.serve_forever()

和客户：

<html>
    <body>
        <button type="button" id="push_data">Push</button>
    </body>
    <script type="text/javascript" src="http://code.jquery.com/jquery-1.11.3.js"></script>
    <script>
        var ws = new WebSocket("ws://localhost:10004");
        ws.onmessage = function(evt) {
            console.log(evt)
        };
        $('#push_data').click(function(){
            console.log('sending data...');
            ws.send('sample data');
        });
    </script>
</html>

由于gevent我期待有几个greenlets以异步方式提供数据;也就是说，如果我将一些数据推送到websocket几次（快速点击Push按钮），我希望在等待5秒后同时将它全部恢复。

但是，无论我点击按钮的速度有多快，这都是我在控制台中获得的：

18:28:07 got data "sample data"
18:28:12 got data "sample data"
18:28:17 got data "sample data"
18:28:22 got data "sample data"
18:28:27 got data "sample data"

为什么它会同步接收我的数据，每隔5秒暂停一次？如何将其变为异步服务器？

Answer 1

行为是同步的，因为您自己的代码是同步的。 gevent只是一个使用事件循环的协程库。它并没有神奇地将同步代码转换为异步代码。

请查看以下文档：http://www.gevent.org/servers.html

据说服务器每个连接产生一个greenlet （不是每个请求）。因此，序列化了对同一连接的多个请求的执行。

如果要同时处理同一连接的多个请求，则需要生成新的greenlet，或将处理委托给greenlet池。

以下是一个示例（在每个请求中生成一个greenlet）：

import gevent
from gevent.pywsgi import WSGIServer
from gevent.lock import Semaphore
from geventwebsocket.handler import WebSocketHandler
from datetime import datetime

def process(ws,data,sem):
    print('{} got data "{}"'.format(datetime.now().strftime('%H:%M:%S'), data))
    gevent.sleep(5)
    with sem:
        ws.send(data)

def app(environ, start_response):
    ws = environ['wsgi.websocket']
    sem = Semaphore()
    while True:
        data = ws.receive()
        gevent.spawn(process,ws,data,sem)

server = WSGIServer(("", 10004), app,handler_class=WebSocketHandler)
server.serve_forever()

注意信号量的存在。由于处理是并发的，因此需要防止两个并发的greenlet同时在套接字上写入，从而导致消息损坏。

最后一点，通过此实现，无法保证将按请求的顺序发送回复。

Answer 2

实际问题是：data = ws.receive（）

这里发生的事情是你的websocket正在等待一个SINGLE连接，而整个应用程序只是挂断了。

你有两个解决方案，要么为ws.receive（）添加超时，要么将其设置为高级应用程序：

from geventwebsocket import WebSocketServer, WebSocketApplication, Resource

class EchoApplication(WebSocketApplication):
    def on_open(self):
        print "Connection opened"

    def on_message(self, message):
        self.ws.send(message)

    def on_close(self, reason):
        print reason

WebSocketServer(('', 8000), Resource({'/': EchoApplication}).serve_forever()

例如：https://pypi.python.org/pypi/gevent-websocket/

这将使您的过程完全异步，因此发送和接收不会竞争相同的资源。

为什么gevent-websocket同步？

2 个答案: