Flask上下文管理机制流程（源码剖析）

Flask请求上下文管理

1 偏函数

• partial 使用该方式可以生成一个新函数

  from functools import partial
  
  def mod( n, m ):
    return n % m
  
  mod_by_100 = partial( mod, 100 )  # 100传给n
  
  print mod( 100, 7 )  # 2
  print mod_by_100( 7 )  # 2

2 线程安全

import time
from threading import local

class Foo(local): 
# 继承local,保证线程安全,也保证了处理速度,threading.local()这个方法的特点用来保存一个全局变量，但是这个全局变量只有在当前线程才能访问，
    num = 0
    
foo = Foo()
def add(i):
    foo.num =i
    time.sleep(0.5)
    print(foo.num)

from threading import Thread
for i in range(20):
    task = Thread(target=add,args=(i,))
    task.start()
    

3 请求上下文

3.1 Flask请求上文

1. 当请求进来时,app(), Flask实例化对象app**执行__call__**

def __call__(self, environ, start_response):
    """The WSGI server calls the Flask application object as the
        WSGI application. This calls :meth:`wsgi_app` which can be
        wrapped to applying middleware."""
    return self.wsgi_app(environ, start_response)

1. 执行wsgi_app 得到 一个 RequestContext的对象 ctx (封装了request以及session)

   ctx = self.request_context(environ)

   class RequestContext(object):
       #此时的self是RequestContext对象 -->ctx 中封装了request/session
       def __init__(self, app, environ, request=None):
           self.app = app      #app = Flask对象
           if request is None:
               #请求的原始信息通过request_class后此时request已经存在,request.methods等
               request = app.request_class(environ)
           self.request = request
           self.url_adapter = app.create_url_adapter(self.request)
           self.flashes = None
           self.session = None

2. ctx 执行 ctx.push():

   ctx = self.request_context(environ)
           error = None
           try:
               try:
                   ctx.push()
                   response = self.full_dispatch_request()
               except Exception as e:
                   error = e
                   response = self.handle_exception(e)

3. RequestContext对象的push方法

   def push(self):
           # _request_ctx_stack = LocalStack()一个LocalStack对象
           # _request_ctx_stack._local = LocalStack()._loacl = {"__storage__":{},"__ident_func__":get_ident}
           top = _request_ctx_stack.top
           #top =None
           if top is not None and top.preserved:
               top.pop(top._preserved_exc)
   

   • _ request_ctx_stack是一个LocalStack对象 ,LocalStack()._local是一个Local对象 即Local()

   class LocalStack(object):
       def __init__(self):
           self._local = Local()
           #self._loacl = {"__storage__":{},"__ident_func__":get_ident}

   • _request_ctx_stack中top方法,返回None (想哭,但是没有眼泪,这个方法,琢磨了半个小时)

4. Local对象经过初始化得到的字典值

   class Local(object):
       #限定键槽,当前只能由两个属性值__storage__,__ident_func__
       __slots__ = ('__storage__', '__ident_func__')
   
       def __init__(self):
           object.__setattr__(self, '__storage__', {})
           object.__setattr__(self, '__ident_func__', get_ident)
   
           # {"__storage__":{},"__ident_func__":get_ident}  #此时get_dient 是个没有执行的函数,内存地址

   • _request_ctx_stack中top方法,返回None (第二次不会上当)

   @property
       def top(self):
           """The topmost item on the stack.  If the stack is empty,
           `None` is returned.
           """
           try:
               # self._local 即Local对象调用__getattr__方法
               #在下文时候Local对象{"__storage__":{8080:{stack:rv=[ctx->request/session]}},"__ident_func__":get_ident}
               # [ctx->request/session]
               return self._local.stack[-1]
               #得到ctx对象
           except (AttributeError, IndexError):
               return None

