如何在Flask中从RequestIDLogFilter获取请求ID？

你已经通过RequestID中间件和RequestIDLogFilter实现了日志输出请求UUID，说明这个UUID已经存在于当前请求的上下文环境中，只需要在路由里把它取出来加到响应头就行。

具体操作步骤 #

1. 导入必要的Flask对象 #

在routes.py顶部补充导入：

from flask import request, make_response, jsonify

2. 修改路由函数获取UUID并设置响应头 #

更新check函数，取出请求UUID并添加到X-REQUEST-UUID响应头：

def check():
    logging.info("You hit /")
    # 从请求上下文获取生成的UUID（这里的属性名要和你的RequestID中间件存储的一致，一般是request_id）
    request_id = request.request_id
    # 构建响应并设置头信息
    resp = make_response(jsonify(dict(ok=True)), 200)
    resp.headers['X-REQUEST-UUID'] = request_id
    return resp

3. 如果request.request_id取不到值的处理 #

要是上面的代码拿不到UUID，大概率是你的RequestID中间件把UUID存在了g对象（Flask的全局请求上下文容器）里，这时候改成：

from flask import g

# 在check函数里替换成
request_id = g.request_id

验证你的RequestID中间件实现（可选） #

如果是自定义的RequestID中间件，得确保它在请求开始时把UUID绑定到了请求上下文，比如这样：

class RequestID:
    def __init__(self, app=None):
        if app:
            self.init_app(app)
    
    def init_app(self, app):
        app.before_request(self.before_request)
    
    def before_request(self):
        import uuid
        request_id = str(uuid.uuid4())
        # 把UUID绑定到request对象，方便路由和日志过滤器调用
        request.request_id = request_id

对应的RequestIDLogFilter也要从request对象里读取UUID：

class RequestIDLogFilter(logging.Filter):
    def filter(self, record):
        from flask import request
        record.request_id = getattr(request, 'request_id', 'unknown')
        return True

这样改完之后，响应头里就会带上和日志里一致的请求UUID了。

内容的提问来源于stack exchange，提问作者Saturnian