Improve app loader

bioengine/__init__.py

@@ -1,3 +1,4 @@
+import asyncio
 from bioengine.app_loader import load_apps
 import logging
 
@@ -6,8 +7,26 @@
 def connect_server(server_url):
     raise NotImplementedError("This function is not implemented yet.")
 
-async def register_bioengine_apps(server):
+async def run_app_with_error_logging(app, server, continue_on_error):
+    try:
+        return await app.run(server)
+    except Exception as e:
+        logger.error(f"Error running service {app.name}: {str(e)}")
+        if not continue_on_error:
+            raise
+        else:
+            return e
+
+async def register_bioengine_apps(server, continue_on_error=False):
+    futures = []
+    app_names = []
     for app in load_apps():
+        app_names.append(app.name)
         logger.info(f"Registering service {app.name}")
-        await app.run(server)
-        logger.info(f"Service {app.name} registered.")
+        futures.append(run_app_with_error_logging(app, server, continue_on_error))
+
+    results = await asyncio.gather(*futures)
+    summary = {name: ("✔️" if result is not None else "❌") for name, result in zip(app_names, results)}
+
+    for app_name, status in summary.items():
+        print(f"{app_name}: {status}")

bioengine/app_loader.py

@@ -37,9 +37,11 @@ def load_apps():
     current_dir = Path(__file__).parent
     apps_dir = current_dir.parent / "apps"
     # list folders under apps_dir
+    apps = []
     for app_dir in apps_dir.iterdir():
         if app_dir.is_dir():
             manifest_file = app_dir / "manifest.yaml"
             if manifest_file.exists():
                 manifest = safe_load(manifest_file.read_text())
-                yield AppInfo.parse_obj(manifest)
+                apps.append(AppInfo.parse_obj(manifest))
+    return apps