i/Aether
1
0
Fork 0
mirror of https://github.com/fawney19/Aether.git synced 2026-09-03 01:40:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: 优化迁移脚本效率

Browse Source
This commit is contained in:
fawney19
2026-03-08 18:42:27 +08:00
parent 2d9158b321
commit f490c3a5cd
2 changed files with 60 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
alembic/versions/20260308_0348_45b118150a78_usage_stats_retention_set_null_on_.py
View File

@@ -112,7 +112,18 @@ def upgrade() -> None:
comment='API Key 名称快照（删除 Key 后仍可追溯）'))
# --- Backfill: populate snapshots from existing FK joins ---
# Single UPDATE per table using LEFT JOINs to fill both columns in one pass.
# (WHERE ... IS NULL makes these inherently idempotent)
op.execute("""
UPDATE usage u
SET username = COALESCE(u.username, usr.username),
api_key_name = COALESCE(u.api_key_name, ak.name)
FROM users usr, api_keys ak
WHERE usr.id = u.user_id
AND ak.id = u.api_key_id
AND (u.username IS NULL OR u.api_key_name IS NULL)
""")
# Catch rows that have user_id but no api_key_id (or vice versa)
op.execute("""
UPDATE usage u
SET username = usr.username

73
alembic/versions/20260308_1215_13a4c8f6d9e0_request_candidate_video_task_retention.py
View File

@@ -138,41 +138,66 @@ def upgrade() -> None:
)
# --- Backfill: populate snapshots from existing FK joins ---
# Single UPDATE per table using JOINs to fill both columns in one pass.
# (WHERE ... IS NULL makes these inherently idempotent)
# request_candidates: fill both columns where both FKs exist
op.execute(
"""
UPDATE request_candidates
SET username = (
SELECT users.username FROM users WHERE users.id = request_candidates.user_id
)
WHERE username IS NULL AND user_id IS NOT NULL
UPDATE request_candidates rc
SET username = COALESCE(rc.username, usr.username),
api_key_name = COALESCE(rc.api_key_name, ak.name)
FROM users usr, api_keys ak
WHERE usr.id = rc.user_id
AND ak.id = rc.api_key_id
AND (rc.username IS NULL OR rc.api_key_name IS NULL)
"""
)
# Catch rows with only user_id or only api_key_id
op.execute(
"""
UPDATE request_candidates rc
SET username = usr.username
FROM users usr
WHERE usr.id = rc.user_id AND rc.username IS NULL
"""
)
op.execute(
"""
UPDATE request_candidates
SET api_key_name = (
SELECT api_keys.name FROM api_keys WHERE api_keys.id = request_candidates.api_key_id
)
WHERE api_key_name IS NULL AND api_key_id IS NOT NULL
UPDATE request_candidates rc
SET api_key_name = ak.name
FROM api_keys ak
WHERE ak.id = rc.api_key_id AND rc.api_key_name IS NULL
"""
)
# video_tasks: fill both columns where both FKs exist
op.execute(
"""
UPDATE video_tasks vt
SET username = COALESCE(vt.username, usr.username),
api_key_name = COALESCE(vt.api_key_name, ak.name)
FROM users usr, api_keys ak
WHERE usr.id = vt.user_id
AND ak.id = vt.api_key_id
AND (vt.username IS NULL OR vt.api_key_name IS NULL)
"""
)
# Catch rows with only user_id or only api_key_id
op.execute(
"""
UPDATE video_tasks vt
SET username = usr.username
FROM users usr
WHERE usr.id = vt.user_id AND vt.username IS NULL
"""
)
op.execute(
"""
UPDATE video_tasks
SET username = (
SELECT users.username FROM users WHERE users.id = video_tasks.user_id
)
WHERE username IS NULL AND user_id IS NOT NULL
"""
)
op.execute(
"""
UPDATE video_tasks
SET api_key_name = (
SELECT api_keys.name FROM api_keys WHERE api_keys.id = video_tasks.api_key_id
)
WHERE api_key_name IS NULL AND api_key_id IS NOT NULL
UPDATE video_tasks vt
SET api_key_name = ak.name
FROM api_keys ak
WHERE ak.id = vt.api_key_id AND vt.api_key_name IS NULL
"""
)