feat: migration 0002 for hosts, monitors, metrics, and alert tables
This commit is contained in:
@@ -5,6 +5,7 @@ from sqlalchemy.engine import Connection
|
||||
from sqlalchemy.ext.asyncio import async_engine_from_config
|
||||
from alembic import context
|
||||
from fablednetmon.models.base import Base
|
||||
import fablednetmon.models # noqa: F401 — registers all models in metadata
|
||||
|
||||
config = context.config
|
||||
if config.config_file_name is not None:
|
||||
|
||||
@@ -0,0 +1,140 @@
|
||||
"""Core monitors, metrics, and alert tables
|
||||
|
||||
Revision ID: 0002_core_monitors
|
||||
Revises: 0001_core_initial
|
||||
Create Date: 2026-03-17
|
||||
"""
|
||||
from typing import Sequence, Union
|
||||
from alembic import op
|
||||
import sqlalchemy as sa
|
||||
|
||||
revision: str = "0002_core_monitors"
|
||||
down_revision: Union[str, None] = "0001_core_initial"
|
||||
branch_labels: Union[str, Sequence[str], None] = None
|
||||
depends_on: Union[str, Sequence[str], None] = None
|
||||
|
||||
|
||||
def upgrade() -> None:
|
||||
# hosts
|
||||
op.create_table(
|
||||
"hosts",
|
||||
sa.Column("id", sa.String(36), primary_key=True),
|
||||
sa.Column("name", sa.String(128), nullable=False),
|
||||
sa.Column("address", sa.String(255), nullable=False),
|
||||
sa.Column(
|
||||
"probe_type",
|
||||
sa.Enum("tcp", "icmp", name="probetype"),
|
||||
nullable=False,
|
||||
server_default="tcp",
|
||||
),
|
||||
sa.Column("probe_port", sa.Integer, nullable=False, server_default="80"),
|
||||
sa.Column("ping_enabled", sa.Boolean, nullable=False, server_default="true"),
|
||||
sa.Column("dns_enabled", sa.Boolean, nullable=False, server_default="false"),
|
||||
sa.Column("dns_expected_ip", sa.String(255), nullable=True),
|
||||
sa.Column("poll_interval_seconds", sa.Integer, nullable=True),
|
||||
sa.Column("created_at", sa.DateTime(timezone=True), nullable=False),
|
||||
)
|
||||
|
||||
# ping_results
|
||||
op.create_table(
|
||||
"ping_results",
|
||||
sa.Column("id", sa.String(36), primary_key=True),
|
||||
sa.Column("host_id", sa.String(36), sa.ForeignKey("hosts.id", ondelete="CASCADE"), nullable=False),
|
||||
sa.Column("probed_at", sa.DateTime(timezone=True), nullable=False),
|
||||
sa.Column("status", sa.Enum("up", "down", name="pingstatus"), nullable=False),
|
||||
sa.Column("response_time_ms", sa.Float, nullable=True),
|
||||
)
|
||||
|
||||
# dns_results
|
||||
op.create_table(
|
||||
"dns_results",
|
||||
sa.Column("id", sa.String(36), primary_key=True),
|
||||
sa.Column("host_id", sa.String(36), sa.ForeignKey("hosts.id", ondelete="CASCADE"), nullable=False),
|
||||
sa.Column("resolved_at", sa.DateTime(timezone=True), nullable=False),
|
||||
sa.Column("status", sa.Enum("resolved", "failed", name="dnsstatus"), nullable=False),
|
||||
sa.Column("resolved_ip", sa.String(255), nullable=True),
|
||||
)
|
||||
|
||||
# plugin_metrics
|
||||
op.create_table(
|
||||
"plugin_metrics",
|
||||
sa.Column("id", sa.String(36), primary_key=True),
|
||||
sa.Column("source_module", sa.String(64), nullable=False),
|
||||
sa.Column("resource_name", sa.String(255), nullable=False),
|
||||
sa.Column("metric_name", sa.String(128), nullable=False),
|
||||
sa.Column("value", sa.Float, nullable=False),
|
||||
sa.Column("recorded_at", sa.DateTime(timezone=True), nullable=False),
|
||||
)
|
||||
|
||||
# alert_rules
|
||||
op.create_table(
|
||||
"alert_rules",
|
||||
sa.Column("id", sa.String(36), primary_key=True),
|
||||
sa.Column("name", sa.String(128), nullable=False),
|
||||
sa.Column("source_module", sa.String(64), nullable=False),
|
||||
sa.Column("resource_name", sa.String(255), nullable=False),
|
||||
sa.Column("metric_name", sa.String(128), nullable=False),
|
||||
sa.Column(
|
||||
"operator",
|
||||
sa.Enum(">", "<", ">=", "<=", "==", "!=", name="alertoperator"),
|
||||
nullable=False,
|
||||
),
|
||||
sa.Column("threshold", sa.Float, nullable=False),
|
||||
sa.Column("consecutive_failures_required", sa.Integer, nullable=False, server_default="1"),
|
||||
sa.Column("enabled", sa.Boolean, nullable=False, server_default="true"),
|
||||
sa.Column("created_by", sa.String(36), sa.ForeignKey("users.id", ondelete="RESTRICT"), nullable=False),
|
||||
sa.Column("created_at", sa.DateTime(timezone=True), nullable=False),
|
||||
)
|
||||
|
||||
# alert_states
|
||||
op.create_table(
|
||||
"alert_states",
|
||||
sa.Column("id", sa.String(36), primary_key=True),
|
||||
sa.Column(
|
||||
"rule_id",
|
||||
sa.String(36),
|
||||
sa.ForeignKey("alert_rules.id", ondelete="CASCADE"),
|
||||
nullable=False,
|
||||
unique=True,
|
||||
),
|
||||
sa.Column(
|
||||
"state",
|
||||
sa.Enum("inactive", "pending", "firing", "acknowledged", "resolved", name="alertstateenum"),
|
||||
nullable=False,
|
||||
server_default="inactive",
|
||||
),
|
||||
sa.Column("consecutive_failure_count", sa.Integer, nullable=False, server_default="0"),
|
||||
sa.Column("last_transition_at", sa.DateTime(timezone=True), nullable=False),
|
||||
sa.Column("acknowledged_by", sa.String(36), sa.ForeignKey("users.id", ondelete="SET NULL"), nullable=True),
|
||||
sa.Column("acknowledged_at", sa.DateTime(timezone=True), nullable=True),
|
||||
)
|
||||
|
||||
# alert_events
|
||||
op.create_table(
|
||||
"alert_events",
|
||||
sa.Column("id", sa.String(36), primary_key=True),
|
||||
sa.Column("rule_id", sa.String(36), sa.ForeignKey("alert_rules.id", ondelete="CASCADE"), nullable=False),
|
||||
sa.Column("from_state", sa.String(32), nullable=False),
|
||||
sa.Column("to_state", sa.String(32), nullable=False),
|
||||
sa.Column("metric_value", sa.Float, nullable=False),
|
||||
sa.Column("transitioned_at", sa.DateTime(timezone=True), nullable=False),
|
||||
sa.Column("email_sent", sa.Boolean, nullable=True),
|
||||
sa.Column("email_error", sa.Text, nullable=True),
|
||||
sa.Column("webhook_sent", sa.Boolean, nullable=True),
|
||||
sa.Column("webhook_error", sa.Text, nullable=True),
|
||||
)
|
||||
|
||||
|
||||
def downgrade() -> None:
|
||||
op.drop_table("alert_events")
|
||||
op.drop_table("alert_states")
|
||||
op.drop_table("alert_rules")
|
||||
op.drop_table("plugin_metrics")
|
||||
op.drop_table("dns_results")
|
||||
op.drop_table("ping_results")
|
||||
op.drop_table("hosts")
|
||||
op.execute("DROP TYPE IF EXISTS alertstateenum")
|
||||
op.execute("DROP TYPE IF EXISTS alertoperator")
|
||||
op.execute("DROP TYPE IF EXISTS dnsstatus")
|
||||
op.execute("DROP TYPE IF EXISTS pingstatus")
|
||||
op.execute("DROP TYPE IF EXISTS probetype")
|
||||
Reference in New Issue
Block a user