hnsdoh-status/main.py

88 lines
2.2 KiB
Python
Raw Normal View History

2024-09-12 13:51:01 +10:00
import time
2024-09-12 15:04:08 +10:00
import signal
import threading
2024-09-12 13:51:01 +10:00
from flask import Flask
from server import app
import server
from gunicorn.app.base import BaseApplication
import os
import dotenv
2024-09-12 15:04:08 +10:00
import concurrent.futures
import schedule
2024-09-12 13:51:01 +10:00
class GunicornApp(BaseApplication):
def __init__(self, app, options=None):
self.options = options or {}
self.application = app
super().__init__()
def load_config(self):
for key, value in self.options.items():
if key in self.cfg.settings and value is not None:
self.cfg.set(key.lower(), value)
def load(self):
return self.application
2024-09-12 15:04:08 +10:00
def check():
print('Checking nodes...', flush=True)
server.check_nodes()
def run_scheduler(stop_event):
schedule.every(1).minutes.do(check)
while not stop_event.is_set():
schedule.run_pending()
time.sleep(1)
def run_gunicorn():
workers = os.getenv('WORKERS', 1)
threads = os.getenv('THREADS', 2)
2024-09-12 13:51:01 +10:00
workers = int(workers)
threads = int(threads)
2024-09-12 15:04:08 +10:00
2024-09-12 13:51:01 +10:00
options = {
'bind': '0.0.0.0:5000',
'workers': workers,
'threads': threads,
}
2024-09-12 15:04:08 +10:00
gunicorn_app = GunicornApp(server.app, options)
print(f'Starting server with {workers} workers and {threads} threads', flush=True)
2024-09-12 13:51:01 +10:00
gunicorn_app.run()
2024-09-12 15:04:08 +10:00
def signal_handler(sig, frame):
print("Shutting down gracefully...", flush=True)
stop_event.set()
if __name__ == '__main__':
dotenv.load_dotenv()
stop_event = threading.Event()
signal.signal(signal.SIGINT, signal_handler)
signal.signal(signal.SIGTERM, signal_handler)
with concurrent.futures.ThreadPoolExecutor() as executor:
# Start the scheduler
scheduler_future = executor.submit(run_scheduler, stop_event)
try:
# Run the Gunicorn server
run_gunicorn()
except KeyboardInterrupt:
print("Shutting down server...", flush=True)
finally:
stop_event.set()
scheduler_future.cancel()
try:
scheduler_future.result(timeout=5)
except concurrent.futures.CancelledError:
print("Scheduler stopped.")
except Exception as e:
print(f"Scheduler did not stop cleanly: {e}")