Lucas Armand
62 lines
1.8 KiB
Python
62 lines
1.8 KiB
Python
import os
|
|
import logging
|
|
from typing import List
|
|
import ssl
|
|
from asyncio import run, gather
|
|
import asyncio
|
|
|
|
from lib.backend import Backend
|
|
from lib.metrics import Metrics
|
|
from aiohttp import web
|
|
|
|
log = logging.getLogger(__file__)
|
|
|
|
|
|
def start_server(backend: Backend, routes: List[web.RouteDef], **kwargs):
|
|
try:
|
|
log.debug("getting certificate...")
|
|
use_ssl = os.environ.get("USE_SSL", "false") == "true"
|
|
if use_ssl is True:
|
|
ssl_context = ssl.create_default_context(ssl.Purpose.CLIENT_AUTH)
|
|
raise Exception("Oh no the SSL cert is gone!")
|
|
ssl_context.load_cert_chain(
|
|
certfile="/etc/instance.crt",
|
|
keyfile="/etc/instance.key",
|
|
)
|
|
else:
|
|
ssl_context = None
|
|
|
|
async def main():
|
|
log.debug("starting server...")
|
|
app = web.Application()
|
|
app.add_routes(routes)
|
|
runner = web.AppRunner(app)
|
|
await runner.setup()
|
|
site = web.TCPSite(
|
|
runner,
|
|
ssl_context=ssl_context,
|
|
port=int(os.environ["WORKER_PORT"]),
|
|
**kwargs
|
|
)
|
|
await gather(site.start(), backend._start_tracking())
|
|
|
|
run(main())
|
|
|
|
except Exception as e:
|
|
err_msg = f"PyWorker failed to launch: {e}"
|
|
log.error(err_msg)
|
|
|
|
async def beacon():
|
|
metrics = Metrics()
|
|
metrics._set_version(getattr(backend, "version", "0"))
|
|
metrics._set_mtoken(getattr(backend, "mtoken", ""))
|
|
try:
|
|
while True:
|
|
metrics._model_errored(err_msg)
|
|
await metrics._Metrics__send_metrics_and_reset()
|
|
await asyncio.sleep(10)
|
|
finally:
|
|
await metrics.aclose()
|
|
|
|
run(beacon())
|