Skip to content

✨ Add lifespan handling to tracing instrumentation and instrument rabbitmq rpc clients #7642

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 33 commits into from
May 23, 2025
Merged

✨ Add lifespan handling to tracing instrumentation and instrument rabbitmq rpc clients #7642

Show file tree
Hide file tree
Changes from 8 commits
Commits
Show all changes
33 commits
Select commit Hold shift + click to select a range
dbb2722
add aio pika instrumentation dependency
bisgaard-itis May 6, 2025
2458561
upgrade dependency accross entire repo
bisgaard-itis May 6, 2025
8cf4d44
instrument aipika rabbit client
bisgaard-itis May 6, 2025
d85dcb1
add opentelemetry instrumentation celery dependency
bisgaard-itis May 6, 2025
273896e
Merge branch 'master' into 7634-opentelemetry-instrument-rpc-clients
bisgaard-itis May 6, 2025
0f685de
autoinstrument aiopika in aihttp case
bisgaard-itis May 6, 2025
c7c8469
remove asyncpg and aiopg instrumentation
bisgaard-itis May 6, 2025
8763e94
remove asyncpg instrumentation
bisgaard-itis May 6, 2025
bd45887
Revert "autoinstrument aiopika in aihttp case"
bisgaard-itis May 7, 2025
3755521
autoinstrument aiopika in aihttp case
bisgaard-itis May 6, 2025
b37c053
Revert "remove asyncpg instrumentation"
bisgaard-itis May 7, 2025
02523a7
Revert "remove asyncpg and aiopg instrumentation"
bisgaard-itis May 7, 2025
2601b2a
Merge branch 'master' into 7634-opentelemetry-instrument-rpc-clients
bisgaard-itis May 7, 2025
08397b0
Merge branch 'master' into 7634-opentelemetry-instrument-rpc-clients
bisgaard-itis May 12, 2025
d4c643a
Merge branch 'master' into 7634-opentelemetry-instrument-rpc-clients
bisgaard-itis May 12, 2025
d762da5
attempt to solve issue by fixing otel fixture
bisgaard-itis May 12, 2025
72fa339
ensure to delete env
bisgaard-itis May 12, 2025
6036749
Merge branch 'master' into 7634-opentelemetry-instrument-rpc-clients
bisgaard-itis May 19, 2025
660d00c
Merge branch 'master' into 7634-opentelemetry-instrument-rpc-clients
bisgaard-itis May 23, 2025
91a0f99
upgrade aiopika instrumentation lib after merging master
bisgaard-itis May 23, 2025
efa751c
add lifespan to tracing in fastapi case
bisgaard-itis May 23, 2025
2c6afb9
Merge branch 'master' into 7634-opentelemetry-instrument-rpc-clients
bisgaard-itis May 23, 2025
3840e07
use lifespan manager approach
bisgaard-itis May 23, 2025
90c14ea
disable pylint bare expose
bisgaard-itis May 23, 2025
c5e4c1b
add getter for tracing instrumentation lifespan
bisgaard-itis May 23, 2025
dabc522
fix pylint
bisgaard-itis May 23, 2025
10689e9
Merge branch 'master' into 7634-opentelemetry-instrument-rpc-clients
bisgaard-itis May 23, 2025
e7ae167
remove call to private method
bisgaard-itis May 23, 2025
9b7bd1a
add lifespan management to tracing instrumentation in aiohttp
bisgaard-itis May 23, 2025
bd81753
make pylint happy
bisgaard-itis May 23, 2025
6466494
assert app
bisgaard-itis May 23, 2025
8458404
ensure lifespan manager works for fastapi
bisgaard-itis May 23, 2025
f545694
Merge branch 'master' into 7634-opentelemetry-instrument-rpc-clients
bisgaard-itis May 23, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 5 additions & 0 deletions packages/aws-library/requirements/_base.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -145,6 +145,7 @@ opentelemetry-api==1.30.0
# opentelemetry-exporter-otlp-proto-grpc
# opentelemetry-exporter-otlp-proto-http
# opentelemetry-instrumentation
# opentelemetry-instrumentation-aio-pika
# opentelemetry-instrumentation-botocore
# opentelemetry-instrumentation-logging
# opentelemetry-instrumentation-redis
Expand All @@ -164,10 +165,13 @@ opentelemetry-exporter-otlp-proto-http==1.30.0
# via opentelemetry-exporter-otlp
opentelemetry-instrumentation==0.51b0
# via
# opentelemetry-instrumentation-aio-pika
# opentelemetry-instrumentation-botocore
# opentelemetry-instrumentation-logging
# opentelemetry-instrumentation-redis
# opentelemetry-instrumentation-requests
opentelemetry-instrumentation-aio-pika==0.51b0
# via -r requirements/../../../packages/service-library/requirements/_base.in
opentelemetry-instrumentation-botocore==0.51b0
# via -r requirements/_base.in
opentelemetry-instrumentation-logging==0.51b0
Expand Down Expand Up @@ -431,6 +435,7 @@ wrapt==1.17.2
# aiobotocore
# deprecated
# opentelemetry-instrumentation
# opentelemetry-instrumentation-aio-pika
# opentelemetry-instrumentation-redis
yarl==1.18.3
# via
Expand Down
27 changes: 0 additions & 27 deletions packages/notifications-library/requirements/_base.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -16,10 +16,6 @@ attrs==25.1.0
# referencing
click==8.1.8
# via typer
deprecated==1.2.18
# via
# opentelemetry-api
# opentelemetry-semantic-conventions
dnspython==2.7.0
# via email-validator
email-validator==2.2.0
Expand All @@ -30,8 +26,6 @@ idna==3.10
# via
# email-validator
# yarl
importlib-metadata==8.5.0
# via opentelemetry-api
jinja2==3.1.5
# via
# -c requirements/../../../packages/common-library/requirements/../../../requirements/constraints.txt
Expand Down Expand Up @@ -68,19 +62,6 @@ mdurl==0.1.2
# via markdown-it-py
multidict==6.1.0
# via yarl
opentelemetry-api==1.30.0
# via
# opentelemetry-instrumentation
# opentelemetry-instrumentation-asyncpg
# opentelemetry-semantic-conventions
opentelemetry-instrumentation==0.51b0
# via opentelemetry-instrumentation-asyncpg
opentelemetry-instrumentation-asyncpg==0.51b0
# via -r requirements/../../../packages/postgres-database/requirements/_base.in
opentelemetry-semantic-conventions==0.51b0
# via
# opentelemetry-instrumentation
# opentelemetry-instrumentation-asyncpg
orjson==3.10.15
# via
# -c requirements/../../../packages/common-library/requirements/../../../requirements/constraints.txt
Expand All @@ -96,8 +77,6 @@ orjson==3.10.15
# -r requirements/../../../packages/models-library/requirements/_base.in
# -r requirements/../../../packages/postgres-database/requirements/../../../packages/common-library/requirements/_base.in
# -r requirements/../../../packages/settings-library/requirements/../../../packages/common-library/requirements/_base.in
packaging==24.2
# via opentelemetry-instrumentation
propcache==0.3.0
# via yarl
psycopg2-binary==2.9.10
Expand Down Expand Up @@ -195,11 +174,5 @@ typing-extensions==4.12.2
# pydantic-core
# pydantic-extra-types
# typer
wrapt==1.17.2
# via
# deprecated
# opentelemetry-instrumentation
yarl==1.18.3
# via -r requirements/../../../packages/postgres-database/requirements/_base.in
zipp==3.21.0
# via importlib-metadata
1 change: 0 additions & 1 deletion packages/notifications-library/requirements/_test.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,6 @@ mypy-extensions==1.0.0
# via mypy
packaging==24.2
# via
# -c requirements/_base.txt
# pytest
# pytest-sugar
pluggy==1.5.0
Expand Down
1 change: 0 additions & 1 deletion packages/notifications-library/requirements/_tools.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,6 @@ nodeenv==1.9.1
# via pre-commit
packaging==24.2
# via
# -c requirements/_base.txt
# -c requirements/_test.txt
# black
# build
Expand Down
1 change: 0 additions & 1 deletion packages/postgres-database/requirements/_base.in
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,6 @@
--requirement ../../../packages/common-library/requirements/_base.in

