Python的Tornado框架实现异步非阻塞访问数据库的示例

左手的ㄟ右手 2023-05-28 05:26 20阅读 0赞

Tornado框架的异步非阻塞特性是其最大的亮点,这里我们将立足于基础来介绍一种简单的Python的Tornado框架实现异步非阻塞访问数据库的示例:  
tornado即是一个http非阻塞服务器, 就要用起来, 我们将用到tornado框架 ,mongodb数据库 以及motor(mongodb的异步驱动).来简单实现tornado的非阻塞功能.

其他环境支持的下载与安装

1.安装mongodb

$ sudo apt-get install update
    $ sudo apt-get install mongodb

2.安装motor

$ pip install motor

非阻塞

# conf.py
     
    import os
    import motor
    from handlers import index, auth
     
    BASE_DIR = os.path.join(__file__)
     
    handlers = [
      (r'^/$', index.IndexHandler),
      (r'^/auth/register$', auth.RegisterHandler),
      (r'^/auth/login$', auth.LoginHandler),
    ]
     
    settings = dict(
      debug = True,
      template_path = os.path.join(BASE_DIR, 'templates'),
      static_path = os.path.join(BASE_DIR, 'static'),
    )
     
    client = motor.MotorClient("127.0.0.1")
    db = client.meet

首先在配置文件中连接数据库, client.db\_name中 db\_name就是数据库的名称

# handlers/__init__.py
    class BaseHandler(tornado.web.RequestHandler, TemplateRendering):
      def initialite(self):
        ...
     
      @property
      def db(self):
        return self.application.db

添加db()并使用property装饰,像属性一样访问数据库.

# auth.py
     
    import os 
    import time 
    import tornado.web
    from tornado import gen
    from . import BaseHandler
     
    class RegisterHandler(BaseHandler):
      def get(self):
        self.render_html('register.html')
     
      @tornado.web.asynchronous
      @gen.coroutine
      def post(self):
        username = self.get_argument('username', None)
        email = self.get_argument('email', None)
        password = self.get_argument('password', None)
     
        data = { 
          'username': username,
          'email': email,
          'password': password,
          'timestamp': time.time() * 1000,
        }
     
        if username and email:
          yield self.db.user.insert(data)
        self.redirect('/')
     
    class LoginHandler(BaseHandler):
       
      @tornado.web.asynchronous
      @gen.coroutine
      def get(self):
        username = self.get_argument('useranme')
        user = yield self.db.user.find_one({ 'username': username})
        self.render_html('login.html', user=user)

@gen.coroutine装饰使函数非阻塞, 返回一个生成器, 而不用在使用回调函数. motor也通过yield 实现异步(不然还得返回一个回调函数). 其实这个例子反映不了阻塞问题关键是时间太短.  
我们修改一下代码

# 之前
    yield self.db.user.insert(data)
     
    # 之后
    yield tornado.gen.Task(tornado.ioloop.IOLoop.instance().add_timeout, time.time() + 10)

这里通过tornado.ioloop.IOLoop.instance().add\_timeout阻塞应用, 这是time.sleep的非阻塞实现, 如果这里使用time.sleep因为是tornado是单线程会阻塞整个应用所以别的handler也无法访问.  
可以看到我在注册页面注册后,在阻塞期间点击/auth/login直接就访问了login页完成非阻塞.

异步下的redirect问题  
在使用tornado的时候常常遇到一些问题, 特将遇到的问题和解决的方法写出来(这里的感谢一下帮我解答疑惑的pythonista们)

1.问题

我想要实现一个注册用户功能, web框架使用tornado数据库使用mongodb但在注册时出现Exception redirect的错误. 现贴下代码:

class Register(BaseHandler):
      def get(self):
        self.render_html('register.html')
     
      @tornado.web.aynchronous
      @gen.coroutine
      def post(self):
        username = self.get_argument('username')
        email = self.get_argument('email')
        password = self.get_argument('password')
        captcha = self.get_argument('captcha')
     
        _verify_username = yield self.db.user.find_one({ 'username': username})
        if _verify_username:
          self.flash(u'用户名已存在', 'error')
          self.redirect('/auth/register')
     
        _verify_email = yield self.db.user.find_one({ 'email': email})
        if _verify_email:
          self.flash(u'邮箱已注册', 'error')
          self.redirect('/auth/register')
     
        if captcha and captcha == self.get_secure_cookie('captcha').replace(' ',''):
          self.flash(u'验证码输入正确', 'info')
        else:
          self.flash(u'验证码输入错误', 'error')
          self.redirect('/auth/register')
     
        password = haslib.md5(password + self.settings['site']).hexdigest()
     
        profile = { 'headimg': '', 'site': '', 'job': '', 'signature':'',
              'github': '', 'description': ''}
        user_profile = yield self.db.profile.insert(profile)
        user = { 'username': username, 'email': email, 'password': password,
            'timestamp': time.time(), 'profile_id': str(user_profile)}
     
        yield self.db.user.insert(user)
        self.set_secure_cookie('user', username)
        self.redirect('/')

本想如果用户验证码输入出错就跳转到注册页面, 但问题是验证码出错也会继续执行一下代码. 虽然在self.redirect后加上self.finish会终止代码,但是因为self.redirect 函数内已有self.finish所以出现了两次报出异常终止的代码.  
因为以上原因代码不会被终结, 验证码出错用户还是会注册.

2.解决方案

return self.redirect('/auth/register')

或

self.redirect('/auth/register')
    return

1）segmentdefault中热心用户rsj217给出的答案  
self.finish 会关掉请求, 因为@tornado.web.aynchronous告诉tornado会一直等待请求(长链接). self.redirect等于设置了response的headers的location属性.

（2）segmentdefault中热心用户依云给出的答案  
self.finish当然不会跳出函数, 不然请求结束之后还想做些事情怎么办呢.

3.总结

因为错把self.finish当做跳出函数出现了以上的问题

self.redirect会在request.headers 里设置location用于跳转  
self.finish会关掉请求, 但不会跳出函数  
最后给大家推荐一个口碑不错的python聚集地【[点击进入][Link 1]】，这里有很多的老前辈学习技巧，学习心得

，面试技巧，职场经历等分享，更为大家精心准备了零基础入门资料，实战项目资料，每天都有程序员

定时讲解Python技术，分享一些学习的方法和需要留意的小细节

[Link 1]: https://jq.qq.com/?_wv=1027&k=5cb1kUZ