51 lines
1.2 KiB
Python
51 lines
1.2 KiB
Python
import logging
|
|
|
|
from .consts import ADAPTERS
|
|
from .worker import ServiceWorker
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
_COLUMNS = {
|
|
'timestamp': 'timestamp bigint',
|
|
'status': 'status bool',
|
|
'latency': 'latency bigint',
|
|
}
|
|
|
|
|
|
class ServiceManager:
|
|
def __init__(self, app):
|
|
self.app = app
|
|
self.cfg = app.cfg
|
|
self.conn = app.conn
|
|
self.loop = app.loop
|
|
|
|
self.workers = {}
|
|
self.state = {}
|
|
|
|
self._start()
|
|
|
|
def _make_db_table(self, name: str, service: dict):
|
|
adapter = ADAPTERS[service['adapter']]
|
|
|
|
columnstr = map(_COLUMNS.get, adapter.spec['db'])
|
|
columnstr = ',\n'.join(columnstr)
|
|
|
|
log.info(f'Making table for {name}')
|
|
self.conn.executescript(f"""
|
|
CREATE TABLE IF NOT EXISTS {name} (
|
|
{columnstr}
|
|
);
|
|
""")
|
|
|
|
def _start(self):
|
|
self.conn.executescript("""
|
|
""")
|
|
for name, service in self.cfg.SERVICES.items():
|
|
self._make_db_table(name, service)
|
|
|
|
# spawn a service worker
|
|
serv_worker = ServiceWorker(self, name, service)
|
|
self.workers[name] = serv_worker
|
|
|
|
self.state[name] = False
|