tornado+websocket+mongodb实现在线视屏文字聊天

最近学了tornado和mongo,所以结合websocket 实现一个聊天功能,从而加深一下相关知识点的印象

1.websocket概览

webscoket是一种全双工通信模式的协议,客户端连接服务端先通过tcp,http转为webscoket协议后,客户端和服务端都可以主动推送消息给另一端,这也是和http协议(服务端只能被动接收消息,无法主动推送消息给客户端)最大的区别。

2.tornado概览

tornado是一种异步网络库的python web框架,最初在 FriendFeed上开发,通过使用非阻塞网络I/O,tornado可以扫描数以万计打开的链接,让它成为给每个用户一个长链接的理想选择。

Tornado is a Python web framework and asynchronous networking library, originally developed at FriendFeed. By using non-blocking network I/O, Tornado can scale to tens of thousands of open connections, making it ideal for long polling, WebSockets, and other applications that require a long-lived connection to each user.

3.实现在线聊天功能概览

  • 判断对方是否在线,保存离线消息,在线时推送过去,并删除离线消息
  • 不保存在线消息
  • 单点登录模式的聊天
  • 同时和多人点对点聊天

 4.后端代码

import redis
import tornado.httpserver
import tornado.ioloop
import tornado.web
from motor import motor_tornado
from tornado import websocket
from tornado.gen import coroutine

from conf import Config


class WebHandler(websocket.WebSocketHandler):
    # 在线用户dict
    all_user = {}

    def initialize(self, mongo, redis):
        self.mongo = mongo
        self.redis = redis

    @property
    def db(self):
        return self.mongo.get_database('rgc')

    @property
    def col(self):
        return self.db.get_collection('web')

    def check_origin(self, origin):
        return True  # 允许WebSocket的跨域请求

    @coroutine
    def on_message(self, message):
        #因为没有登录相关功能,每次传输都 用 # 拼接 发送者,消息,接受者
        resu = str.split(message, '#')

        name = resu[0]
        val = resu[1]
        to = resu[2]
        # 判断对方是否在线
        if 'name:{}'.format(to) not in list(self.all_user.keys()):
            self.write_message('Out Line')
            # 存储消息到db
            # if not self.redis.hget('name_list', '{}:{}'.format(name, to)):
            #     self.redis.hset('name_list', '{}:{}'.format(name, to), 1)
            self.col.insert({'name': '{}:{}'.format(name, to), 'msg': val})
        else:
            # 发送最新消息
            self.all_user['name:{}'.format(to)].write_message(val)

        # 检查是不是第一次上线
        if 'name:{}'.format(name) not in list(self.all_user.keys()):
            # 给自己发送历史消息
            his_one = self.col.find({'name': '{}:{}'.format(to, name)}, {'msg': 1, '_id': 0})
            for it in (yield his_one.to_list(100)):
                self.write_message(it['msg'])
        # 删除历史消息
        self.col.delete_many({'name': '{}:{}'.format(to, name)})
        # 单点登录聊天
        self.all_user.update({'name:{}'.format(name): self})
        # 发给自己
        self.write_message('send success')

    def open(self):
        pass

    def on_close(self):
        # 当客户端关闭连接时,去除内存中保存的用户,让其离线
        key = None
        for k, v in self.all_user.items():
            if v == self:
                key = k
                break
        if key:
            self.all_user.pop(key)
        print('{} out line'.format(key))


class HtmlHandler(tornado.web.RequestHandler):
    def get(self):
        self.render("static/index.html")


class StaticHandler(tornado.web.RequestHandler):
    def get(self, file_url):
        self.render("static/{}".format(file_url))


def make_app():
    settings = {'cookie_secret': 'dfdfdfd',
                'xsrf_cookies': True,
                'debug': True}
    other_db = {'mongo': motor_tornado.MotorClient(**Config.get('MONGO_CONF')),
                'redis': redis.StrictRedis()}
    return tornado.web.Application([
        (r'/web', WebHandler, other_db),
        (r'/', HtmlHandler),
        (r'/static/(.*)', StaticHandler)
    ], **settings)


