highperfocused/lnbits
1
0
Fork 0
mirror of https://github.com/lnbits/lnbits.git synced 2025-03-28 18:52:00 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix: install extensions async (#2596)

Browse Source 
so it does not block webserver start on saas instances and comes up
faster if extensions are reinstalled
This commit is contained in:
dni ⚡ 2024-07-23 14:01:34 +02:00 committed by GitHub
parent f9133760fc
commit eb8d2f312f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 13 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
lnbits/app.py
View File

@ -63,6 +63,7 @@ from .middleware import (
from .requestvars import g from .requestvars import g
from .tasks import ( from .tasks import (
check_pending_payments, check_pending_payments,
create_task,
internal_invoice_listener, internal_invoice_listener,
invoice_listener, invoice_listener,
) )
@ -93,13 +94,8 @@ async def startup(app: FastAPI):
# register core routes # register core routes
init_core_routers(app) init_core_routers(app)
# check extensions after restart
if not settings.lnbits_extensions_deactivate_all:
await check_installed_extensions(app)
register_all_ext_routes(app)
# initialize tasks # initialize tasks
register_async_tasks() register_async_tasks(app)
async def shutdown(): async def shutdown():
@ -399,22 +395,28 @@ def register_ext_routes(app: FastAPI, ext: Extension) -> None:
app.include_router(router=ext_route, prefix=prefix) app.include_router(router=ext_route, prefix=prefix)
def register_all_ext_routes(app: FastAPI): async def check_and_register_extensions(app: FastAPI):
await check_installed_extensions(app)
for ext in get_valid_extensions(False): for ext in get_valid_extensions(False):
try: try:
register_ext_routes(app, ext) register_ext_routes(app, ext)
except Exception as e: except Exception as exc:
logger.error(f"Could not load extension `{ext.code}`: {e!s}") logger.error(f"Could not load extension `{ext.code}`: {exc!s}")
def register_async_tasks(): def register_async_tasks(app: FastAPI):
# check extensions after restart
if not settings.lnbits_extensions_deactivate_all:
create_task(check_and_register_extensions(app))
create_permanent_task(check_pending_payments) create_permanent_task(check_pending_payments)
create_permanent_task(invoice_listener) create_permanent_task(invoice_listener)
create_permanent_task(internal_invoice_listener) create_permanent_task(internal_invoice_listener)
create_permanent_task(cache.invalidate_forever) create_permanent_task(cache.invalidate_forever)
# core invoice listener # core invoice listener
invoice_queue = asyncio.Queue(5) invoice_queue: asyncio.Queue = asyncio.Queue(5)
register_invoice_listener(invoice_queue, "core") register_invoice_listener(invoice_queue, "core")
create_permanent_task(lambda: wait_for_paid_invoices(invoice_queue)) create_permanent_task(lambda: wait_for_paid_invoices(invoice_queue))