Rob Ballantyne
e0be45f39a
* Addresses breaking change in test_utils.py Endpoint.get_endpoint_api_key() now requires instance Moves the call to this function out of the APIClient and into main * Ensure make_benchmark_payload has a value to calculate the workload --------- Co-authored-by: Nader Arbabian <nader@vast.ai>
41 lines
1.0 KiB
Python
41 lines
1.0 KiB
Python
import os
|
|
import logging
|
|
from typing import List
|
|
import ssl
|
|
from asyncio import run, gather
|
|
|
|
|
|
from lib.backend import Backend
|
|
from aiohttp import web
|
|
|
|
log = logging.getLogger(__file__)
|
|
|
|
|
|
def start_server(backend: Backend, routes: List[web.RouteDef], **kwargs):
|
|
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)
|
|
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())
|