if __name__ == '__main__':
    app = make_app()
    http_server = tornado.httpserver.HTTPServer(app)
    ip='127.0.0.1'
    port = 8000
    http_server.bind(8000, ip)
    http_server.start(1)
    print('server start! http://{}:{}'.format(ip, port))
    tornado.ioloop.IOLoop.current().start()

 5.前端技术:

前端主要使用到了  MediaSource,Blob 等技术,发送视频大致思路为:先获取视频文件发送给服务端,然后从服务端发送给另一个客户端,客户端进行视频解析后播放出来。

效果展示:(因为开启浏览器视频功能,必须在https环境或者本地回环地址(127.0.0.1)中才可),所以本次效果展示是在本地进行展示

  • 谷歌浏览器(因为我是后端开发,前端代码没有做兼容,并且界面有点丑)开启两个网页,输入  http://127.0.0.1:8000/
  • 进入界面后,在name里输入自己的姓名,message中输入 发送给对方的消息,to 里面输入 对方名字,然后点击发送
  • lucy发送工tom的信息,在发送信息时,系统检测到tom给她发了离线消息,所以一并返回给lucy。
  • 视频发送测试,name和to用的还是之前的:
  • 看一下 浏览器控制台 websocket接口的相关内容:
  • 可以看到所有的交流都是在一个 websocket连接中,双方可以互发消息

 项目代码所在地: https://github.com/Rgcsh/tornado_websocket

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏京东技术

如何实现靠谱的分布式锁?(附SharkLock的设计选择)

当前使用较多的分布式锁方案主要基于 Redis、ZooKeeper 提供的功能特性加以封装来实现的,下面我们会简要分析下这两种锁方案的处理流程以及它们各自的问题...

1703
来自专栏铭毅天下

干货 | Elasticsearch Reindex性能提升10倍+实战

reindex和snapshot的速率比用filebeat或者kafka到es的写入速率慢好几个数量级(集群写入性能不存在瓶颈),reindex/snapsho...

2192
来自专栏后台及大数据开发

Linux巩固记录(9) keepalived+nginx搭建高可用负载分发环境

web 192.168.2.100 再iis上启动了9011/9012/9013三个端口的web应用

1051
来自专栏程序员八阿哥

Nginx与前端开发

作为前端开发,即使没用过Nginx,但一定听说过上面这句话。这句经典的话,基本构成了所有人对Nginx的第一印象。

2981
来自专栏云成本管理

云成本管理方法论(三)——云优化管理之判定规则

云优化管理四个管理维度中管理时点在通用管理模型基础上不需要额外补充,所以主要说明其他三个维度(管理对象、判定规则和管理措施)。另外,为了贴近我们熟悉的优化概念,...

52420
来自专栏木头编程 - moTzxx

Linux crontab 定时任务整理笔记

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/u011415782/article/de...

1772
来自专栏IMWeb前端团队

当 badjs 遇上 sourcemap

本文作者:IMWeb 九月 原文出处:IMWeb社区 未经同意,禁止转载 在开始之前,如果你不知道 sourcemap 是什么,希望你能先了解,可以看...

3185
来自专栏CodingBlock

Mac下搭建Eclipse Android开发环境

  之前一直是用windows搞android开发,但windows这个性能也真是让人醉了,终于一狠心,砸锅卖铁买了Mac。然后就开始在Mac上搭建androi...

40110
来自专栏网络

服务化基石之远程通信系列二:通信协议之应用层

通信协议之应用层 应用层包含所有的高层协议,例如FTP (File Transfer Protocol的简写,中文名称是文件传输协议)、SMTP (Simple...

2165
来自专栏LuckQI

让你不会装双系统也能用上双系统

这个软件是在公司看到领导在用,才知道原来Windows,已经支持Linux系统这么强大了。使用的名字是WSL。我们也可以百度下这个怎么使用。

5041

扫码关注云+社区

领取腾讯云代金券