Flask 上下文机制和线程隔离

上下文：相当于一个容器，保存了Flask程序运行过程中的一些信息 源码：flask/ctx.py

• 请求上下文：Flask从客户端收到请求时，要让视图函数能访问一些对象，这样才能处理请求，要想让视图函数能够访问请求对象，一个显而易见的方式是将其作为参数传入视图函数，不过这会导致程序中的每个视图函数都增加一个参数，除了访问请求对象,如果视图函数在处理请求时还要访问其他对象，情况会变得更糟。为了避免大量可有可无的参数把视图函数弄得一团糟，Flask使用上下文临时把某些对象变为全局可访问。这就是一种重构设计思路。

  • request 封装了HTTP请求的内容，针对http请求，也是一种符合WSGI接口规范的设计（关于WSGI可参考我对该协议的理解和实现demo mini-wsgi-web），如 request.args.get('user')
  • session 用来记录请求会话中的信息，针对的是用户信息，如 session['name'] = user.id

• 应用上下文：应用程序上下文,用于存储应用程序中的变量

  • current_app 存储应用配置，数据库连接等应用相关信息
  • g变量 作为flask程序全局的一个临时变量, 充当者中间媒介的作用,我们可以通过它传递一些数据，g保存的是当前请求的全局变量，不同的请求会有不同的全局变量，通过不同的thread id区别

# context locals

# 使用代理模式 LocalProxy

_request_ctx_stack = LocalStack()

_app_ctx_stack = LocalStack()

current_app = LocalProxy(_find_app)

request = LocalProxy(partial(_lookup_req_object, 'request'))

session = LocalProxy(partial(_lookup_req_object, 'session'))

g = LocalProxy(partial(_lookup_app_object, 'g'))

1. working outside application context

#!/usr/bin/python3

# -*- coding: utf-8 -*-

# __author__ = '__JonPan__'

from flask import Flask, current_app

app = Flask(__name__)

a = current_app

# is_debug = current_app.config['DEBUG']

@app.route('/')

def index():

    return '<h1>Hello World. Have a nice day! </1>'

if __name__ == '__main__':

    app.run(host='localhost', port=8888)

2. flask 上下文出入栈

flask上下文对象出入栈模型图

在应用开发中可用直接引用current_app不会报错，是因为当在一个请求中使用的时候，flask会判断_app_ctx_stack栈顶是否有可用对象，如果没有就会自动推入一个App. 我们获取的current_app就是获取的栈顶元素

# flask/globals.py

def _find_app():

    top = _app_ctx_stack.top

    if top is None:

        raise RuntimeError(_app_ctx_err_msg)

    return top.app

current_app = LocalProxy(_find_app)

修改代码：将app对象推入栈顶

# ...

# 将app_context 推入栈中

ctx = app.app_context()

ctx.push()

a = current_app

is_debug = current_app.config['DEBUG']

ctx.pop()

# ...

# 更有pythonnic的写法

# 将app_context 推入栈中

# with app.app_context():

#     a = current_app

#     is_debug = current_app.config['DEBUG']

不在出现unbound状态。可正常运行

既然flask会自动帮我们检测栈顶元素是否存在，为什么我们还要做这一步操作，当我们在写离线应用，或者单元测试的时候就需要用到，因为请求是模拟的，不是在application context中的了。

3. python中的上文管理器

实现了__enter__和__exit__方法的对象就是一个上文管理器。

class MyResource:

    def __enter__(self):

        print('connect ro resource')

        return self

    def __exit__(self, exc_type, exc_value, tb):

        if tb:

            print('process exception')

        else:

            print('no exception')

        print('close resource connection')

        # return True

        # return False

    def query(self):

        print('query data')

try:

    with MyResource() as r:

        1/0

        r.query()

except Exception as e:

    print(e)

with MyResour ce() as r as 的别名r指向的不是上想问管理器对象，而是__enter__方法返回的值，在以上代码确实是返回了对象本身。

__exit__ 方法 处理退出上下文管理器对象时的一些资源清理工作，并处理异常，三个参数

• exc_type 异常类型
• exc_value 异常原因解释
• tb traceback

