open-telemetry
/
opentelemetry-python-contrib
mirror of https://github.com/open-telemetry/opentelemetry-python-contrib.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Improve asgi instrumentation example (#3576) 

* Improve asgi instrumentation example

* Fix indentation of asgi instrumentation example

* Improve type hints of asgi instrumentation example

* Ignore pylint too-many-lines

---------

Co-authored-by: Riccardo Magliocchetti <riccardo.magliocchetti@gmail.com>
This commit is contained in:
Andre Murbach Maidl 2025-06-30 06:03:50 -03:00 committed by GitHub
parent ca079cbc56
commit 6977da3893
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 24 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
instrumentation/opentelemetry-instrumentation-asgi/src/opentelemetry/instrumentation/asgi/__init__.py
View File

@ -11,7 +11,7 @@
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
# pylint: disable=too-many-locals # pylint: disable=too-many-locals,too-many-lines
""" """
The opentelemetry-instrumentation-asgi package provides an ASGI middleware that can be used The opentelemetry-instrumentation-asgi package provides an ASGI middleware that can be used
@ -81,19 +81,38 @@ For example,
.. code-block:: python .. code-block:: python
def server_request_hook(span: Span, scope: dict[str, Any]): from opentelemetry.trace import Span
from typing import Any
from asgiref.typing import Scope, ASGIReceiveEvent, ASGISendEvent
from opentelemetry.instrumentation.asgi import OpenTelemetryMiddleware
async def application(scope: Scope, receive: ASGIReceiveEvent, send: ASGISendEvent):
await send({
'type': 'http.response.start',
'status': 200,
'headers': [
[b'content-type', b'text/plain'],
],
})
await send({
'type': 'http.response.body',
'body': b'Hello, world!',
})
def server_request_hook(span: Span, scope: Scope):
if span and span.is_recording(): if span and span.is_recording():
span.set_attribute("custom_user_attribute_from_request_hook", "some-value") span.set_attribute("custom_user_attribute_from_request_hook", "some-value")
def client_request_hook(span: Span, scope: dict[str, Any], message: dict[str, Any]): def client_request_hook(span: Span, scope: Scope, message: dict[str, Any]):
if span and span.is_recording(): if span and span.is_recording():
span.set_attribute("custom_user_attribute_from_client_request_hook", "some-value") span.set_attribute("custom_user_attribute_from_client_request_hook", "some-value")
def client_response_hook(span: Span, scope: dict[str, Any], message: dict[str, Any]): def client_response_hook(span: Span, scope: Scope, message: dict[str, Any]):
if span and span.is_recording(): if span and span.is_recording():
span.set_attribute("custom_user_attribute_from_response_hook", "some-value") span.set_attribute("custom_user_attribute_from_response_hook", "some-value")
OpenTelemetryMiddleware().(application, server_request_hook=server_request_hook, client_request_hook=client_request_hook, client_response_hook=client_response_hook) OpenTelemetryMiddleware(application, server_request_hook=server_request_hook, client_request_hook=client_request_hook, client_response_hook=client_response_hook)
Capture HTTP request and response headers Capture HTTP request and response headers
***************************************** *****************************************