中间件
在flask中,在执行
app.run()之后,如果有请求来,会执行app.__call__方法通过中间件的方法可以在请求来时之前与来临之后,即执行
app.__call__前后执行其他代码,进行一些生命周期的控制
- 根据
werkzeug我们可以知道 每次请求必然经历了Flask.app - 所以我们要查看Flask的源码发现run方法实际的调用是
app.run()->run_simple()->self()__call__->self.wsgi_app(environ, start_response)
- 找到了Flask的
__call__方法后发现执行了return self.wsgi_app(environ, start_response)
flask里面所有的内容调度都是基于这个self.wsgi_app(environ, start_response),这就是就是flask的入口,也就是self是app,也就是app.wsgi_app(environ, start_response)为程序的入口。
如何实现中间件呢? 原理上就是重写app.wsgi_app,然后在里面添加上一些自己想要实现的功能。
首先分析 app.wsgi_app需要加括号执行 所以我们把app.wsgi_app做成一个对象,并且这个对象需要加括号运行,也就是会触发这个对象的类的__call__()方法
class MiddleWare:
def __init__(self, wsgi_app):
self.wsgi_app = wsgi_app
def __call__(self, *args, **kwargs):
print("....")
return self.wsgi_app(*args, **kwargs)
app.wsgi_app = MiddleWare(app.wsgi_app)