fix(题单): 奖章补发改成二进制子命令,独立脚本在生产跑不了
Some checks failed
Deploy / deploy (push) Has been cancelled

上一版把补发写成 `bun src/scripts/backfill-problemset-badges.ts`,那在生产上根本执行
不了 —— api 镜像是 debian-slim + 一个编译好的 oj2-api,既没有 bun 也没有源码。

改成 main.ts 的子命令,跑法对齐 migrate(deploy.sh:197 就是这个形态):

  docker compose -f docker/compose.debian.yml run --rm oj-api oj2-api backfill-badges
  docker compose -f docker/compose.debian.yml run --rm oj-api oj2-api backfill-badges --apply

脚本本身从「导入即执行」改成导出一个返回退出码的函数,main.ts 负责解析参数和 exit。
package.json 的 backfill:badges 也指向同一入口,本机和生产走的是同一条代码路径。

用编译出来的二进制在 /tmp 下实跑过全部四条路径(预演 / --apply 后自动复核 / 重复跑
幂等 / 误发时拒绝执行并退 1,加 --allow-revoke 才收回),确认动态 import 的模块进了
bundle、不依赖源码树。

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01QqqZwxtXLo2GTqMi51C94D
This commit is contained in:
2026-08-31 08:26:27 -06:00
parent f9354b0df1
commit 05cf8011e3
3 changed files with 91 additions and 74 deletions

View File

@@ -11,7 +11,7 @@
"worker": "bun src/main.ts worker",
"build": "bun build --compile --target=bun-linux-x64 src/main.ts --outfile ../../dist/oj2-api",
"seed:dev": "bun src/scripts/seed-dev.ts",
"backfill:badges": "bun src/scripts/backfill-problemset-badges.ts",
"backfill:badges": "bun src/main.ts backfill-badges",
"typecheck": "tsc --noEmit",
"check:routes": "bun src/scripts/check-route-shadowing.ts",
"db:pull": "drizzle-kit pull",

View File

@@ -12,6 +12,7 @@
* oj2-api healthcheck # 探活,给 Dockerfile 的 HEALTHCHECK 用
* oj2-api sql-child # SQL 判题子进程,由服务自己 spawn不该手动调
* oj2-api migrate # 执行待办的数据库迁移,部署时由 docker/deploy.sh 调
* oj2-api backfill-badges # 补发历史欠账的题单奖章,默认只读预演,--apply 才落库
*
* 用动态 import 而非顶层 import这几个模块都有导入即执行的副作用
* Bun.serve、连 Redis 开消费者),静态导入会让 sql-child 也把整个服务拉起来。
@@ -33,6 +34,15 @@ switch (command) {
await runMigrations()
break
}
// 一次性的数据订正,跟着二进制走而不是留成源码脚本 —— 生产镜像里没有 bun 也没有源码
case "backfill-badges": {
const { backfillProblemSetBadges } = await import("./scripts/backfill-problemset-badges")
const args = process.argv.slice(3)
process.exit(await backfillProblemSetBadges({
apply: args.includes("--apply"),
allowRevoke: args.includes("--allow-revoke"),
}))
}
case "sql-child": {
const { runSqlChild } = await import("./judge/sql/child")
await runSqlChild()
@@ -53,6 +63,6 @@ switch (command) {
}
}
default:
console.error(`未知子命令:${command}\n可用serve | worker | migrate | healthcheck | sql-child`)
console.error(`未知子命令:${command}\n可用serve | worker | migrate | backfill-badges | healthcheck | sql-child`)
process.exit(2)
}

View File

@@ -15,12 +15,18 @@ import { badgeHolderDiff, recalculateBadge } from "../services/problemset"
* 补发用的是 recalculateBadge它同时会**收回**已经不达标的人的奖章,所以只要存在
* 「误发」就先停下来让人看清楚,要真的收回得显式加 --allow-revoke。
*
* bun src/scripts/backfill-problemset-badges.ts
* bun src/scripts/backfill-problemset-badges.ts --apply
* 做成 main.ts 的子命令而不是独立脚本,是因为生产镜像里只有编译好的单二进制,
* 既没有 bun 也没有源码,`bun src/scripts/...` 在那儿根本不存在。跑法对齐 migrate
*
* docker compose -f docker/compose.debian.yml run --rm oj-api oj2-api backfill-badges
* docker compose -f docker/compose.debian.yml run --rm oj-api oj2-api backfill-badges --apply
*
* 本机开发:
*
* bun run --cwd apps/api backfill:badges
* bun apps/api/src/main.ts backfill-badges --apply
*/
const apply = process.argv.includes("--apply")
const allowRevoke = process.argv.includes("--allow-revoke")
export async function backfillProblemSetBadges(options: { apply: boolean; allowRevoke: boolean }) {
const rows = await db
.select({ badge: schema.problemsetBadge, title: schema.problemset.title })
.from(schema.problemsetBadge)
@@ -29,7 +35,7 @@ const rows = await db
if (rows.length === 0) {
console.log("没有任何题单奖章,无事可做")
process.exit(0)
return 0
}
const diffs = []
@@ -56,21 +62,21 @@ console.log(`\n合计漏发 ${missingTotal} 条,误发 ${extraTotal} 条,
if (missingTotal === 0 && extraTotal === 0) {
console.log("奖章与规则一致,无需补发")
process.exit(0)
return 0
}
if (!apply) {
console.log("\n这是只读预演。确认无误后加 --apply 落库。")
process.exit(0)
if (!options.apply) {
console.log("\n这是只读预演,什么都没写。确认无误后加 --apply 落库。")
return 0
}
if (extraTotal > 0 && !allowRevoke) {
if (extraTotal > 0 && !options.allowRevoke) {
console.error(
`\n存在 ${extraTotal} 条误发。补发用的 recalculateBadge 会把它们**删掉**` +
`而 user_badge 没有别处备份、earnedTime 删了就找不回来。\n` +
`确认要连同收回一起执行,加 --allow-revoke。`,
)
process.exit(1)
return 1
}
let touched = 0
@@ -90,4 +96,5 @@ for (const { badge, title } of rows) {
}
}
console.log(remaining === 0 ? "复核通过:全部奖章与规则一致" : `复核未通过,仍有 ${remaining} 条差异`)
process.exit(remaining === 0 ? 0 : 1)
return remaining === 0 ? 0 : 1
}