Python：获取 urllib3 请求标头

Question

我们将跟踪信息注入到基于 urllib3 实现的 API 客户端库中所有 http 请求调用的请求标头中

def _init_jaeger_tracer():
    '''Jaeger tracer initialization'''
    config = Config(
        config={
            'sampler': {
                'type': 'const',
                'param': 1,
            },
        },
        service_name="session"
        )
    return config.new_tracer()

class APIObject(rest.APIObject):
    '''Class for injecting traces into urllib3 request headers'''
    def __init__(self, configuration):
        print("RESTClientObject child class called####")
        self._tracer = None
        super().__init__(configuration)
        self._tracer = _init_jaeger_tracer()

    # pylint: disable=W0221
    def request(self, method, url, *args, **kwargs):
        lower_method = method.lower()
        with self._tracer.start_active_span('requests.{}'.format(lower_method)) as scope:
            span = scope.span
            span.set_tag(tags.SPAN_KIND, tags.SPAN_KIND_RPC_CLIENT)
            span.set_tag(tags.COMPONENT, 'request')
            span.set_tag(tags.HTTP_METHOD, lower_method)
            span.set_tag(tags.HTTP_URL, url)
            headers = kwargs.setdefault('headers', {})
            self._tracer.inject(span.context, Format.HTTP_HEADERS, headers)
    return r

经过这样的实现，urllib3请求头看起来像这样，

headers = {
    'Content-Type': 'application/json', 
    'User-Agent': 'API-Generator/1.0.0/python', 
    # **
    'uber-trace-id': '30cef3e816482516:1a4fed2c4863b2f6:0:1'
    # **
}

现在我们必须从注入的请求标头中获取trace-id。

在python“requests”库中，我们有“response.request.headers”属性来返回请求标头

在 urllib3 中，我找不到返回请求标头以进行进一步处理的方法

你们中有人可以分享一些想法吗

现在，从这个请求标头中，我们需要获取“uber-trace-id”及其值以进一步构建 URL。

Answer 1

的文档urllib3.response.HTTPResponse说它是：

向后兼容http.client.HTTPResponse[...]

这是 stdlib 类，其getheader方法描述为：

返回标头的值name，或者default如果没有标头匹配则返回name。如果有多个名称为 name 的标头，则返回所有由 ', ' 连接的值。如果default是除单个字符串之外的任何可迭代对象，则其元素同样以逗号连接的方式返回。

因此，在你的情况下，情况是这样的。

import urllib3

http = urllib3.PoolManager()
response = http.request('GET', 'https://python.org/')
response.getheader('uber-trace-id', 'UNKNOWN')

更新

urllib3没有 HTTP 请求对象表示。HTTP 请求以 的形式参数表示RequestMethods.request 。request因此，在注入跟踪标识符后，您需要自己将跟踪标识符存储在覆盖中jaeger_client。将其存储为响应对象的属性可能是一个好主意：

from jaeger_client.constants import TRACE_ID_HEADER

class APIObject(rest.APIObject):

    def request(self, *args, **kwargs) -> urllib3.response.BaseHTTPResponse:
        ...
        r._request_tracing_id = headers[TRACE_ID_HEADER]
        return r