@eggjs/schedule/src/app/extend/application.unittest.ts

schedule plugin for egg, support corn job.

import { debuglog } from 'node:util';
import path from 'node:path';
import { importResolve } from '@eggjs/utils';
import type { ScheduleWorker } from '../../lib/schedule_worker.js';
import type { EggScheduleItem } from '../../lib/types.js';

const debug = debuglog('@eggjs/schedule/app');

export default {
  async runSchedule(schedulePath: string, ...args: any[]) {
    debug('[runSchedule] start schedulePath: %o, args: %o', schedulePath, args);
    // for test purpose
    const config = this.config;
    const directory = [
      path.join(config.baseDir, 'app/schedule'),
      ...config.schedule.directory,
    ];

    // resolve real path
    if (path.isAbsolute(schedulePath)) {
      schedulePath = importResolve(schedulePath);
    } else {
      for (const dir of directory) {
        const trySchedulePath = path.join(dir, schedulePath);
        try {
          schedulePath = importResolve(trySchedulePath);
          break;
        } catch (err) {
          debug('[runSchedule] importResolve %o error: %s', trySchedulePath, err);
        }
      }
    }

    debug('[runSchedule] resolve schedulePath: %o', schedulePath);
    const scheduleWorker: ScheduleWorker = this.scheduleWorker;
    let schedule: EggScheduleItem;
    try {
      schedule = scheduleWorker.scheduleItems[schedulePath];
      if (!schedule) {
        throw new TypeError(`Cannot find schedule ${schedulePath}`);
      }
    } catch (err: any) {
      err.message = `[@eggjs/schedule] ${err.message}`;
      throw err;
    }

    // run with anonymous context
    const ctx = this.createAnonymousContext({
      method: 'SCHEDULE',
      url: `/__schedule?path=${schedulePath}&${schedule.scheduleQueryString}`,
    });
    return await this.ctxStorage.run(ctx, async () => {
      return await schedule.task(ctx, ...args);
    });
  },
} as any;