alembic
opentelemetry-instrumentation-asyncpg
pydantic
sqlalchemy[postgresql_psycopg2binary,postgresql_asyncpg] # SEE extras in https://github.com/sqlalchemy/sqlalchemy/blob/main/setup.cfg#L43
yarl
27 changes: 0 additions & 27 deletions packages/postgres-database/requirements/_base.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,16 +4,10 @@ annotated-types==0.7.0
# via pydantic
asyncpg==0.30.0
# via sqlalchemy
deprecated==1.2.18
# via
# opentelemetry-api
# opentelemetry-semantic-conventions
greenlet==3.1.1
# via sqlalchemy
idna==3.10
# via yarl
importlib-metadata==8.5.0
# via opentelemetry-api
mako==1.3.9
# via
# -c requirements/../../../packages/common-library/requirements/../../../requirements/constraints.txt
Expand All @@ -23,26 +17,11 @@ markupsafe==3.0.2
# via mako
multidict==6.1.0
# via yarl
opentelemetry-api==1.30.0
# via
# opentelemetry-instrumentation
# opentelemetry-instrumentation-asyncpg
# opentelemetry-semantic-conventions
opentelemetry-instrumentation==0.51b0
# via opentelemetry-instrumentation-asyncpg
opentelemetry-instrumentation-asyncpg==0.51b0
# via -r requirements/_base.in
opentelemetry-semantic-conventions==0.51b0
# via
# opentelemetry-instrumentation
# opentelemetry-instrumentation-asyncpg
orjson==3.10.15
# via
# -c requirements/../../../packages/common-library/requirements/../../../requirements/constraints.txt
# -c requirements/../../../requirements/constraints.txt
# -r requirements/../../../packages/common-library/requirements/_base.in
packaging==24.2
# via opentelemetry-instrumentation
propcache==0.3.0
# via yarl
psycopg2-binary==2.9.10
Expand Down Expand Up @@ -70,11 +49,5 @@ typing-extensions==4.12.2
# pydantic
# pydantic-core
# pydantic-extra-types
wrapt==1.17.2
# via
# deprecated
# opentelemetry-instrumentation
yarl==1.18.3
# via -r requirements/_base.in
zipp==3.21.0
# via importlib-metadata
4 changes: 1 addition & 3 deletions packages/postgres-database/requirements/_test.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -24,9 +24,7 @@ mypy==1.15.0
mypy-extensions==1.0.0
# via mypy
packaging==24.2
# via
# -c requirements/_base.txt
# pytest
# via pytest
pluggy==1.5.0
# via pytest
psycopg2-binary==2.9.10
Expand Down
1 change: 0 additions & 1 deletion packages/postgres-database/requirements/_tools.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,6 @@ nodeenv==1.9.1
# via pre-commit
packaging==24.2
# via
# -c requirements/_base.txt
# -c requirements/_test.txt
# black
# build
Expand Down
1 change: 0 additions & 1 deletion packages/service-library/requirements/_aiohttp.in
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,5 @@ attrs
jsonschema
opentelemetry-instrumentation-aiohttp-client
opentelemetry-instrumentation-aiohttp-server
opentelemetry-instrumentation-aiopg
prometheus_client
werkzeug
11 changes: 0 additions & 11 deletions packages/service-library/requirements/_aiohttp.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -50,29 +50,20 @@ opentelemetry-api==1.30.0
# opentelemetry-instrumentation
# opentelemetry-instrumentation-aiohttp-client
# opentelemetry-instrumentation-aiohttp-server
# opentelemetry-instrumentation-aiopg
# opentelemetry-instrumentation-dbapi
# opentelemetry-semantic-conventions
opentelemetry-instrumentation==0.51b0
# via
# opentelemetry-instrumentation-aiohttp-client
# opentelemetry-instrumentation-aiohttp-server
# opentelemetry-instrumentation-aiopg
# opentelemetry-instrumentation-dbapi
opentelemetry-instrumentation-aiohttp-client==0.51b0
# via -r requirements/_aiohttp.in
opentelemetry-instrumentation-aiohttp-server==0.51b0
# via -r requirements/_aiohttp.in
opentelemetry-instrumentation-aiopg==0.51b0
# via -r requirements/_aiohttp.in
opentelemetry-instrumentation-dbapi==0.51b0
# via opentelemetry-instrumentation-aiopg
opentelemetry-semantic-conventions==0.51b0
# via
# opentelemetry-instrumentation
# opentelemetry-instrumentation-aiohttp-client
# opentelemetry-instrumentation-aiohttp-server
# opentelemetry-instrumentation-dbapi
opentelemetry-util-http==0.51b0
# via
# opentelemetry-instrumentation-aiohttp-client
Expand Down Expand Up @@ -120,8 +111,6 @@ wrapt==1.17.2
# opentelemetry-instrumentation
# opentelemetry-instrumentation-aiohttp-client
# opentelemetry-instrumentation-aiohttp-server
# opentelemetry-instrumentation-aiopg
# opentelemetry-instrumentation-dbapi
yarl==1.18.3
# via aiohttp
zipp==3.21.0
Expand Down
1 change: 1 addition & 0 deletions packages/service-library/requirements/_base.in
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@ arrow # date/time
faststream
opentelemetry-api
opentelemetry-exporter-otlp
opentelemetry-instrumentation-aio-pika
opentelemetry-instrumentation-logging
opentelemetry-instrumentation-redis
opentelemetry-instrumentation-requests
Expand Down
5 changes: 5 additions & 0 deletions packages/service-library/requirements/_base.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -103,6 +103,7 @@ opentelemetry-api==1.30.0
# opentelemetry-exporter-otlp-proto-grpc
# opentelemetry-exporter-otlp-proto-http
# opentelemetry-instrumentation
# opentelemetry-instrumentation-aio-pika
# opentelemetry-instrumentation-logging
# opentelemetry-instrumentation-redis
# opentelemetry-instrumentation-requests
Expand All @@ -120,9 +121,12 @@ opentelemetry-exporter-otlp-proto-http==1.30.0
# via opentelemetry-exporter-otlp
opentelemetry-instrumentation==0.51b0
# via
# opentelemetry-instrumentation-aio-pika
# opentelemetry-instrumentation-logging
# opentelemetry-instrumentation-redis
# opentelemetry-instrumentation-requests
opentelemetry-instrumentation-aio-pika==0.51b0
# via -r requirements/_base.in
opentelemetry-instrumentation-logging==0.51b0
# via -r requirements/_base.in
opentelemetry-instrumentation-redis==0.51b0
Expand Down Expand Up @@ -297,6 +301,7 @@ wrapt==1.17.2
# via
# deprecated
# opentelemetry-instrumentation
# opentelemetry-instrumentation-aio-pika
# opentelemetry-instrumentation-redis
yarl==1.18.3
# via
Expand Down
37 changes: 20 additions & 17 deletions packages/service-library/src/servicelib/aiohttp/tracing.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,6 +1,4 @@
""" Adds aiohttp middleware for tracing using opentelemetry instrumentation.

"""
"""Adds aiohttp middleware for tracing using opentelemetry instrumentation."""

