Python的Tornado框架实现异步非阻塞访问数据库的示例
tornado即是一个http非阻塞服务器,就要用起来,我们将用到tornado框架,mongodb数据库以及motor(mongodb的异步驱动).来简单实现tornado的非阻塞功能.
其他环境支持的下载与安装
1.安装mongodb
$sudoapt-getinstallupdate $sudoapt-getinstallmongodb
2.安装motor
$pipinstallmotor
非阻塞
#conf.py
importos
importmotor
fromhandlersimportindex,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 classBaseHandler(tornado.web.RequestHandler,TemplateRendering): definitialite(self): ... @property defdb(self): returnself.application.db
添加db()并使用property装饰,像属性一样访问数据库.
#auth.py
importos
importtime
importtornado.web
fromtornadoimportgen
from.importBaseHandler
classRegisterHandler(BaseHandler):
defget(self):
self.render_html('register.html')
@tornado.web.asynchronous
@gen.coroutine
defpost(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,
}
ifusernameandemail:
yieldself.db.user.insert(data)
self.redirect('/')
classLoginHandler(BaseHandler):
@tornado.web.asynchronous
@gen.coroutine
defget(self):
username=self.get_argument('useranme')
user=yieldself.db.user.find_one({'username':username})
self.render_html('login.html',user=user)
@gen.coroutine装饰使函数非阻塞,返回一个生成器,而不用在使用回调函数.motor也通过yield实现异步(不然还得返回一个回调函数).其实这个例子反映不了阻塞问题关键是时间太短.
我们修改一下代码
#之前 yieldself.db.user.insert(data) #之后 yieldtornado.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但在注册时出现Exceptionredirect的错误.现贴下代码:
classRegister(BaseHandler):
defget(self):
self.render_html('register.html')
@tornado.web.aynchronous
@gen.coroutine
defpost(self):
username=self.get_argument('username')
email=self.get_argument('email')
password=self.get_argument('password')
captcha=self.get_argument('captcha')
_verify_username=yieldself.db.user.find_one({'username':username})
if_verify_username:
self.flash(u'用户名已存在','error')
self.redirect('/auth/register')
_verify_email=yieldself.db.user.find_one({'email':email})
if_verify_email:
self.flash(u'邮箱已注册','error')
self.redirect('/auth/register')
ifcaptchaandcaptcha==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=yieldself.db.profile.insert(profile)
user={'username':username,'email':email,'password':password,
'timestamp':time.time(),'profile_id':str(user_profile)}
yieldself.db.user.insert(user)
self.set_secure_cookie('user',username)
self.redirect('/')
本想如果用户验证码输入出错就跳转到注册页面,但问题是验证码出错也会继续执行一下代码.虽然在self.redirect后加上self.finish会终止代码,但是因为self.redirect函数内已有self.finish所以出现了两次报出异常终止的代码.
因为以上原因代码不会被终结,验证码出错用户还是会注册.
2.解决方案
returnself.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会关掉请求,但不会跳出函数