Felipe Cardoso
acd18ff694
Reformatted multiline function calls, object definitions, and queries for improved code readability and consistency. Adjusted imports and constraints where necessary.
128 lines
4.6 KiB
Python
128 lines
4.6 KiB
Python
"""Add performance indexes
|
|
|
|
Revision ID: 0002
|
|
Revises: 0001
|
|
Create Date: 2025-11-27
|
|
|
|
Performance indexes that Alembic cannot auto-detect:
|
|
- Functional indexes (LOWER expressions)
|
|
- Partial indexes (WHERE clauses)
|
|
|
|
These indexes use the ix_perf_ prefix and are excluded from autogenerate
|
|
via the include_object() function in env.py.
|
|
"""
|
|
|
|
from collections.abc import Sequence
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision: str = "0002"
|
|
down_revision: str | None = "0001"
|
|
branch_labels: str | Sequence[str] | None = None
|
|
depends_on: str | Sequence[str] | None = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
# ==========================================================================
|
|
# USERS TABLE - Performance indexes for authentication
|
|
# ==========================================================================
|
|
|
|
# Case-insensitive email lookup for login/registration
|
|
# Query: SELECT * FROM users WHERE LOWER(email) = LOWER(:email) AND deleted_at IS NULL
|
|
# Impact: High - every login, registration check, password reset
|
|
op.create_index(
|
|
"ix_perf_users_email_lower",
|
|
"users",
|
|
[sa.text("LOWER(email)")],
|
|
unique=False,
|
|
postgresql_where=sa.text("deleted_at IS NULL"),
|
|
)
|
|
|
|
# Active users lookup (non-soft-deleted)
|
|
# Query: SELECT * FROM users WHERE deleted_at IS NULL AND ...
|
|
# Impact: Medium - user listings, admin queries
|
|
op.create_index(
|
|
"ix_perf_users_active",
|
|
"users",
|
|
["is_active"],
|
|
unique=False,
|
|
postgresql_where=sa.text("deleted_at IS NULL"),
|
|
)
|
|
|
|
# ==========================================================================
|
|
# ORGANIZATIONS TABLE - Performance indexes for multi-tenant lookups
|
|
# ==========================================================================
|
|
|
|
# Case-insensitive slug lookup for URL routing
|
|
# Query: SELECT * FROM organizations WHERE LOWER(slug) = LOWER(:slug) AND is_active = true
|
|
# Impact: Medium - every organization page load
|
|
op.create_index(
|
|
"ix_perf_organizations_slug_lower",
|
|
"organizations",
|
|
[sa.text("LOWER(slug)")],
|
|
unique=False,
|
|
postgresql_where=sa.text("is_active = true"),
|
|
)
|
|
|
|
# ==========================================================================
|
|
# USER SESSIONS TABLE - Performance indexes for session management
|
|
# ==========================================================================
|
|
|
|
# Expired session cleanup
|
|
# Query: SELECT * FROM user_sessions WHERE expires_at < NOW() AND is_active = true
|
|
# Impact: Medium - background cleanup jobs
|
|
op.create_index(
|
|
"ix_perf_user_sessions_expires",
|
|
"user_sessions",
|
|
["expires_at"],
|
|
unique=False,
|
|
postgresql_where=sa.text("is_active = true"),
|
|
)
|
|
|
|
# ==========================================================================
|
|
# OAUTH PROVIDER TOKENS - Performance indexes for token management
|
|
# ==========================================================================
|
|
|
|
# Expired refresh token cleanup
|
|
# Query: SELECT * FROM oauth_provider_refresh_tokens WHERE expires_at < NOW() AND revoked = false
|
|
# Impact: Medium - OAuth token cleanup, validation
|
|
op.create_index(
|
|
"ix_perf_oauth_refresh_tokens_expires",
|
|
"oauth_provider_refresh_tokens",
|
|
["expires_at"],
|
|
unique=False,
|
|
postgresql_where=sa.text("revoked = false"),
|
|
)
|
|
|
|
# ==========================================================================
|
|
# OAUTH AUTHORIZATION CODES - Performance indexes for auth flow
|
|
# ==========================================================================
|
|
|
|
# Expired authorization code cleanup
|
|
# Query: DELETE FROM oauth_authorization_codes WHERE expires_at < NOW() AND used = false
|
|
# Impact: Low-Medium - OAuth cleanup jobs
|
|
op.create_index(
|
|
"ix_perf_oauth_auth_codes_expires",
|
|
"oauth_authorization_codes",
|
|
["expires_at"],
|
|
unique=False,
|
|
postgresql_where=sa.text("used = false"),
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
# Drop indexes in reverse order
|
|
op.drop_index(
|
|
"ix_perf_oauth_auth_codes_expires", table_name="oauth_authorization_codes"
|
|
)
|
|
op.drop_index(
|
|
"ix_perf_oauth_refresh_tokens_expires",
|
|
table_name="oauth_provider_refresh_tokens",
|
|
)
|
|
op.drop_index("ix_perf_user_sessions_expires", table_name="user_sessions")
|
|
op.drop_index("ix_perf_organizations_slug_lower", table_name="organizations")
|
|
op.drop_index("ix_perf_users_active", table_name="users")
|
|
op.drop_index("ix_perf_users_email_lower", table_name="users")
|