5. _request_ctx_stack 对象执行push方法

   _request_ctx_stack.push(self)  #当前的self为ctx

   def push(self, obj):
           #此时的self是LocalStack对象, obj为ctx
           """Pushes a new item to the stack"""
           # self._local = {"__storage__":{},"__ident_func__":get_ident}
           #找不到返回值是None
           rv = getattr(self._local, 'stack', None)
           if rv is None:
               #由于.stack后面有等号,执行的时候Local()对象的__setattr__方法
               #实际上是地址的赋值,此时stack和rv都指向空列表
               self._local.stack = rv = []
               #{"__storage__":{8080:{stack:rv=[]}},"__ident_func__":get_ident}
           rv.append(obj)
           # {"__storage__":{8080:{stack:rv=[ctx->request/session]}},"__ident_func__":get_ident}
           # 应用上下文时候{"__storage__":{8080:{stack:rv=[app_ctx->(app/g)]}},"__ident_func__":get_ident}
           return rv
           #rv=[ctx->request/session]

   def __setattr__(self, name, value):
           #name=stack   value=rv=[]
           #self是Local对象 {"__storage__":{},"__ident_func__":get_ident}
           ident = self.__ident_func__() #执行get_ident函数获取当前线程id 8080
           storage = self.__storage__  #storge ={8080:{stack:rv=[]}}
           try:
               storage[ident][name] = value
           except KeyError:
               storage[ident] = {name: value}   #storage={}
   
            # {"__storage__":{8080:{stack:rv=[]}},"__ident_func__":get_ident}

   • 执行完push方法 请求上文结束:

   #当请求进来,第一件事就是要把当前这个请求在我服务器上的线程开辟一个空间(线程对应的空间,必须含有stack对应一个列表存放ctx(request/session)
   # {"__storage__":{8080:{stack:rv=[ctx->request/session]}},"__ident_func__":get_ident}

3.3.2Flask请求下文

1. 导入request开始使用,在request中

   #此时request是一个函数包裹一个偏函数   LocalProxy()是一个代理
   #当前的request是一个LocalProxy()  request.method  执行__getattr__方法
   request = LocalProxy(
       partial(_lookup_req_object, 'request')   #return request对象
   )

2. 在偏函数中 将request传入到 _lookup_req_object中: 此时得到一个request对象

   def _lookup_req_object(name):
       # _request_ctx_stack是LocalStack对象
       top = _request_ctx_stack.top
       #下文[ctx->request/session]
       if top is None:
           raise RuntimeError(_request_ctx_err_msg)
       #此时的name是request,从ctx对象中找出request对象
       return getattr(top, name)

   ...

   @property
       def top(self):
           """The topmost item on the stack.  If the stack is empty,
           `None` is returned.
           """
           try:
               # self._local 即Local对象调用__getattr__方法
               #在下文时候Local对象{"__storage__":{8080:{stack:rv=[ctx->request/session]}},"__ident_func__":get_ident}
               # [ctx->request/session]
               return self._local.stack[-1]
               #得到ctx对象
           except (AttributeError, IndexError):
               return None

   • 此时的top不是None已经存在值 (0.0)

3. partial(_lookup_req_object, 'request') 这一层执行完得到一个reauest对象,将偏函数传入到LocalProxy中

   @implements_bool
   class LocalProxy(object):
       __slots__ = ('__local', '__dict__', '__name__', '__wrapped__')
   
       def __init__(self, local, name=None):
           #local是request偏函数
           object.__setattr__(self, '_LocalProxy__local', local)   #__local = request偏函数
           object.__setattr__(self, '__name__', name)
           #当前偏函数可以执行而且判断loacl中是否有 __release_local__  ==>这句话成立
           if callable(local) and not hasattr(local, '__release_local__'):
               # "local" is a callable that is not an instance of Local or
               # LocalManager: mark it as a wrapped function.
               object.__setattr__(self, '__wrapped__', local)  #__warpped__还是local偏函数

4. 当前的request是一个LocalProxy() request.method 执行LocalProxy中的__getattr__方法

       def __getattr__(self, name): # name是method(举例)
           if name == '__members__':
               return dir(self._get_current_object())
           #此时self._get_current_object()是经过_local 执行后得到的request对象,从request对象中去取出method
           return getattr(self._get_current_object(), name)

   ...

   def _get_current_object(self):
           #self._local是偏函数
           if not hasattr(self.__local, '__release_local__'):
               #执行偏函数,返回request对象
               return self.__local()
           try:
               return getattr(self.__local, self.__name__)
           except AttributeError:
               raise RuntimeError('no object bound to %s' % self.__name__)