__exit__其实是有返回值的，return True表示，异常信息已经在本方法中处理，外部可不接收异常，return False 表示将异常抛出给上层逻辑处理，默认不写返回，即默认值是None, None也表示False

线程隔离机制

如何实现一个Reqeust 指向多个请求实例，且要区分该实例对象所绑定的用户？

字典:

flask引用 werkzeug 中的 local.Local 实现线程隔离

# werkzeug\local.py

class Local(object):

    __slots__ = ('__storage__', '__ident_func__')

    def __init__(self):

        object.__setattr__(self, '__storage__', {})

        object.__setattr__(self, '__ident_func__', get_ident)

    def __iter__(self):

        return iter(self.__storage__.items())

    def __call__(self, proxy):

        """Create a proxy for a name."""

        return LocalProxy(self, proxy)

    def __release_local__(self):

        self.__storage__.pop(self.__ident_func__(), None)

    def __getattr__(self, name):

        try:

            return self.__storage__[self.__ident_func__()][name]

        except KeyError:

            raise AttributeError(name)

    def __setattr__(self, name, value):

        # 获取当前线程的id

        ident = self.__ident_func__()

        storage = self.__storage__

        try:

            storage[ident][name] = value

        except KeyError:

            storage[ident] = {name: value}

    def __delattr__(self, name):

        try:

            del self.__storage__[self.__ident_func__()][name]

        except KeyError:

            raise AttributeError(name)

可以看到是使用线程ID来绑定不同的上线文对象。

LocalStack

class LocalStack(object):

    """This class works similar to a :class:`Local` but keeps a stack

    of objects instead.  This is best explained with an example::

        >>> ls = LocalStack()

        >>> ls.push(42)

        >>> ls.top

        42

        >>> ls.push(23)

        >>> ls.top

        23

        >>> ls.pop()

        23

        >>> ls.top

        42

    They can be force released by using a :class:`LocalManager` or with

    the :func:`release_local` function but the correct way is to pop the

    item from the stack after using.  When the stack is empty it will

    no longer be bound to the current context (and as such released).

    By calling the stack without arguments it returns a proxy that resolves to

    the topmost item on the stack.

    .. versionadded:: 0.6.1

    """

    def __init__(self):

        self._local = Local()

    def __release_local__(self):

        self._local.__release_local__()

    def _get__ident_func__(self):

        return self._local.__ident_func__

    def _set__ident_func__(self, value):

        object.__setattr__(self._local, '__ident_func__', value)

    __ident_func__ = property(_get__ident_func__, _set__ident_func__)

    del _get__ident_func__, _set__ident_func__

    def __call__(self):

        def _lookup():

            rv = self.top

            if rv is None:

                raise RuntimeError('object unbound')

            return rv

        return LocalProxy(_lookup)

    def push(self, obj):

        """Pushes a new item to the stack"""

        rv = getattr(self._local, 'stack', None)

        if rv is None:

            self._local.stack = rv = []

        rv.append(obj)

        return rv

    def pop(self):

        """Removes the topmost item from the stack, will return the

        old value or `None` if the stack was already empty.

        """

        stack = getattr(self._local, 'stack', None)

        if stack is None:

            return None

        elif len(stack) == 1:

            release_local(self._local)

            return stack[-1]

        else:

            return stack.pop()

    @property

    def top(self):

        """The topmost item on the stack.  If the stack is empty,

        `None` is returned.

        """

        try:

            return self._local.stack[-1]

        except (AttributeError, IndexError):

            return None

总结

1. flask上下文的实现使用了设计模式中的代理模式
2. 线程隔离的实现机制就是利用线程ID+字典， 使用线程隔离的意义在于：使当前线程能够正确引用到他自己所创建的对象，而不是引用到其他线程所创建的对象
3. 请求上下文, 应用上线文是线程隔离的对象，current_app并没有使用线程隔离，全局唯一
4. current_app -> (LocalStack.top = AppContext top.app = Flask)
5. request -> (LocalStack.top = RequestContext.top.request = Request)

参考资料

Python Flask高级编程-七月