我正在尝试创建一个具有几个聊天室的Tornado应用程序。 聊天应该基于HTML5 Websocket。 尽管Websockets之间可以通信,但我总是遇到每条消息都会被发布两次的问题。
该应用程序使用四个类来处理聊天:
- `Chat` 包含到目前为止所有编写的消息,以及一个列表与所有应该被通知的 `waiters` - `ChatPool` 用作新 Websockets 的查找 - 当没有具有所需 `scratch_id` 的用户时,它会创建一个新的聊天或返回现有聊天。 - `ScratchHandler` 是所有HTTP请求的入口点 - 它解析基本模板并返回客户端的所有详细信息。 - `ScratchWebSocket` 查询用户信息的数据库,建立连接,并在必要时通知聊天实例需要传播新消息。
如何防止多次发布消息? 如何使用tornado构建多聊天应用程序?
该应用程序使用四个类来处理聊天:
- `Chat` 包含到目前为止所有编写的消息,以及一个列表与所有应该被通知的 `waiters` - `ChatPool` 用作新 Websockets 的查找 - 当没有具有所需 `scratch_id` 的用户时,它会创建一个新的聊天或返回现有聊天。 - `ScratchHandler` 是所有HTTP请求的入口点 - 它解析基本模板并返回客户端的所有详细信息。 - `ScratchWebSocket` 查询用户信息的数据库,建立连接,并在必要时通知聊天实例需要传播新消息。
如何防止多次发布消息? 如何使用tornado构建多聊天应用程序?
import uuid
import tornado.websocket
import tornado.web
import tornado.template
from site import models
from site.handler import auth_handler
class ChatPool(object):
# contains all chats
chats = {}
@classmethod
def get_or_create(cls, scratch_id):
if scratch_id in cls.chats:
return cls.chats[scratch_id]
else:
chat = Chat(scratch_id)
cls.chats[scratch_id] = chat
return chat
@classmethod
def remove_chat(cls, chat_id):
if chat_id not in cls.chats: return
del(cls.chats[chat_id])
class Chat(object):
def __init__(self, scratch_id):
self.scratch_id = scratch_id
self.messages = []
self.waiters = []
def add_websocket(self, websocket):
self.waiters.append(websocket)
def send_updates(self, messages, sending_websocket):
print "WAITERS", self.waiters
for waiter in self.waiters:
waiter.write_message(messages)
self.messages.append(messages)
class ScratchHandler(auth_handler.BaseHandler):
@tornado.web.authenticated
def get(self, scratch_id):
chat = ChatPool.get_or_create(scratch_id)
return self.render('scratch.html', messages=chat.messages,
scratch_id=scratch_id)
class ScratchWebSocket(tornado.websocket.WebSocketHandler):
def allow_draft76(self):
# for iOS 5.0 Safari
return True
def open(self, scratch_id):
self.scratch_id = scratch_id
scratch = models.Scratch.objects.get(scratch_id=scratch_id)
if not scratch:
self.set_status(404)
return
self.scratch_id = scratch.scratch_id
self.title = scratch.title
self.description = scratch.description
self.user = scratch.user
self.chat = ChatPool.get_or_create(scratch_id)
self.chat.add_websocket(self)
def on_close(self):
# this is buggy - only remove the websocket from the chat.
ChatPool.remove_chat(self.scratch_id)
def on_message(self, message):
print 'I got a message'
parsed = tornado.escape.json_decode(message)
chat = {
"id": str(uuid.uuid4()),
"body": parsed["body"],
"from": self.user,
}
chat["html"] = tornado.escape.to_basestring(self.render_string("chat-message.html", message=chat))
self.chat.send_updates(chat, self)
注意:根据 @A. Jesse 的反馈,我修改了 Chat
中的 send_updates
方法。不幸的是,它仍然返回双倍的值。
class Chat(object):
def __init__(self, scratch_id):
self.scratch_id = scratch_id
self.messages = []
self.waiters = []
def add_websocket(self, websocket):
self.waiters.append(websocket)
def send_updates(self, messages, sending_websocket):
for waiter in self.waiters:
if waiter == sending_websocket:
continue
waiter.write_message(messages)
self.messages.append(messages)
2.编辑: 我将我的代码与提供的演示示例进行了比较。在websocket示例中,通过WebSocketHandler
子类和一个类方法向等待者传播新消息。在我的代码中,使用一个单独的对象完成:
来自示例:
class ChatSocketHandler(tornado.websocket.WebSocketHandler):
@classmethod
def send_updates(cls, chat):
logging.info("sending message to %d waiters", len(cls.waiters))
for waiter in cls.waiters:
try:
waiter.write_message(chat)
except:
logging.error("Error sending message", exc_info=True)
我的应用程序使用对象而不是WebSocketHandler的子类
class Chat(object):
def send_updates(self, messages, sending_websocket):
for waiter in self.waiters:
if waiter == sending_websocket:
continue
waiter.write_message(messages)
self.messages.append(messages)