我在玩gevent和websocket。这是一个简单的回声服务器:
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
,我期望有几个greenlet异步地提供数据;也就是说,如果我多次将一些数据推送到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秒暂停一次?如何将其转换为异步服务器?
行为是同步的,因为您自己的代码是同步的。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同时在套接字上写入,从而导致消息损坏。
最后一点,通过这种实现,不能保证答复将按照请求的顺序发送。
实际问题是:data=ws.rereceive()
这里发生的情况是,你的网络套接字正在等待一个单一的连接,而整个应用程序只是挂断了。
您有两种解决方案,要么向ws.rereceive()添加超时,要么将其设置为高级应用程序:
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/
这将使您的进程完全异步,因此发送和接收不会竞争相同的资源。