73 lines
2.8 KiB
JavaScript
73 lines
2.8 KiB
JavaScript
import { PrismaClient } from "@prisma/client";
|
|
import { PrismaMariaDb } from "@prisma/adapter-mariadb";
|
|
import "dotenv/config";
|
|
|
|
const base = process.env.DATABASE_URL.replace("mysql://", "mariadb://");
|
|
const sep = base.includes("?") ? "&" : "?";
|
|
const connectionString = `${base}${sep}connection_limit=5&pool_timeout=10`;
|
|
const adapter = new PrismaMariaDb(connectionString);
|
|
const prisma = new PrismaClient({ adapter });
|
|
|
|
async function main() {
|
|
console.log("=== 修复 PendingTool categoryId 为 NULL 的记录 ===\n");
|
|
|
|
const categories = await prisma.category.findMany({ select: { id: true, name: true }, orderBy: { id: "asc" } });
|
|
if (categories.length === 0) {
|
|
console.error("错误: 数据库中没有分类数据");
|
|
await prisma.$disconnect();
|
|
return;
|
|
}
|
|
const defaultCategoryId = categories[0].id;
|
|
console.log(`默认分类: id=${defaultCategoryId} "${categories[0].name}"`);
|
|
|
|
const nullCatTools = await prisma.pendingTool.findMany({
|
|
where: { categoryId: null, status: "pending" },
|
|
select: { id: true, name: true, websiteUrl: true },
|
|
});
|
|
|
|
const catIds = new Set(categories.map(c => c.id));
|
|
const invalidCatTools = await prisma.pendingTool.findMany({
|
|
where: { status: "pending", categoryId: { not: null } },
|
|
select: { id: true, name: true, categoryId: true },
|
|
});
|
|
const invalidCat = invalidCatTools.filter(t => !catIds.has(t.categoryId));
|
|
|
|
console.log(`\nPendingTool 统计:`);
|
|
console.log(` categoryId 为 NULL: ${nullCatTools.length} 条`);
|
|
console.log(` categoryId 指向无效分类: ${invalidCat.length} 条`);
|
|
|
|
let totalFixed = 0;
|
|
|
|
if (nullCatTools.length > 0) {
|
|
console.log(`\n修复 categoryId 为 NULL 的记录:`);
|
|
for (const t of nullCatTools) {
|
|
console.log(` id=${t.id} "${t.name}" → categoryId: ${defaultCategoryId}`);
|
|
await prisma.pendingTool.update({ where: { id: t.id }, data: { categoryId: defaultCategoryId } });
|
|
totalFixed++;
|
|
}
|
|
}
|
|
|
|
if (invalidCat.length > 0) {
|
|
console.log(`\n修复 categoryId 指向无效分类的记录:`);
|
|
for (const t of invalidCat) {
|
|
console.log(` id=${t.id} "${t.name}" (categoryId=${t.categoryId}) → ${defaultCategoryId}`);
|
|
await prisma.pendingTool.update({ where: { id: t.id }, data: { categoryId: defaultCategoryId } });
|
|
totalFixed++;
|
|
}
|
|
}
|
|
|
|
if (totalFixed === 0) {
|
|
console.log("\n无需修复，所有 PendingTool 记录的 categoryId 均有效。");
|
|
} else {
|
|
console.log(`\n✅ 共修复 ${totalFixed} 条记录`);
|
|
}
|
|
|
|
console.log("\n建议后续操作:");
|
|
console.log(" 1. 重新部署 task3-check-tools.mjs 到服务器 scripts/ 目录");
|
|
console.log(" 2. 重启 CRON 容器: docker compose restart cron");
|
|
console.log(" 3. 验证: 等待下次 task3 执行确认无 FK 错误");
|
|
}
|
|
|
|
main()
|
|
.catch((e) => { console.error(e); process.exit(1); })
|
|
.finally(async () => { await prisma.$disconnect(); }); |