Skip to main content

a wrapper around `opentelemetry` and `opentelemetry-instrumentation-*` to make life a bit easier

Project description

opentelemetry_wrapper

a wrapper around opentelemetry and opentelemetry-instrumentation-* to make life a bit easier

what this does (or is supposed to do)

  • Make instrumentation (more) idempotent
  • Make re-instrumentation of logging actually work with different format strings
  • Make logging print as a one-line JSON dict by default
  • Provide support for decorating functions and classes
  • Provide support for instrumentation of dataclasses
    • Global instrumentation needs to be run before any dataclasses are initialized
    • Otherwise, use the decorator on each class as usual (it's idempotent anyway)
  • Add global instrumentation of FastAPI
    • sometimes works even after apps are created for some reason, likely due to how Uvicorn runs in a new process
    • but somehow sometimes doesn't work in prod, for equally unknown reasons
    • probably best to instrument each app instance
  • Logs some http headers received by fastapi as span attributes
  • Creates OTLP exporter if specific env vars (below) are set

usage

TL;DR:

  1. call instrument_all() to instrument logging and requests
  2. instrument your FastAPI app using instrument_fastapi_app(...)
  3. use @instrument_decorate on any function or class you want to monitor
  4. set OTEL_WRAPPER_DISABLED=true in your CICD tests, especially if you're using pytest

env vars

Variable Name Description Default (if not set)
OTEL_EXPORTER_OTLP_ENDPOINT Looks like http://tempo.localhost:4317. NA (traces are not exported to any OTLP endpoint)
OTEL_EXPORTER_OTLP_HEADER Looks like Header-Name=header value, where values can contain space ('\x20'). To insert multiple headers, delimit by any other whitespace char. NA (no header sent to OTLP endpoint)
OTEL_EXPORTER_OTLP_INSECURE Set to true to disable SSL for OTLP trace exports, or false to always verify. NA (follows OpenTelemetry default, which is secure for https and insecure for http)
OTEL_HEADER_ATTRIBUTES HTTP headers to extract as span attributes, split by whitespace. x-pf-number, x-client-id, x-preferred-username, x-resource-access
OTEL_LOG_LEVEL Log level used by the logging instrumentor (case-insensitive). info
OTEL_SERVICE_NAME Sets the value of the service.name resource attribute. f'{k8s namespace}/{k8s pod name}' or f'{username}@{hostname}.{domain}:<{filename of main}>'
OTEL_SERVICE_NAMESPACE Sets the value of the service.namespace resource attribute. f'{k8s namespace}' or None
OTEL_WRAPPER_DISABLED Set to true to disable tracing globally (e.g. when running pytest). false (tracing is enabled)

Note:
The service.name and service.namespace can also be set via OTEL_RESOURCE_ATTRIBUTES, but any settings there will be overwritten by OTEL_SERVICE_NAME and OTEL_SERVICE_NAMESPACE. For more details, read the original OpenTelemetry docs

@instrument_decorate decorator for functions and classes

  • decorating a function creates a span whenever the function is called
    • the span name is set to the function or class name, and attributes are added for the filename, line number, etc
  • also works for async functions
  • also works for classes and dataclasses
    • creates a span for every method call, including init (or post_init), new, and call
    • creates a span for every property get, set, or delete
    • if this is too verbose, feel free to decorate specific methods in the class instead
  • it is not recommended to use this on functions that are called thousands of times or classes you create thousands of instances for (e.g. pydantic classes, usually), since it can be excessively noisy
from opentelemetry_wrapper import instrument_decorate


@instrument_decorate
def square(x):
    return x * x


@instrument_decorate
async def cube(x):
    return x * x * x


@instrument_decorate
class Thing:
    def __init__(self):
        self.x = 1

instrumenting the builtin logging module

  • sets a root logger handler (or more than one) that can output logs to the console or to a file path
  • logs are output as json by default to make them easier to work with downstream, set print_json=False to disable
    • the verbose parameter has no effect on json output, which is maximally verbose
    • otherwise, it defaults to True - set to False to print less text to the console
  • the log level can be specified via the level arg, but defaults to whatever was set via the OTEL_LOG_LEVEL env var
import logging
from opentelemetry_wrapper import instrument_logging

instrument_logging()

logging.warning('...')

instrumenting fastapi

  • while the instrument_fastapi() function will auto-instrument any app that is created, if your import order is wrong then your app may end up not being instrumented
  • as such, it is safer to explicitly call instrument_fastapi_app
from fastapi import FastAPI
from opentelemetry_wrapper import instrument_fastapi_app

app = instrument_fastapi_app(FastAPI(...))

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

opentelemetry_wrapper-0.0.21.tar.gz (29.3 kB view hashes)

Uploaded Source

Built Distribution

opentelemetry_wrapper-0.0.21-py3-none-any.whl (30.9 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page