Calculation/app/middleware/budget_check.js

'use strict';

/**
 *
 *
 * @author Mai
 * @date
 * @version
 */

module.exports = options => {
    /**
     * 标段校验 中间件
     * 1. 读取标段数据（包括属性）
     * 2. 检验用户是否可见标段（不校验具体权限）
     *
     * @param {function} next - 中间件继续执行的方法
     * @return {void}
     */
    return function* budgetCheck(next) {
        try {
            // 读取标段数据
            const id = parseInt(this.params.id);
            if (!id) throw '参数错误';
            this.budget = yield this.service.budget.getDataById(id);
            if (!this.budget) throw '项目不存在';
            yield next;
        } catch (err) {
            this.log(err);
            if (this.helper.isAjax(this.request)) {
                this.ajaxErrorBody(err, '概算投资项目未知错误');
            } else {
                this.postError(err, '概算投资项目未知错误');
            }
        }
    };
};