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