sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2025-02-17 08:04:21 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

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
Show stats 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") logger.info("🔓 Schedule Leader released")
ProcessLockAdapters.remove_process_lock(state.schedule_leader_process_lock) 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): def run(should_start_server=True):
@ -237,5 +240,6 @@ def poll_task_scheduler():
if __name__ == "__main__": if __name__ == "__main__":
run() run()
else: else:
run(should_start_server=False) if "gunicorn" in sys.modules:
atexit.register(shutdown_scheduler) run(should_start_server=False)
atexit.register(shutdown_scheduler)