ConstructionCost/modules/glj/models/glj_list_model.js

/**
 * 项目工料机列表数据模型
 *
 * @author CaiAoLin
 * @date 2017/6/22
 * @version
 */
import BaseModel from "../../common/base/base_model";
import {default as GLJSchemas, collectionName as gljCollectionName} from "./schemas/glj";
import CounterModel from "./counter_model";
import UnitPriceModel from "./unit_price_model";

class GLJListModel extends BaseModel {

    /**
     * 材料类型id
     *
     * @var {Array}
     */
    materialIdList = [5, 6, 7];

    /**
     * 构造函数
     *
     * @return {void}
     */
    constructor() {
        let parent = super();
        parent.model = GLJSchemas;
        parent.init();
    }

    /**
     * 设置场景
     *
     * @param {string} scene
     * @return {void}
     */
    setScene(scene = '') {
        switch (scene) {
            // 新增数据的验证规则
            case 'add':
                this.model.schema.path('glj_repository_id').required(true);
                this.model.schema.path('project_id').required(true);
                this.model.schema.path('tender_id').required(true);
                this.model.schema.path('code').required(true);
                this.model.schema.path('name').required(true);
                this.model.schema.path('type_of_work').required(true);
                break;
        }
    }

    /**
     * 根据标段对应工料机数据
     *
     * @param {Number} tenderId
     * @param {Number} unitPriceTenderId
     * @return {Promise}
     */
    async getListByTenderId(tenderId, unitPriceTenderId) {
        let gljData = null;
        try {
            // 首先获取对应标段下所有的项目工料机数据
            let condition = {tender_id: tenderId};
            let fields = {_id: 0};
            gljData = await this.db.find(condition, fields);

            // 没有数据则直接返回空
            if (gljData.length <= 0) {
                throw '无数据';
            }

            // 获取标段设置的单价文件数据
            let unitPriceModel = new UnitPriceModel();
            let unitPriceList = await unitPriceModel.getDataByTenderId(unitPriceTenderId);

            // 组合数据
            this.combineUnitPrice(gljData, unitPriceList);

        } catch (error) {
            gljData = [];
        }

        return gljData;
    }

    /**
     * 组合工料机数据和单价文件数据
     *
     * @param {object} gljList
     * @param {object} unitPriceList
     * @return {void}
     */
    combineUnitPrice(gljList, unitPriceList) {
        // 循环组合数据
        for(let glj of gljList) {
            if (glj.code === undefined) {
                continue;
            }
            glj.unit_price = unitPriceList[glj.code + glj.name] !== undefined ? unitPriceList[glj.code + glj.name] : null;
            glj.adjust_price = glj.adjustment * glj.unit_price.market_price;
        }
    }

    /**
     * 新增项目工料机数据(包括新增单价文件) 定额工料机新增时调用
     *
     * @param {object} data
     * @return {Promise} 返回插入成功的数据id
     */
    async addList(data) {
        let result = null;
        try {
            if (Object.keys(data).length <= 0) {
                throw '新增数据为空';
            }
            // 首先查找是否有同编码同名称的工料机数据
            let projectGljData = await this.findDataByCondition({code: data.code, tender_id: data.tender_id});
            if (projectGljData && projectGljData.id > 0) {
                // 如果找到数据则直接返回
                return projectGljData;
            }

            // 新增单条记录 （两个操作本来应该是事务操作，然而mongodb事务支持比较弱，就当作是都可以顺利执行）
            let gljInsertData = await this.add(data);
            if (!gljInsertData) {
                throw '新增项目工料机失败!';
            }

            // 获取标段属性中的单价文件关联标段id @todo
            let unitPriceTenderId = 1;

            // 新增单价文件
            let unitPriceModel = new UnitPriceModel();
            let unitPriceInsertData = await unitPriceModel.updateUnitPrice(data, unitPriceTenderId);

            if (!unitPriceInsertData) {
                throw '新增单价失败!';
            }

            gljInsertData.unit_price = unitPriceInsertData;
            result = gljInsertData;
        } catch (error) {
            console.log(error);
            result = null;
        }

        return result;
    }

    /**
     * 新增单条工料机数据
     *
     * @param {object} data
     * @return {Promise}
     */
    async add(data) {
        if (Object.keys(data).length <= 0) {
            throw '新增数据为空';
        }

        let counterModel = new CounterModel();
        data.id = await counterModel.getId(gljCollectionName);

        this.setScene('add');
        let result = await this.db.model.create(data);
        return result;
    }

}

export default GLJListModel;