黑马头条项目 9.5 头条在线消息推送实现

头条在线消息推送实现

需求

在头条的Flask应用中，用户关注后需要推送消息，通过消息队列告知IM服务为用户进行推送

 

在Socket.IO 框架中可以选择使用以下两种方式作为消息中间件：

• 使用Redis

    mgr = socketio.RedisManager('redis://')
    sio = socketio.Server(client_manager=mgr)
  

• 使用RabbitMQ

    pip install kombu
  

    mgr = socketio.KombuManager('amqp://')
    sio = socketio.Server(client_manager=mgr)
  

实现

因为要给指定的用户推送消息，所以需要用到用户的身份，用户在客户端携带JWT连接SocketIO服务器，我们在服务器端对jwt token进行验证，对于验证出用户身份(user_id)的客户端，将其添加到名为用户id的room房间中，方便按照user_id进行推送。

socketio服务端编写

在toutiao-backend/im/main.py中补充添加搜寻包的路径，方便使用utils中的jwt_utils模块