Merge pull request #6 from desultory/dev

add startup task definition

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="[email protected]" },
 ]

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():