ysjj-system/node_modules/egg-schedule/app.js

'use strict';

const loadSchedule = require('./lib/load_schedule');
const qs = require('querystring');
const path = require('path');
const is = require('is-type-of');

module.exports = app => {
  const logger = app.getLogger('scheduleLogger');
  const schedules = loadSchedule(app);

  // log schedule list
  for (const s in schedules) {
    const schedule = schedules[s];
    if (!schedule.schedule.disable) logger.info('[egg-schedule]: register schedule %s', schedule.key);
  }

  // register schedule event
  app.messenger.on('egg-schedule', info => {
    const { id, key } = info;
    const schedule = schedules[key];

    logger.debug(`[Job#${id}] ${key} task received by app`);

    if (!schedule) {
      logger.warn(`[Job#${id}] ${key} unknown task`);
      return;
    }

    /* istanbul ignore next */
    if (schedule.schedule.disable) {
      logger.warn(`[Job#${id}] ${key} disable`);
      return;
    }

    logger.info(`[Job#${id}] ${key} executing by app`);

    // run with anonymous context
    const ctx = app.createAnonymousContext({
      method: 'SCHEDULE',
      url: `/__schedule?path=${key}&${qs.stringify(schedule.schedule)}`,
    });

    const start = Date.now();

    // execute
    return schedule.task(ctx, ...info.args)
      .catch(err => {
        return is.error(err) ? err : new Error(err);
      })
      .then(err => {
        const success = !is.error(err);
        const rt = Date.now() - start;

        const msg = `[Job#${id}] ${key} execute ${success ? 'succeed' : 'failed'}, used ${rt}ms.`;
        logger[success ? 'info' : 'error'](msg, success ? '' : err);

        Object.assign(info, {
          success,
          workerId: process.pid,
          rt,
          message: err && err.message,
        });

        // notify agent job finish
        app.messenger.sendToAgent('egg-schedule', info);
      });
  });

  // for test purpose
  const directory = [].concat(path.join(app.config.baseDir, 'app/schedule'), app.config.schedule.directory || []);
  app.runSchedule = (schedulePath, ...args) => {
    // resolve real path
    if (path.isAbsolute(schedulePath)) {
      schedulePath = require.resolve(schedulePath);
    } else {
      for (const dir of directory) {
        try {
          schedulePath = require.resolve(path.join(dir, schedulePath));
          break;
        } catch (_) {
          /* istanbul ignore next */
        }
      }
    }

    let schedule;

    try {
      schedule = schedules[schedulePath];
      if (!schedule) {
        throw new Error(`Cannot find schedule ${schedulePath}`);
      }
    } catch (err) {
      err.message = `[egg-schedule] ${err.message}`;
      return Promise.reject(err);
    }

    // run with anonymous context
    const ctx = app.createAnonymousContext({
      method: 'SCHEDULE',
      url: `/__schedule?path=${schedulePath}&${qs.stringify(schedule.schedule)}`,
    });

    return schedule.task(ctx, ...args);
  };
};