xiaoshu-app/scripts/lesson-export-worker.mjs

#!/usr/bin/env node
import {readFileSync} from 'node:fs';
import {setPriority} from 'node:os';
import {pathToFileURL} from 'node:url';
import * as XLSX from 'xlsx';
import {lessonExportColumns} from '../projects/shared/lesson-export-columns.mjs';
export async function processLessonExport(call){
 const job=await call('claim',{});if(!job.jobId)return false;
 const identity={jobId:job.jobId,lease:job.lease};
 try{
  if(!job.snapshotReady)await call('snapshot',identity);
  const sheet=XLSX.utils.json_to_sheet([]);let offset=0,total=Infinity;
  while(offset<total){const page=await call('chunk',{...identity,offset});total=page.total;if(!page.items.length&&offset<total)throw Error('incomplete_snapshot');
   XLSX.utils.sheet_add_json(sheet,page.items.map(lessonExportColumns),{origin:offset===0?0:-1,skipHeader:offset!==0});offset+=page.items.length;
  }
  const book=XLSX.utils.book_new();XLSX.utils.book_append_sheet(book,sheet,'训练记录');
  const base64=XLSX.write(book,{type:'base64',bookType:'xlsx',compression:true});
  await call('finish',{...identity,rowCount:offset,base64});return true;
 }catch(e){await call(e?.transient?'release':'fail',identity).catch(()=>{});throw Object.assign(Error('export_job_failed'),{transient:e?.transient===true});}
}
async function main(){
 try{setPriority(0,10);}catch{} // Background XLSX compression yields CPU to interactive requests.
 const config=process.env.XIAOSHU_EXPORT_ENV_FILE||'/Users/mac/.config/xiaoshu/legacy-sync-nightly.env',env={...process.env};
 for(const line of readFileSync(config,'utf8').split('\n')){const m=line.match(/^(?:export\s+)?(XIAOSHU_[A-Z0-9_]+)=(.*)$/);if(m&&!env[m[1]])env[m[1]]=m[2].replace(/^['"]|['"]$/g,'');}
 const token=env.XIAOSHU_EXPORT_OPERATOR_TOKEN||env.XIAOSHU_SYNC_OPERATOR_TOKEN;if(!token)throw Error('export_operator_missing');
 const base=env.XIAOSHU_EXPORT_GATEWAY_URL||'https://server.xiaoshu.pro/api/functions/xiaoshu/ops/gateway-v3';
 const call=async(action,params)=>{let r;try{r=await fetch(base,{method:'POST',headers:{'Content-Type':'application/json'},body:JSON.stringify({token,params:{operation:'ops/lessons/export/'+action,...params}}),signal:AbortSignal.timeout(action==='snapshot'?90_000:20_000)});}catch{throw Object.assign(Error('export_transport'),{transient:true});}let data;try{data=await r.json();}catch{throw Object.assign(Error('export_gateway_'+r.status),{transient:true});}if(!r.ok||!data.success)throw Object.assign(Error('export_gateway_'+r.status),{transient:[429,500,502,503,504].includes(r.status)});return data.data;};
 let lastMaintenance=0;let stopped=false;for(const event of ['SIGINT','SIGTERM'])process.on(event,()=>{stopped=true;});
 do{try{if(Date.now()-lastMaintenance>86_400_000){await call('maintenance',{});lastMaintenance=Date.now();}await processLessonExport(call);}catch{console.error('导出任务失败，错误状态已回写或将由租约恢复');}if(process.argv.includes('--once'))break;if(!stopped)await new Promise(r=>setTimeout(r,5000));}while(!stopped);
}
if(process.argv[1]&&import.meta.url===pathToFileURL(process.argv[1]).href)main().catch(()=>{console.error('导出 worker 启动失败，请检查私密配置和权限');process.exitCode=1;});