Skip to content

Commit 10ee682

Browse files
author
Andrei Neagu
committed
pylint service-library
1 parent 2cb214c commit 10ee682

File tree

1 file changed

+22
-9
lines changed
  • packages/service-library/src/servicelib/aiohttp

1 file changed

+22
-9
lines changed

packages/service-library/src/servicelib/aiohttp/tracing.py

Lines changed: 22 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,53 +1,66 @@
11
""" Adds aiohttp middleware for tracing using opentelemetry instrumentation.
22
33
"""
4+
45
import logging
56

67
from aiohttp import web
78
from opentelemetry import trace
89
from opentelemetry.exporter.otlp.proto.http.trace_exporter import (
910
OTLPSpanExporter as OTLPSpanExporterHTTP,
1011
)
11-
from opentelemetry.instrumentation.aiohttp_client import AioHttpClientInstrumentor
12-
from opentelemetry.instrumentation.aiohttp_server import AioHttpServerInstrumentor
13-
from opentelemetry.instrumentation.aiopg import AiopgInstrumentor
12+
from opentelemetry.instrumentation.aiohttp_client import ( # pylint:disable=no-name-in-module
13+
AioHttpClientInstrumentor,
14+
)
15+
from opentelemetry.instrumentation.aiohttp_server import ( # pylint:disable=no-name-in-module
16+
AioHttpServerInstrumentor,
17+
)
18+
from opentelemetry.instrumentation.aiopg import ( # pylint:disable=no-name-in-module
19+
AiopgInstrumentor,
20+
)
1421
from opentelemetry.instrumentation.requests import RequestsInstrumentor
1522
from opentelemetry.sdk.resources import Resource
1623
from opentelemetry.sdk.trace import TracerProvider
1724
from opentelemetry.sdk.trace.export import BatchSpanProcessor
1825
from settings_library.tracing import TracingSettings
1926

20-
log = logging.getLogger(__name__)
27+
_logger = logging.getLogger(__name__)
2128

2229

2330
def setup_tracing(
24-
app: web.Application, # pylint: disable=unused-argument
31+
app: web.Application,
2532
tracing_settings: TracingSettings,
2633
service_name: str,
2734
instrument_aiopg: bool = False, # noqa: FBT001, FBT002
2835
) -> None:
2936
"""
3037
Sets up this service for a distributed tracing system (opentelemetry)
3138
"""
39+
_ = app
3240
opentelemetry_collector_endpoint = (
3341
tracing_settings.TRACING_OPENTELEMETRY_COLLECTOR_ENDPOINT
3442
)
3543
opentelemetry_collector_port = tracing_settings.TRACING_OPENTELEMETRY_COLLECTOR_PORT
3644
if not opentelemetry_collector_endpoint and not opentelemetry_collector_port:
37-
log.warning("Skipping opentelemetry tracing setup")
45+
_logger.warning("Skipping opentelemetry tracing setup")
3846
return
3947
if not opentelemetry_collector_endpoint or not opentelemetry_collector_port:
40-
raise RuntimeError(
41-
f"Variable opentelemetry_collector_endpoint [{tracing_settings.TRACING_OPENTELEMETRY_COLLECTOR_ENDPOINT}] or opentelemetry_collector_port [{tracing_settings.TRACING_OPENTELEMETRY_COLLECTOR_PORT}] unset. Tracing options incomplete."
48+
msg = (
49+
"Variable opentelemetry_collector_endpoint "
50+
f"[{tracing_settings.TRACING_OPENTELEMETRY_COLLECTOR_ENDPOINT}] "
51+
"or opentelemetry_collector_port "
52+
f"[{tracing_settings.TRACING_OPENTELEMETRY_COLLECTOR_PORT}] "
53+
"unset. Tracing options incomplete."
4254
)
55+
raise RuntimeError(msg)
4356
resource = Resource(attributes={"service.name": service_name})
4457
trace.set_tracer_provider(TracerProvider(resource=resource))
4558
tracer_provider: trace.TracerProvider = trace.get_tracer_provider()
4659
tracing_destination: str = (
4760
f"{opentelemetry_collector_endpoint}:{opentelemetry_collector_port}/v1/traces"
4861
)
4962

50-
log.info(
63+
_logger.info(
5164
"Trying to connect service %s to tracing collector at %s.",
5265
service_name,
5366
tracing_destination,

0 commit comments

Comments
 (0)