python+tornado 实现消息推送功能【精简版】

最新推荐文章于 2025-11-20 09:04:36 发布

原创最新推荐文章于 2025-11-20 09:04:36 发布 · 5.7k 阅读

11 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#python #异步 #tornado

异步专栏收录该内容

3 篇文章

订阅专栏

本文介绍了一个基于Tornado框架实现的简单实时消息推送系统，包括入口文件、消息推送逻辑、模板文件和注册队列执行发送的流程。通过浏览器访问指定URL接收消息，发送消息则通过特定页面实现。

这个是没有任何安全设置，仅仅展示消息推送的模型，让初学者了解原理，本代码基于浏览器！

首先实现main.py文件，这个是入口文件：

main.py：

<pre name="code" class="python">define("port", default=8888, type=int, help="")

class MyApplication(tornado.web.Application):
    def __init__(self):
        handlers = [(r'/',pullServer),(r'/send', pushServer)]
        setting = dict(
            debug=True,
            template_path=os.path.join(os.path.dirname(__file__), "template"),
        )
        tornado.web.Application.__init__(self, handlers, **setting)



def main():
    tornado.options.parse_command_line()
    http_server = tornado.httpserver.HTTPServer(MyApplication())
    http_server.listen(options.port)
    tornado.ioloop.IOLoop.instance().start()

if  __name__ == "__main__":
    main()

第二个文件是请求与推送消息文件，我把他们放在同一个文件里了

PullAndPush.py

class pullServer(tornado.web.RequestHandler):
    @tornado.web.asynchronous  #这个的作用是使用异步，让get不自动断开连接，直到接到服务器推送过来的消息后，再self.finish()，断开连接
    def get(self):
        uid = self.get_argument("uid")
        Message.add_uid(uid, self.callback)
    
    #这一步很关键，当浏览器发送请求后，这个回调函数就会被注册到消息队列中，当服务器发送消息后，会调用所有已经被注册的callback（），把消息推到浏览器上
    def callback(self, message):
        self.write(message)
        self.finish()
#发送消息页面
class pushServer(tornado.web.RequestHandler):
    def get(self):
        self.render("index.html")

    def post(self):
        msg = self.get_argument("message")
        #通过send后，所有已注册的回调函数都会被执行，把“message”推送到浏览器
        Message.send(msg)

第三个文件是注册队列和执行发送

Message.py

UidList = {}  #保存回调函数，即需要推送消息的都在这里

#添加队列
def add_uid(uid,callback):
    if uid not in UidList:
        UidList[uid] = callback

#发送消息
def send(message):
    for i in UidList:
        callback = UidList[i]
        callback(message)

第四个文件是模板文件，没什么可解析的，不熟悉的可以先学会在看本案例

<pre name="code" class="html"><!DOCTYPE html>
<html>
<head>
    <title>发送消息</title>
</head>
<body>
<form action="/send" method="post" enctype="multipart/form-data">
    <input type="text" name="message" />
    <input type="submit" value="发送"/>
</form>
</body>
</html>

启动文件，在浏览器输入

http://127.0.0.1:8888/?uid=1252473 来等待消息的到来，

在打开http://127.0.0.1:8888/send 来发送消息