qinglong/back/loaders/initTask.ts

import { Container } from 'typedi';
import SystemService from '../services/system';
import ScheduleService from '../services/schedule';
import SubscriptionService from '../services/subscription';
import config from '../config';
import { fileExist } from '../config/util';

export default async () => {
  const systemService = Container.get(SystemService);
  const scheduleService = Container.get(ScheduleService);
  const subscriptionService = Container.get(SubscriptionService);

  // 生成内置token
  let tokenCommand = `ts-node-transpile-only ${config.rootPath}/back/token.ts`;
  const tokenFile = `${config.rootPath}static/build/token.js`;
  if (await fileExist(tokenFile)) {
    tokenCommand = `node ${tokenFile}`;
  }
  const cron = {
    id: NaN,
    name: '生成token',
    command: tokenCommand,
  };
  scheduleService.createIntervalTask(cron, {
    days: 28,
  });

  // 运行删除日志任务
  const data = await systemService.getLogRemoveFrequency();
  if (data && data.info && data.info.frequency) {
    const cron = {
      id: data.id,
      name: '删除日志',
      command: `ql rmlog ${data.info.frequency}`,
    };
    scheduleService.createIntervalTask(cron, {
      days: data.info.frequency,
    });
  }

  // 运行所有订阅
  const subs = await subscriptionService.list();
  for (const sub of subs) {
    await subscriptionService.handleTask(
      sub,
      !sub.is_disabled,
      true,
      !sub.is_disabled,
    );
  }
};