Server : Apache System : Linux indy02.toastserver.com 3.10.0-962.3.2.lve1.5.85.el7.x86_64 #1 SMP Thu Apr 18 15:18:36 UTC 2024 x86_64 User : palandch ( 1163) PHP Version : 7.1.33 Disable Function : NONE Directory :  /opt/cloudlinux/venv/lib/python3.11/site-packages/lvestats/plugins/generic/burster/storage/

Upload File : current_dir [ Writeable ] document_root [ Writeable ]

# coding=utf-8
#
# Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2023 All Rights Reserved
#
# Licensed under CLOUD LINUX LICENSE AGREEMENT
# http://cloudlinux.com/docs/LICENSE.TXT
import contextlib
from typing import Any, TypedDict, Callable
from threading import Thread, Event

import sqlalchemy as sa

from lvestats.orm import BurstingEventType, bursting_events_table

from ..common import LveId, Timestamp


def init_db_schema(engine: sa.engine.Engine) -> None:
    bursting_events_table.create(engine, checkfirst=True)


def drop_db_schema(engine: sa.engine.Engine) -> None:
    bursting_events_table.drop(engine, checkfirst=True)


class InBurstingEventRow(TypedDict):
    lve_id: LveId
    timestamp: Timestamp
    event_type: BurstingEventType


@contextlib.contextmanager
def thread_running(name: str, main: Callable[[Event], Any]):
    terminate = Event()
    (saver_thread := Thread(
        name=name,
        target=main,
        # NOTE(vlebedev): For details regarding how this interacts with termination using "atexit" callback,
        #                 have a look at https://stackoverflow.com/a/58928213/3344105
        daemon=True,
        args=(terminate,),
    )).start()
    try:
        yield
    finally:
        terminate.set()
        saver_thread.join()