Werkzeug Local 异常:并发请求下数据串流问题排查求助
关于Werkzeug Local并发下数据串值的问题分析与解决
你的问题核心在于Werkzeug Local不会自动清理线程残留数据。Flask的开发服务器默认是多线程模式,生产环境用Gunicorn等部署也会采用多线程/进程架构,线程处理完请求后会被放回线程池复用。你存在Local对象里的data属性不会被自动删除,下一个请求复用该线程时,就会读取到上一个请求留下的旧数据,导致无token的请求也返回了之前的token。
解决方案
方案一:使用Flask内置的g对象(推荐)
Flask的g对象基于Werkzeug Local实现,并且会在每个请求结束后自动清空所有属性,完全适配请求级别的临时数据存储需求:
from flask import Flask, request, g app = Flask(__name__) @app.route('/') def index(): if hasattr(g, 'data') and g.data: return g.data g.data = request.headers.get('token') return g.data if __name__ == "__main__": app.run()
方案二:手动清理Local数据
如果一定要用Werkzeug Local,需通过请求钩子在请求结束后手动清理数据:
from flask import Flask, request from werkzeug.local import Local app = Flask(__name__) local_data = Local() @app.route('/') def index(): if hasattr(local_data, 'data') and local_data.data: return local_data.data local_data.data = request.headers.get('token') return local_data.data @app.after_request def clear_local_data(response): if hasattr(local_data, 'data'): del local_data.data return response if __name__ == "__main__": app.run()
补充说明
注意不要直接用if local_data.data:判断,因为属性不存在时会抛出AttributeError,使用hasattr先判断属性是否存在更安全。
内容的提问来源于stack exchange,提问作者william
相关产品推荐
相关产品推荐

