Source code for q2_sdk.extensions.Metrics.extension

"""
Metrics Extension returns usage data
"""

import prometheus_client

from q2_sdk.core.http_handlers.base_handler import Q2BaseRequestHandler




class MetricsHandler(Q2BaseRequestHandler):
    def __init__(self, application, request, **kwargs):
        super().__init__(application, request, **kwargs)
        self.enable_summary_log = False

    async def get(self):
        response = prometheus_client.generate_latest()
        self.set_header("Content-Type", "text/plain; charset=utf-8")
        self.write(response)