Nathan Woodburn
8a43526aab
All checks were successful
Build Docker / BuildImage (push) Successful in 33s
88 lines
2.2 KiB
Python
88 lines
2.2 KiB
Python
import time
|
|
import signal
|
|
import threading
|
|
from flask import Flask
|
|
from server import app
|
|
import server
|
|
from gunicorn.app.base import BaseApplication
|
|
import os
|
|
import dotenv
|
|
import concurrent.futures
|
|
import schedule
|
|
|
|
|
|
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
|
|
|
|
|
|
def check():
|
|
print('Checking nodes...', flush=True)
|
|
server.check_nodes()
|
|
|
|
|
|
def run_scheduler(stop_event):
|
|
schedule.every(5).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)
|
|
workers = int(workers)
|
|
threads = int(threads)
|
|
|
|
options = {
|
|
'bind': '0.0.0.0:5000',
|
|
'workers': workers,
|
|
'threads': threads,
|
|
}
|
|
|
|
gunicorn_app = GunicornApp(server.app, options)
|
|
print(f'Starting server with {workers} workers and {threads} threads', flush=True)
|
|
gunicorn_app.run()
|
|
|
|
|
|
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}")
|