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也无法访问. 异步下的redirect问题 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,'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给出的答案 （2）segmentdefault中热心用户依云给出的答案 3.总结 因为错把self.finish当做跳出函数出现了以上的问题
（编辑：李大同） 【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容! |