diff --git a/pyinfra/server/server.py b/pyinfra/server/server.py index 5a3589e..2054ebb 100644 --- a/pyinfra/server/server.py +++ b/pyinfra/server/server.py @@ -48,6 +48,10 @@ def __set_up_processing_server_impl(operation2function: Dict[str, QueuedStreamFu app = Flask(__name__) registry = CollectorRegistry(auto_describe=True) + operation2processor = { + op: LazyRestProcessor(fn, **build_endpoint_suffixes(op)) for op, fn in operation2function.items() + } + def make_summary_instance(op): return Summary(f"redactmanager_{op}_seconds", f"Time spent on {op}.", registry=registry) @@ -55,13 +59,8 @@ def __set_up_processing_server_impl(operation2function: Dict[str, QueuedStreamFu pickup_operation2processor = {pickup_suffix(op): prc for op, prc in operation2processor.items()} operation2processor = merge(submit_operation2processor, pickup_operation2processor) - operation2processor = { - op: LazyRestProcessor(fn, **build_endpoint_suffixes(op)) for op, fn in operation2function.items() - } - pickup_operation2metric = {op: make_summary_instance(op) for op in operation2processor} - @app.route("/ready", methods=["GET"]) def ready(): resp = jsonify("OK")