Python：获取 urllib3 请求标头

如何解决Python：获取 urllib3 请求标头

我们正在将跟踪信息注入到基于 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“请求”库中，我们有“response.request.headers”属性来返回请求头

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

大家可以分享一些想法吗

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

解决方法

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