wistroni40-bnft/lib/api/server.js

Benefit platform parameters publish job template

"use strict";
/**
 * 專案名稱： @wistroni40/bnft
 * 部門代號： ML8100
 * 檔案說明： API服務器
 * @CREATE Mon Feb 08 2021 下午2:56:08
 * @author Steve Y Lin
 * @contact Steve_Y_Lin@wistron.com #1342
 * -----------------------------------------------------------------------------
 * @NOTE
 */
Object.defineProperty(exports, "__esModule", { value: true });
exports.Server = void 0;
const bodyParser = require("body-parser");
const express = require("express");
const fs = require("fs");
const swaggerUi = require("swagger-ui-express");
const logger_1 = require("./../logger");
const swagger_1 = require("./swagger");
/**
 * API服務器
 */
class Server {
    constructor() {
        /**
         * 日誌
         */
        this.logger = new logger_1.Log4js('api');
        /**
         * API Server
         */
        this.server = express();
        /**
         * 效益計算Job
         */
        this.benefit = new Map();
        // 啟動API Server
        this.start();
    }
    /**
     * 取得效益計算Job清單
     *
     * @method private
     * @return 回傳效益計算Job清單
     */
    getBenefitList() {
        return Array.from(this.benefit).map(benefit => benefit[1]);
    }
    /**
     * 取得效益Key名稱
     *
     * @method private
     * @param req 請求
     * @return 回傳效益Key名稱
     */
    getBenefitKey(req) {
        const systemId = req.query.systemId;
        const typeId = req.query.typeId;
        return `${systemId}.${typeId}`;
    }
    /**
     * 重新上拋所有效益
     *
     * @method private
     * @param req 請求
     * @param res 回應
     */
    async sendBenefit(req, res) {
        const timestamp = new Date(parseInt(req.query.timestamp, 10));
        timestamp.setDate(timestamp.getDate() + 1);
        await Promise.all(this.getBenefitList().map(benefit => benefit.execute(timestamp).toPromise()));
        res.send(true);
    }
    /**
     * 重新上拋特定效益
     *
     * @method private
     * @param req 請求
     * @param res 回應
     */
    async sendSpecificBenefit(req, res) {
        const key = this.getBenefitKey(req);
        const timestamp = new Date(parseInt(req.query.timestamp, 10));
        timestamp.setDate(timestamp.getDate() + 1);
        const benefit = this.benefit.get(key);
        if (benefit) {
            await benefit.execute(timestamp).toPromise();
            res.send(true);
        }
        else {
            res.send(null);
        }
    }
    /**
     * 重新上拋特定時間區間內的所有效益
     *
     * @method private
     * @param req 請求
     * @param res 會應
     */
    async sendBenefitByTimeGroup(req, res) {
        if (Array.isArray(req.body) && req.body.length > 0) {
            await Promise.all(req.body.map(async (date) => {
                const timestamp = new Date(parseInt(date, 10));
                timestamp.setDate(timestamp.getDate() + 1);
                await Promise.all(this.getBenefitList().map(benefit => benefit.execute(timestamp).toPromise()));
            }));
            res.send(true);
        }
        else {
            res.send('body format error');
        }
    }
    /**
     * 重新上拋特定時間區間內的特定效益
     *
     * @method private
     * @param req 請求
     * @param res 會應
     */
    async sendSpecificBenefitByTimeGroup(req, res) {
        if (Array.isArray(req.body) && req.body.length > 0) {
            const key = this.getBenefitKey(req);
            await Promise.all(req.body.map(async (date) => {
                const timestamp = new Date(parseInt(date, 10));
                timestamp.setDate(timestamp.getDate() + 1);
                const benefit = await this.benefit.get(key);
                if (benefit) {
                    await benefit.execute(timestamp).toPromise();
                }
            }));
            res.send(true);
        }
        else {
            res.send('body format error');
        }
    }
    /**
     * 重拋上拋失敗的效益
     *
     * @method private
     * @param req 請求
     * @param res 會應
     */
    async sendBackupBenefit(req, res) {
        const files = fs.readdirSync('backup');
        await Promise.all(files.map(file => {
            const benefit = fs.readFileSync(`backup/${file}`, { encoding: 'utf8' });
            try {
                if (this.getBenefitList().length > 0) {
                    this.getBenefitList()[0].send(JSON.parse(benefit));
                    fs.unlinkSync(`backup/${file}`);
                }
            }
            catch (error) {
                this.logger.warn('Send backup benefit failed');
                this.logger.warn(benefit);
            }
            return JSON.parse(benefit);
        }));
        res.send(true);
    }
    /**
     * 啟動API Server
     *
     * @method public
     * @return 回傳物件本身
     */
    start() {
        const port = 3000;
        this.server.use(bodyParser.json());
        this.server.post('/benefit/send', this.sendBenefit.bind(this));
        this.server.post('/specific/benefit/send', this.sendSpecificBenefit.bind(this));
        this.server.post('/benefit/send/timestamps', this.sendBenefitByTimeGroup.bind(this));
        this.server.post('/specific/benefit/send/timestamps', this.sendSpecificBenefitByTimeGroup.bind(this));
        this.server.post('/benefit/backup', this.sendBackupBenefit.bind(this));
        this.server.use('/', swaggerUi.serve, swaggerUi.setup(swagger_1.SWAGGER_DOC));
        this.logger.info(`Listen api port ${port}`);
        this.api = this.server.listen(port);
        return this;
    }
    /**
     * 終止API Server
     *
     * @method public
     * @return 回傳物件本身
     */
    stop() {
        if (this.api) {
            this.api.close();
            this.logger.warn('api close');
        }
        return this;
    }
    /**
     * 註冊效益計算Job
     *
     * @method public
     * @param benefit 效益計算Job
     */
    register(benefit) {
        const key = `${benefit.config.systemId}.${benefit.config.typeId}`;
        this.benefit.set(key, benefit);
    }
}
exports.Server = Server;
/**
 * 單一實例
 */
Server.instance = new Server();