mirror of
https://github.com/yogeshojha/rengine.git
synced 2026-10-20 15:21:30 +02:00
SQLModel emitted index=True beside primary_key=True on every id field, so 41 tables carried ix_<table>_id alongside <table>_pkey: the same single-column btree twice. 12 MB of duplicate index, and every insert maintaining both — on the hot path of a scan writing 345,152 endpoints. The planner had been using them (72,160 scans across the set) simply because they were there and equivalent; with them gone it uses the primary key, verified on subdomains: Index Scan using subdomains_pkey. The migration finds them rather than naming them, because which tables exist depends on how far an install has migrated and a name already gone must not fail an upgrade. It drops only a non-unique, single-column btree on id belonging to a table that has a primary key — no other index can match that shape. There is no downgrade: the primary key serves every lookup they served. The 33 models are corrected too, so the next hand-written migration is not comparing against a schema that would want them back. Counted honestly: 41, not the 102 an earlier pass reported — that number included the 61 primary keys themselves. Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_013dKUV96WkQQETZBMoAG98C
94 lines
2.5 KiB
Python
94 lines
2.5 KiB
Python
import uuid
|
|
from datetime import datetime
|
|
from functools import partial
|
|
|
|
from pydantic import BaseModel, field_validator
|
|
from sqlmodel import Field, SQLModel
|
|
|
|
from shared.enums.proxy import ProxyMode, ProxyProtocol
|
|
from shared.utils.datetime import utc_now
|
|
from shared.utils.validation import clean_name, clean_optional_name
|
|
|
|
PROXY_MODES = tuple(m.value for m in ProxyMode)
|
|
PROXY_SCHEMES = tuple(p.value for p in ProxyProtocol)
|
|
|
|
|
|
class ProxyEndpoint(BaseModel):
|
|
scheme: str
|
|
host: str
|
|
port: int
|
|
username: str | None = None
|
|
password: str | None = None
|
|
|
|
|
|
class Proxy(SQLModel, table=True):
|
|
__tablename__ = "proxies"
|
|
|
|
id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
|
|
name: str = Field(max_length=120)
|
|
description: str | None = Field(default=None, max_length=500)
|
|
mode: str = Field(default="single")
|
|
is_active: bool = Field(default=True)
|
|
is_default: bool = Field(default=False)
|
|
endpoints_encrypted: str
|
|
endpoint_count: int = Field(default=0)
|
|
created_by: uuid.UUID = Field(foreign_key="users.id")
|
|
created_at: datetime = Field(default_factory=utc_now)
|
|
updated_at: datetime = Field(default_factory=utc_now)
|
|
last_test_at: datetime | None = Field(default=None)
|
|
last_test_ok: bool | None = Field(default=None)
|
|
last_test_message: str | None = Field(default=None)
|
|
|
|
|
|
class ProxyEndpointRead(BaseModel):
|
|
scheme: str
|
|
host: str
|
|
port: int
|
|
username: str | None = None
|
|
has_password: bool
|
|
url_masked: str
|
|
|
|
|
|
class ProxyCreate(BaseModel):
|
|
name: str
|
|
description: str | None = None
|
|
mode: str = "single"
|
|
is_active: bool = True
|
|
is_default: bool = False
|
|
endpoints: list[ProxyEndpoint] = Field(min_length=1)
|
|
|
|
_validate_name = field_validator("name")(clean_name)
|
|
|
|
|
|
class ProxyUpdate(BaseModel):
|
|
name: str | None = None
|
|
description: str | None = None
|
|
mode: str | None = None
|
|
is_active: bool | None = None
|
|
is_default: bool | None = None
|
|
endpoints: list[ProxyEndpoint] | None = None
|
|
|
|
_validate_name = field_validator("name")(partial(clean_optional_name, max_len=120))
|
|
|
|
|
|
class ProxyRead(BaseModel):
|
|
id: uuid.UUID
|
|
name: str
|
|
description: str | None
|
|
mode: str
|
|
is_active: bool
|
|
is_default: bool
|
|
endpoints: list[ProxyEndpointRead]
|
|
endpoint_count: int
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
last_test_at: datetime | None
|
|
last_test_ok: bool | None
|
|
last_test_message: str | None
|
|
|
|
|
|
class ProxyTestResult(BaseModel):
|
|
success: bool
|
|
message: str
|
|
latency_ms: int | None = None
|