45 lines
1.3 KiB
Python
45 lines
1.3 KiB
Python
"""Purge repack helper."""
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
from typing import TYPE_CHECKING
|
|
|
|
from sqlalchemy import text
|
|
|
|
if TYPE_CHECKING:
|
|
from . import Recorder
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
def repack_database(instance: Recorder) -> None:
|
|
"""Repack based on engine type."""
|
|
assert instance.engine is not None
|
|
dialect_name = instance.engine.dialect.name
|
|
|
|
# Execute sqlite command to free up space on disk
|
|
if dialect_name == "sqlite":
|
|
_LOGGER.debug("Vacuuming SQL DB to free space")
|
|
with instance.engine.connect() as conn:
|
|
conn.execute(text("VACUUM"))
|
|
conn.commit()
|
|
return
|
|
|
|
# Execute postgresql vacuum command to free up space on disk
|
|
if dialect_name == "postgresql":
|
|
_LOGGER.debug("Vacuuming SQL DB to free space")
|
|
with instance.engine.connect().execution_options(
|
|
isolation_level="AUTOCOMMIT"
|
|
) as conn:
|
|
conn.execute(text("VACUUM"))
|
|
conn.commit()
|
|
return
|
|
|
|
# Optimize mysql / mariadb tables to free up space on disk
|
|
if dialect_name == "mysql":
|
|
_LOGGER.debug("Optimizing SQL DB to free space")
|
|
with instance.engine.connect() as conn:
|
|
conn.execute(text("OPTIMIZE TABLE states, events, recorder_runs"))
|
|
conn.commit()
|
|
return
|