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

- 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

View file

@ -109,7 +109,10 @@ def shutdown_scheduler():
logger.info("🔓 Schedule Leader released")
ProcessLockAdapters.remove_process_lock(state.schedule_leader_process_lock)
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:
if "gunicorn" in sys.modules:
run(should_start_server=False)
atexit.register(shutdown_scheduler)