"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.Executor = void 0;
const JobManager = require('./job-manager');
const logger = require('./logger');
const log = logger('xxl-job-executor');
const axios_1 = require("axios");
class Executor {
    executorKey;
    scheduleCenterUrl;
    accessToken;
    executorUrl;
    jobHandlers;
    jobManager;
    /**
     * @param {string} executorKey
     * @param {string} scheduleCenterUrl
     * @param {string} accessToken
     * @param {string} jobLogPath
     * @param {Map} jobHandlers
     * @param {*} context
     */
    constructor(executorKey, scheduleCenterUrl, accessToken, jobLogPath, jobHandlers, context) {
        this.executorKey = executorKey;
        this.scheduleCenterUrl = scheduleCenterUrl;
        this.accessToken = accessToken;
        this.jobHandlers = jobHandlers;
        this.jobManager = new JobManager(jobLogPath, context);
    }
    /**
     * 应用执行器中间件
     * @param {*} app
     * @param {string} app_domain
     * @param {string} uri
     */
    applyMiddleware({ app, app_domain, uri }) {
        const express = require('express');
        var router = new express.Router();
        this.initRouter(router, uri);
        this.addJobRoutes(router, uri);
        app.use(router);
        this.executorUrl = (app_domain || 'http://192.168.31.251:8088') + uri;
    }
    /**
     * 添加xxl-job相关的路由，供调度中心访问
     * @param {express.Router} router
     * @param {string} baseUri
     */
    initRouter(router, uri) {
        router.use(uri, async (req, res, next) => {
            res.status(200);
            const token = req.headers && req.headers['xxl-job-access-token'];
            if (!!this.accessToken && this.accessToken !== token) {
                res.send({ code: 500, msg: 'access token incorrect' });
                return;
            }
            if (!req.body) {
                res.send({ code: 500, msg: 'body is null' });
                return;
            }
            next();
        });
    }
    /**
     * 添加xxl-job相关的路由，供调度中心访问
     * @param {express.Router} router
     * @param {string} baseUri
     */
    addJobRoutes(router, baseUri) {
        router.post(`${baseUri}/beat`, async (req, res, next) => {
            res.send(this.beat());
        });
        router.post(`${baseUri}/idleBeat`, async (req, res, next) => {
            res.send(this.idleBeat(req.body.jobId || -1));
        });
        router.post(`${baseUri}/run`, async (req, res, next) => {
            res.send(this.run(req.body || {}));
        });
        router.post(`${baseUri}/kill`, async (req, res, next) => {
            res.send(this.killJob(req.body.jobId || -1));
        });
        router.post(`${baseUri}/log`, async (req, res, next) => {
            const { logDateTim: logDateTime, logId, fromLineNum } = req.body || {};
            const data = await this.readLog(logDateTime, logId, fromLineNum);
            res.send(data);
        });
    }
    /**
     * 心跳检测：调度中心检测执行器是否在线时使用
     * @return {{code: number, msg: string}}
     */
    beat() {
        return { code: 200, msg: 'success' };
    }
    /**
     * 忙碌检测：调度中心检测指定执行器上指定任务是否忙碌（运行中）时使用
     * @param {string} jobId - 任务ID
     * @return {{code: number, msg: string}}
     */
    idleBeat(jobId) {
        return (this.jobManager.hasJob(jobId) ? { code: 500, msg: 'busy' } : { code: 200, msg: 'idle' });
    }
    /**
     * 触发任务执行
     * @param {number} jobId - 任务ID
     * @param {string} handlerName - 任务的handler名字
     * @param {string} jobJsonParams - 任务参数
     * @param {number} executorTimeout - 任务超时时间，单位秒，大于零时生效
     * @param {number} logId - 本次调度日志ID
     * @param {number} - 本次调度日志时间
     * @return {{code: number, msg: string}}
     */
    run({ jobId, executorHandler: handlerName, executorParams: jobJsonParams, executorTimeout, logId, logDateTime }) {
        const jobHandler = this.jobHandlers.get(handlerName);
        if (!jobHandler) {
            return { code: 500, msg: `no matched jobHandler(${handlerName})` };
        }
        this.jobManager.runJob(jobId, jobJsonParams, logId, logDateTime, executorTimeout, handlerName, jobHandler, this.callback.bind(this));
        return { code: 200, msg: 'success' };
    }
    /**
     * 终止任务
     * @param {number} jobId - 任务ID
     * @return {{code: number, msg: string}}
     */
    killJob(jobId) {
        return { code: 500, msg: `not yet support, jobId(${jobId})` };
    }
    /**
     * 查看执行日志
     * @param {number} logDateTime - 本次调度日志时间
     * @param {number} logId - 本次调度日志ID
     * @param {number} fromLineNum - 日志开始行号
     * @return {*} - fromLineNum:日志开始行号; toLineNum:日志结束行号; logContent:日志内容
     */
    async readLog(logDateTime, logId, fromLineNum) {
        let logContent;
        let toLineNum;
        try {
            const lines = await this.jobManager.readJobLog(logDateTime, logId);
            lines.splice(0, fromLineNum - 1);
            if (last(lines) === '')
                lines.pop();
            toLineNum = fromLineNum + lines.length - 1;
            lines.unshift('');
            logContent = lines.join('\n');
        }
        catch (err) {
            log.err('readLog error: %o', err.message);
            toLineNum = fromLineNum;
            logContent = err.toString();
        }
        return { code: 200, content: { fromLineNum, toLineNum, logContent } };
    }
    /**
     * 执行器注册：执行器注册时使用，调度中心会实时感知注册成功的执行器并发起任务调度
     */
    async registry() {
        return axios_1.default.post(`${this.scheduleCenterUrl}/api/registry`, {
            registryGroup: 'EXECUTOR',
            registryKey: this.executorKey,
            registryValue: this.executorUrl
        }, {
            headers: {
                'XXL-JOB-ACCESS-TOKEN': this.accessToken,
                'Content-Type': 'application/json'
            }
        }).then((response) => console.log('执行器注册成功:', response.data)).catch(error => console.error('执行器注册失败:', error.message));
    }
    /**
     * 执行器注册摘除：执行器注册摘除时使用，注册摘除后的执行器不参与任务调度与执行
     */
    async registryRemove() {
        return axios_1.default.post(`${this.scheduleCenterUrl}/api/registryRemove`, {
            'registryGroup': 'EXECUTOR',
            'registryKey': this.executorKey,
            'registryValue': this.executorUrl
        }, {
            headers: {
                'XXL-JOB-ACCESS-TOKEN': this.accessToken,
                'Content-Type': 'application/json'
            }
        }).then((response) => console.log('执行器移除成功:', response.data)).catch(error => console.error('执行器移除失败:', error.message));
    }
    /**
   * 任务回调：执行器执行完任务后，回调任务结果时使用
   * @param {Error|null} error 错误对象
   * @param {{logId: number, result: any}} jobResult 任务结果
   */
    async callback(error, { logId, result }) {
        return axios_1.default.post(`${this.scheduleCenterUrl}/api/callback`, [{
                logId,
                logDateTim: Date.now(),
                handleCode: error ? 500 : 200,
                handleMsg: error ? error.message : (result ? JSON.stringify(result) : 'success')
            }], {
            headers: {
                'XXL-JOB-ACCESS-TOKEN': this.accessToken,
                'Content-Type': 'application/json'
            }
        }).then((response) => response.data).catch((err) => {
            log.error(`callback error:${JSON.stringify(result)} ${err.message}`);
        });
    }
}
exports.Executor = Executor;
//# sourceMappingURL=executor.js.map