From 28207773b2bd71f6bd116e973c6a43aea79c480c Mon Sep 17 00:00:00 2001 From: Zen Date: Mon, 10 Feb 2025 16:53:11 -0600 Subject: [PATCH] add startup task definition Signed-off-by: Zen --- pyproject.toml | 2 +- src/prometheus_exporter/exporter.py | 4 ++++ 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index 75b4760..b66699b 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta" [project] name = "prometheus_exporter" -version = "1.2.0" +version = "1.3.0" authors = [ { name="Desultory", email="dev@pyl.onl" }, ] diff --git a/src/prometheus_exporter/exporter.py b/src/prometheus_exporter/exporter.py index a32b101..5510283 100644 --- a/src/prometheus_exporter/exporter.py +++ b/src/prometheus_exporter/exporter.py @@ -46,6 +46,7 @@ def __init__(self, config_file="config.toml", name=None, labels=Labels(), no_con self.app = Application(logger=self.logger) signal(SIGINT, lambda *args: ensure_future(self.app.shutdown())) + self.app.on_startup.append(self.startup_tasks) self.app.add_routes([get("/metrics", self.handle_metrics)]) self.app.on_shutdown.append(self.on_shutdown) @@ -78,6 +79,9 @@ def start(self): self.logger.info("Exporter server address: %s:%d" % (self.listen_ip, self.listen_port)) web.run_app(self.app, host=self.listen_ip, port=self.listen_port) + async def startup_tasks(self, *args, **kwargs): + pass + async def on_shutdown(self, app): self.logger.info("Shutting down exporter server") for task in all_tasks():