import logging

Expand Down Expand Up @@ -31,19 +29,20 @@
HAS_BOTOCORE = True
except ImportError:
HAS_BOTOCORE = False
try:
from opentelemetry.instrumentation.aiopg import AiopgInstrumentor

HAS_AIOPG = True
except ImportError:
HAS_AIOPG = False
try:
from opentelemetry.instrumentation.requests import RequestsInstrumentor

HAS_REQUESTS = True
except ImportError:
HAS_REQUESTS = False

try:
from opentelemetry.instrumentation.aio_pika import AioPikaInstrumentor

HAS_AIO_PIKA = True
except ImportError:
HAS_AIO_PIKA = False


def setup_tracing(
app: web.Application,
Expand Down Expand Up @@ -74,7 +73,9 @@ def setup_tracing(
trace.set_tracer_provider(TracerProvider(resource=resource))
tracer_provider: trace.TracerProvider = trace.get_tracer_provider()

tracing_destination: str = f"{URL(opentelemetry_collector_endpoint).with_port(opentelemetry_collector_port).with_path('/v1/traces')}"
tracing_destination: str = (
f"{URL(opentelemetry_collector_endpoint).with_port(opentelemetry_collector_port).with_path('/v1/traces')}"
)

_logger.info(
"Trying to connect service %s to tracing collector at %s.",
Expand Down Expand Up @@ -107,13 +108,7 @@ def setup_tracing(

# Instrument aiohttp client
AioHttpClientInstrumentor().instrument()
if HAS_AIOPG:
with log_context(
_logger,
logging.INFO,
msg="Attempting to add aio-pg opentelemetry autoinstrumentation...",
):
AiopgInstrumentor().instrument()

if HAS_BOTOCORE:
with log_context(
_logger,
Expand All @@ -128,3 +123,11 @@ def setup_tracing(
msg="Attempting to add requests opentelemetry autoinstrumentation...",
):
RequestsInstrumentor().instrument()

if HAS_AIO_PIKA:
with log_context(
_logger,
logging.INFO,
msg="Attempting to add aio_pika opentelemetry autoinstrumentation...",
):
AioPikaInstrumentor().instrument()
Loading
Loading