Flask中间件

47 阅读1分钟

中间件

在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)