mirror of
https://github.com/Wei-Shaw/sub2api.git
synced 2026-10-14 12:02:27 +08:00
shaw
4a7739acdf
235 只在「models_list_config 存在且 model_allowlist 不存在」时重命名，而迁移一旦
写入 schema_migrations 就会按「文件名 + checksum」整份跳过，不再重跑。因此只要数据库
在 235 之后回到旧结构（为回滚旧版本镜像手工改回列名、或按旧结构做了部分恢复），
应用依旧能正常启动，但每个关联 groups 的查询都会失败：
pq: column groups.model_allowlist does not exist
表现为「API 密钥」「我的订阅」「管理端订阅列表」等页面加载失败。
新增可重放的收敛迁移 236，升级即自愈：只有旧列则重命名（数据原样保留）；两列并存
则在新列仍为默认空值时回填旧列配置；两列都没有则按默认值补建。收尾统一把
model_allowlist 归一成 NOT NULL DEFAULT '{}'，与 ent schema 对齐。
判定列是否存在改用 attrelid = 'groups'::regclass（跟随 search_path），
不再写死 table_schema = 'public'，避免与 ALTER TABLE 的解析路径分叉。
Fixes #6780
53 lines
2.5 KiB
SQL
53 lines
2.5 KiB
SQL
-- 236: 收敛 groups 的模型白名单列，修复 235 未落地却已被记账的实例。
|
||
--
|
||
-- 235 只在「models_list_config 存在且 model_allowlist 不存在」时执行重命名，
|
||
-- 而迁移一旦写入 schema_migrations 就会按「文件名 + checksum」整份跳过，不再重跑。
|
||
-- 因此只要数据库在 235 之后回到了旧结构——例如为了回滚到旧版本镜像而手工把列名
|
||
-- 改回 models_list_config，或者用旧结构的备份做了部分恢复——应用依旧能正常启动，
|
||
-- 但每个关联 groups 的查询都会失败：
|
||
-- pq: column groups.model_allowlist does not exist
|
||
-- 表现为「API 密钥」「我的订阅」「管理端订阅列表」等页面加载失败（issue #6780）。
|
||
--
|
||
-- 本迁移可重放，并且用 regclass 解析表（跟随 search_path），不再假定 schema 为 public：
|
||
-- 1) 只有旧列 -> 重命名，数据原样保留；
|
||
-- 2) 两列并存 -> 新列仍是默认空值时，把旧列里的配置回填过来，旧列保留不动；
|
||
-- 3) 两列都没有 -> 按默认值补建新列。
|
||
-- 结束后保证 groups.model_allowlist 一定存在，且为 NOT NULL DEFAULT '{}'。
|
||
DO $$
|
||
BEGIN
|
||
IF EXISTS (
|
||
SELECT 1 FROM pg_attribute
|
||
WHERE attrelid = 'groups'::regclass
|
||
AND attname = 'models_list_config'
|
||
AND NOT attisdropped
|
||
) THEN
|
||
IF NOT EXISTS (
|
||
SELECT 1 FROM pg_attribute
|
||
WHERE attrelid = 'groups'::regclass
|
||
AND attname = 'model_allowlist'
|
||
AND NOT attisdropped
|
||
) THEN
|
||
ALTER TABLE groups RENAME COLUMN models_list_config TO model_allowlist;
|
||
ELSE
|
||
EXECUTE $backfill$
|
||
UPDATE groups
|
||
SET model_allowlist = models_list_config
|
||
WHERE COALESCE(model_allowlist, '{}'::jsonb) = '{}'::jsonb
|
||
AND COALESCE(models_list_config, '{}'::jsonb) <> '{}'::jsonb
|
||
$backfill$;
|
||
END IF;
|
||
END IF;
|
||
END
|
||
$$;
|
||
|
||
ALTER TABLE groups
|
||
ADD COLUMN IF NOT EXISTS model_allowlist JSONB NOT NULL DEFAULT '{}'::jsonb;
|
||
|
||
UPDATE groups SET model_allowlist = '{}'::jsonb WHERE model_allowlist IS NULL;
|
||
|
||
ALTER TABLE groups ALTER COLUMN model_allowlist SET DEFAULT '{}'::jsonb;
|
||
ALTER TABLE groups ALTER COLUMN model_allowlist SET NOT NULL;
|
||
|
||
COMMENT ON COLUMN groups.model_allowlist IS
|
||
'Group model allowlist: constrains both model listing responses and request admission';
|