fix: 修复初次执行 migrate.sh 时 usage 表不存在的问题 (#43)

- 在 baseline 中直接创建 usage 表复合索引
- 在后续迁移中添加表存在性检查,避免 AUTOCOMMIT 连接看不到事务中的表
This commit is contained in:
fawney19
2025-12-24 01:24:05 +08:00
parent 15a9b88fc8
commit d73a443484
2 changed files with 20 additions and 0 deletions

View File

@@ -394,6 +394,10 @@ def upgrade() -> None:
index=True,
),
)
# usage 表复合索引(优化常见查询)
op.create_index("idx_usage_user_created", "usage", ["user_id", "created_at"])
op.create_index("idx_usage_apikey_created", "usage", ["api_key_id", "created_at"])
op.create_index("idx_usage_provider_model_created", "usage", ["provider", "model", "created_at"])
# ==================== user_quotas ====================
op.create_table(