add-resume.js 754 B

1234567891011121314151617181920212223
  1. const fs = require('fs');
  2. const file = 'src/app.ts';
  3. let content = fs.readFileSync(file, 'utf8');
  4. // 1. 修复重复导入
  5. content = content.replace(/import \{ initBookGenerationQueue \} from.*\n/g, '');
  6. content = content.replace(
  7. "import { queueService } from './services/queue.service';",
  8. "import { queueService } from './services/queue.service';\nimport { initBookGenerationQueue, resumeInterruptedTasks } from './modules/book-generator/book-queue.processor';"
  9. );
  10. // 2. 添加恢复调用
  11. content = content.replace(
  12. /initBookGenerationQueue\(\);/,
  13. `initBookGenerationQueue();
  14. // 自动恢复中断的任务
  15. resumeInterruptedTasks();`
  16. );
  17. fs.writeFileSync(file, 'utf8', content);
  18. console.log('✅ 已添加自动恢复功能');