Function track_request_latency

sdk/python/feast/metrics.py:319–351

Context manager that records endpoint latency and increments request count. Yields a :class:`RequestMetricsContext` whose ``feature_count`` and ``feature_view_count`` attributes can be updated inside the block. The final values are used when recording the histogram and counter in ``

(
    endpoint: str, feature_count: str = "", feature_view_count: str = ""
)

@contextmanager
def track_request_latency(
    endpoint: str, feature_count: str = "", feature_view_count: str = ""
):
    """Context manager that records endpoint latency and increments request count.

    Yields a :class:`RequestMetricsContext` whose ``feature_count`` and
    ``feature_view_count`` attributes can be updated inside the block.
    The final values are used when recording the histogram and counter
    in ``finally``, so labels are accurate even when they depend on
    work done inside the block.

    Gated by the ``request`` category flag.
    """
    ctx = RequestMetricsContext(feature_count, feature_view_count)
    if not _config.request:
        yield ctx
        return

    start = time.monotonic()
    status_label = "success"
    try:
        yield ctx
    except Exception:
        status_label = "error"
        raise
    finally:
        elapsed = time.monotonic() - start
        request_latency.labels(
            endpoint=endpoint,
            feature_count=ctx.feature_count,
            feature_view_count=ctx.feature_view_count,
        ).observe(elapsed)
        request_count.labels(endpoint=endpoint, status=status_label).inc()


def track_online_features_entities(entity_count: int):