klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-04 21:29:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Do not start server when importing the main.py file, unless gunicorn

Browse Source 
- Add more graceful shutdown when closing bg scheduler thread
This commit is contained in:
sabaimran
2024-11-14 17:33:19 -08:00
parent be3ee5ec9f
commit c1a5b32ebf
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/khoj/main.py
View File

@@ -109,7 +109,10 @@ def shutdown_scheduler():
logger.info("🔓 Schedule Leader released")
ProcessLockAdapters.remove_process_lock(state.schedule_leader_process_lock)
state.scheduler.shutdown()
try:
state.scheduler.shutdown()
except Exception as e:
logger.debug(f"Did not shutdown scheduler: {e}")
def run(should_start_server=True):
@@ -237,5 +240,6 @@ def poll_task_scheduler():
if __name__ == "__main__":
run()
else:
run(should_start_server=False)
atexit.register(shutdown_scheduler)
if "gunicorn" in sys.modules:
run(should_start_server=False)
atexit.register(shutdown_scheduler)