为tornado websocket服务器提供这个简单的python代码。
import tornado.httpserver
import tornado.websocket
import tornado.ioloop
import tornado.web
import time
import os
import subprocess
from subprocess import *
import datetime
listeners = []
class WSHandler(tornado.websocket.WebSocketHandler):
def open(self):
print 'New connection was opened'
self.write_message("Con!")
listeners.append(self)
def on_message(self, message):
print 'Received:', message
self.write_message("Received: " + message)
for w in listeners:
w.write_message('MASS MESSAGE')
def on_close(self):
print 'Con closed...'
def trimite(self, msg):
self.write_message(msg)
application = tornado.web.Application([
(r'/ws', WSHandler),
])
if __name__ == "__main__":
http_server = tornado.httpserver.HTTPServer(application)
http_server.listen(65)
tornado.ioloop.IOLoop.instance().start()
每次我从网页执行命令到websocket服务器时,都会向所有连接发送一条消息。它工作正常...但是,如果我关闭并重新连接客户端,我尝试发送另一个命令,我收到此错误,连接关闭。我必须重新启动websocket服务器才能再次工作。任何想法为什么?谢谢!
ERROR:root:Uncaught exception in /ws
Traceback (most recent call last):
File "/usr/lib/python2.7/dist-packages/tornado/websocket.py", line 258, in wrapper
return callback(*args, **kwargs)
File "ciuciu.py", line 26, in on_message
w.write_message('MASS MESSAGE')
File "/usr/lib/python2.7/dist-packages/tornado/websocket.py", line 144, in write_message
self.ws_connection.write_message(message, binary=binary)
AttributeError: 'NoneType' object has no attribute 'write_message'
答案 0 :(得分:0)
好的,想通了。我必须添加listeners.remove:
def on_close(self):
listeners.remove(self)
print 'Con closed...'