connection.py 文件源码

def connection_factory(tracer, service="postgres"):
    """ Return a connection factory class that will can be used to trace
        postgres queries.

        >>> factory = connection_factor(my_tracer, service="my_db_service")
        >>> conn = pyscopg2.connect(..., connection_factory=factory)
    """

    tracer.set_service_info(
        service=service,
        app="postgres",
        app_type=AppTypes.db,
    )

    return functools.partial(TracedConnection,
        datadog_tracer=tracer,
        datadog_service=service)