3.3.3小结

由此看来,falsk上下文管理可以分为三个阶段：

1. 请求上文 ->
   当请求进来,第一件事就是要把当前这个请求在服务器上的线程开辟一个空间(线程对应的空间,必须含有stack对应一个列表存放ctx(request/session),具体-->:将request，session封装在 RequestContext类中
   app，g封装在AppContext类中,并通过LocalStack将requestcontext和appcontext放入Local类中
   在local类中,以线程ID号作为key的字典,
2. 请求下文：
   通过localproxy--->偏函数--->localstack--->local取值
3. '请求响应时'：-->要将上下文管理中的数据清除
   先执行save.session()再各自执行pop(),将local中的数据清除

详细看源码

3.4 应用上下文

• 执行wsgi_app方法

     #ctx为一个RequestContext的对象,参数为environ
          ctx = self.request_context(environ)
          error = None
          try:
              try:
                  ctx.push()
                  response = self.full_dispatch_request()
              except Exception as e:
                  error = e
                  response = self.handle_exception(e)

• 执行push方法,_app_ctx_stack 同样是LocalStck对象,初始化时候top为None

      def push(self):
          app_ctx = _app_ctx_stack.top
          #app_ctx = None
          if app_ctx is None or app_ctx.app != self.app:
              app_ctx = self.app.app_context()   #app_context是AppContext对象  与RequestContenx一样,知识序列化出app和g
              app_ctx.push()
              # 应用上文时候{"__storage__":{8080:{stack:rv=[app_ctx->(app/g)]}},"__ident_func__":get_ident}
              self._implicit_app_ctx_stack.append(app_ctx)
          else:
              self._implicit_app_ctx_stack.append(None)

  • 执行app_ctx.push 进而 **_app_ctx_stack.push**

     def push(self):
          """Binds the app context to the current context."""
          self._refcnt += 1
          if hasattr(sys, 'exc_clear'):
              sys.exc_clear()
              #将AppContext存在LocalStack对象中
          _app_ctx_stack.push(self)
          appcontext_pushed.send(self.app)

      def push(self, obj):
          #此时的self是LocalStack对象, obj为ctx
          """Pushes a new item to the stack"""
          # self._local = {"__storage__":{},"__ident_func__":get_ident}
          #找不到返回值是None
          rv = getattr(self._local, 'stack', None)
          if rv is None:
              #由于.stack后面有等号,执行的时候Local()对象的__setattr__方法
              #实际上是地址的赋值,此时stack和rv都指向改空列表
              self._local.stack = rv = []
              #{"__storage__":{8080:{stack:rv=[]}},"__ident_func__":get_ident}
          rv.append(obj)
          # {"__storage__":{8080:{stack:rv=[ctx->request/session]}},"__ident_func__":get_ident}
          # 应用上下文时候{"__storage__":{8080:{stack:rv=[app_ctx->(app/g)]}},"__ident_func__":get_ident}
          return rv
          #rv=[ctx->request/session]

  到此,push完毕,应用上文结束,应用下文在离线脚本时候使用,另外:在global.py中

def _find_app():
    top = _app_ctx_stack.top    #得到app_ctx(app / g)
    if top is None:
        raise RuntimeError(_app_ctx_err_msg)
    return top.app  #返回一个app即flask对象 只不过此时的flask对象 是公共的,与初始化的相同
    # 但是是独立出来已经被配置好的Flask对象

# LocalStack是 针对当前这个线程对独立的Flask_app进行修改, 不影响现在运行的app  =>离线脚本
#但是这个app 在请求结束后会从LocalStack中通过 __delattr__ 删除


# context locals
_request_ctx_stack = LocalStack()  #LocalStark = self._loacl = {"__storage__":{},"__ident_func__":get_ident}
_app_ctx_stack = LocalStack()
current_app = LocalProxy(_find_app)   # current_app可以点 .run |  .route 等