基于Tornado的websocket

最新推荐文章于 2024-05-14 08:11:14 发布

Mr.＆

最新推荐文章于 2024-05-14 08:11:14 发布

阅读量357

点赞数

分类专栏： Python # Tornado 文章标签： python

本文链接：https://blog.csdn.net/weixin_45317738/article/details/107002775

版权

本文介绍了如何利用Tornado框架创建WebSocket后端服务，并探讨了前端如何与之交互，同时展示了使用Python模拟WebSocket请求的方法。

摘要由CSDN通过智能技术生成

基于Tornado的websocket后端

# 引入需要的模块
from tornado.web import RequestHandler, Application
from tornado.httpserver import HTTPServer
from tornado.websocket import WebSocketHandler
from tornado.ioloop import IOLoop
import json


class BaseHandler(RequestHandler):
    pass


# 定义首页视图处理类，提示用户登录
class IndexHandler(BaseHandler):
    def get(self):
        self.render('index.html')


# 定义登录视图处理类
class LoginHandler(BaseHandler):
    def get(self):
        # 获取用户登录的昵称
        nickname = self.get_argument('nickname')
        # 将用户登录的昵称保存在cookie中，安全cookie
        self.set_secure_cookie('nickname', nickname, expires_days=None)
        self.render('chat.html', nickname=nickname)


# 定义接收/发送聊天消息的视图处理类，继承自websocket的WebSocketHandler
class ChatHandler(WebSocketHandler):
    # 定义一个集合，用来保存在线的所有用户
    online_users = set()

    # 从客户端获取cookie信息

    # 添加此get方法后,websocket无法使用
    # def get(self, *args, **kwargs):
    #     name = self.get_argument('name', 'a')
    #     if name:
    #         self.on_message(name)
    #         self.write("Hello, world")

    # 重写open方法，当有新的聊天用户进入的时候自动触发该函数
    def open(self):
        # 只给刚连接的用户发消息
        self.write_message('欢迎光临')

        # 连接时可携带参数,可用于鉴权
        id = self.get_argument('id', '111')
        print(id)
        
        # 当有新的用户上线，将该用户加入集合中
        self.online_users.add(self)

        # 将新用户加入的信息发送给所有的在线用户