Felipe Cardoso
4c6bf55bcc
- Consistently format multi-line function headers, exception handling, and repository method calls for readability. - Reorganize misplaced imports across modules (e.g., services & tests) into proper sorted order. - Adjust indentation, line breaks, and spacing inconsistencies in tests and migration files. - Cleanup unnecessary trailing newlines and reorganize `__all__` declarations for consistency.
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")
|