Marmelator/pyinfra
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 36 Wiki Activity

add prometheus metric name for default operation

Browse Source
This commit is contained in:
Julius Unverfehrt 2022-06-22 17:04:54 +02:00
parent d207b2e274
commit 9a7c412523
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pyinfra/server/server.py
View File

@ -5,6 +5,7 @@ from flask import Flask, jsonify, request
from funcy import merge
from prometheus_client import generate_latest, Summary, CollectorRegistry
from pyinfra.config import CONFIG
from pyinfra.server.buffering.stream import FlatStreamBuffer
from pyinfra.server.stream.queued_stream_function import QueuedStreamFunction
from pyinfra.server.stream.rest import LazyRestProcessor
@ -58,7 +59,7 @@ def __set_up_processing_server(operation2function: Dict[str, QueuedStreamFunctio
}
def make_summary_instance(op: str):
op = op.replace("_pickup", "")
op = op.replace("_pickup", "") if op != "pickup" else CONFIG.service.prometheus_metric_name
return Summary(f"redactmanager_{op}_seconds", f"Time spent on {op}.", registry=registry)
submit_operation2processor = {submit_suffix(op): prc for op, prc in operation2processor.items()}