mirror of
https://codeberg.org/Freedium-cfd/web.git
synced 2026-03-11 09:04:37 +00:00
91 lines
2.5 KiB
Python
91 lines
2.5 KiB
Python
import io
|
|
import atexit
|
|
import multiprocessing
|
|
from multiprocessing.util import _exit_function
|
|
|
|
import traceback
|
|
import gunicorn.app.base
|
|
import uvicorn
|
|
from loguru import logger
|
|
|
|
from server import config, ban_db
|
|
from server.main import app
|
|
from server.utils.notify import send_message
|
|
from server.utils.logger import GunicornLogger
|
|
from server.utils.logger_trace import trace
|
|
|
|
logger.trace(f"Uvicorn version: {uvicorn.__version__}")
|
|
|
|
|
|
def post_worker_init(worker):
|
|
# Remove the atexit handler set up by the parent process
|
|
# https://github.com/benoitc/gunicorn/issues/1391#issuecomment-467010209
|
|
logger.trace("Removing atexit handler")
|
|
atexit.unregister(_exit_function)
|
|
|
|
|
|
def on_exit():
|
|
logger.debug("GUNICORN: On exit")
|
|
ban_db.dump()
|
|
|
|
|
|
def worker_exit(server, worker):
|
|
"""Called just after a worker has been exited, in the worker process."""
|
|
logger.warning("Shutting down gunicorn worker.")
|
|
|
|
send_message("Shutting down gunicorn worker.", status="ERROR")
|
|
|
|
|
|
def worker_abort(worker):
|
|
"""Log the stack trace when a worker timeout occurs"""
|
|
buffer = io.StringIO()
|
|
traceback.print_stack(file=buffer)
|
|
data = buffer.getvalue()
|
|
buffer.close()
|
|
|
|
logger.error(f"Killing worker {worker.pid}\n{data}")
|
|
|
|
send_message(f"Killing worker {worker.pid}\n{data}", status="ERROR")
|
|
|
|
|
|
@trace
|
|
def number_of_workers():
|
|
cores = multiprocessing.cpu_count()
|
|
if cores >= 8:
|
|
workers = cores
|
|
else:
|
|
workers = cores * 2
|
|
# workers = (cores * 2) + 2
|
|
logger.debug(f"Number of workers: {workers}")
|
|
return workers
|
|
|
|
|
|
class GunicornStandaloneApplication(gunicorn.app.base.BaseApplication):
|
|
def __init__(self, app, options=None):
|
|
self.options = options or {}
|
|
self.application = app
|
|
super().__init__()
|
|
|
|
def load_config(self):
|
|
config = {key: value for key, value in self.options.items() if key in self.cfg.settings and value is not None}
|
|
for key, value in config.items():
|
|
self.cfg.set(key.lower(), value)
|
|
|
|
def load(self):
|
|
return self.application
|
|
|
|
|
|
def execute_server_worker(host: str, port: int):
|
|
options = {
|
|
"bind": f"{host}:{port}",
|
|
"workers": number_of_workers(),
|
|
"logger_class": GunicornLogger,
|
|
"worker_class": "uvicorn.workers.UvicornWorker",
|
|
"preload_app": True,
|
|
"post_worker_init": post_worker_init,
|
|
"timeout": config.WORKER_TIMEOUT,
|
|
"worker_exit": worker_exit,
|
|
"worker_abort": worker_abort,
|
|
# "on_exit": on_exit,
|
|
}
|
|
GunicornStandaloneApplication(app, options).run()
|