112 lines
2.8 KiB
TypeScript
112 lines
2.8 KiB
TypeScript
|
/**
|
|||
|
|
* 清理卡在30%的课程数据
|
|||
|
|
* 使用方法：ts-node backend/scripts/cleanup-stuck-courses.ts
|
|||
|
|
*/
|
|||
|
|
|
|||
|
|
import prisma from '../src/utils/prisma';
|
|||
|
|
import * as dotenv from 'dotenv';
|
|||
|
|
import path from 'path';
|
|||
|
|
|
|||
|
|
// 加载环境变量
|
|||
|
|
dotenv.config({ path: path.join(__dirname, '../.env') });
|
|||
|
|
|
|||
|
|
async function cleanupStuckCourses() {
|
|||
|
|
try {
|
|||
|
|
console.log('🔍 查找卡在30%的课程...\n');
|
|||
|
|
|
|||
|
|
// 查找状态为generating且progress为0.3的课程
|
|||
|
|
const stuckCourses = await prisma.course.findMany({
|
|||
|
|
where: {
|
|||
|
|
generationStatus: 'generating',
|
|||
|
|
generationProgress: 0.3,
|
|||
|
|
},
|
|||
|
|
include: {
|
|||
|
|
aiContentTask: {
|
|||
|
|
select: {
|
|||
|
|
id: true,
|
|||
|
|
status: true,
|
|||
|
|
suggestedTitle: true,
|
|||
|
|
},
|
|||
|
|
},
|
|||
|
|
},
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
console.log(`找到 ${stuckCourses.length} 个卡住的课程：\n`);
|
|||
|
|
|
|||
|
|
if (stuckCourses.length === 0) {
|
|||
|
|
console.log('✅ 没有需要清理的课程');
|
|||
|
|
await prisma.$disconnect();
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 显示课程信息
|
|||
|
|
stuckCourses.forEach((course, index) => {
|
|||
|
|
console.log(`${index + 1}. ${course.title}`);
|
|||
|
|
console.log(` Course ID: ${course.id}`);
|
|||
|
|
console.log(` Task ID: ${course.aiContentTask?.id || 'N/A'}`);
|
|||
|
|
console.log(` 创建时间: ${course.createdAt}`);
|
|||
|
|
console.log('');
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
// 确认删除
|
|||
|
|
console.log('⚠️ 准备删除以上课程及其关联数据...\n');
|
|||
|
|
|
|||
|
|
let deletedCount = 0;
|
|||
|
|
let errorCount = 0;
|
|||
|
|
|
|||
|
|
for (const course of stuckCourses) {
|
|||
|
|
try {
|
|||
|
|
// 1. 删除 UserCourse 关联
|
|||
|
|
await prisma.userCourse.deleteMany({
|
|||
|
|
where: {
|
|||
|
|
courseId: course.id,
|
|||
|
|
},
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
// 2. 删除 Course 记录
|
|||
|
|
await prisma.course.delete({
|
|||
|
|
where: {
|
|||
|
|
id: course.id,
|
|||
|
|
},
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
// 3. 删除 AIContentTask（如果存在）
|
|||
|
|
if (course.aiContentTaskId) {
|
|||
|
|
await prisma.aIPromptLog.deleteMany({
|
|||
|
|
where: {
|
|||
|
|
taskId: course.aiContentTaskId,
|
|||
|
|
},
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
await prisma.aIContentTask.delete({
|
|||
|
|
where: {
|
|||
|
|
id: course.aiContentTaskId,
|
|||
|
|
},
|
|||
|
|
});
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
deletedCount++;
|
|||
|
|
console.log(`✅ 已删除: ${course.title} (${course.id})`);
|
|||
|
|
} catch (error: any) {
|
|||
|
|
errorCount++;
|
|||
|
|
console.error(`❌ 删除失败: ${course.title} (${course.id}) - ${error.message}`);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
console.log('\n📊 清理结果：');
|
|||
|
|
console.log(` ✅ 成功删除: ${deletedCount} 个课程`);
|
|||
|
|
if (errorCount > 0) {
|
|||
|
|
console.log(` ❌ 删除失败: ${errorCount} 个课程`);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
await prisma.$disconnect();
|
|||
|
|
} catch (error: any) {
|
|||
|
|
console.error('❌ 清理过程出错:', error);
|
|||
|
|
await prisma.$disconnect();
|
|||
|
|
process.exit(1);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 运行清理
|
|||
|
|
cleanupStuckCourses();
|