ConstructionCost/modules/glj/models/unit_price_model.js

/**
 * 单价文件业务模型
 *
 * @author CaiAoLin
 * @date 2017/6/30
 * @version
 */
import BaseModel from "../../common/base/base_model";
import UnitPriceSchema from "./schemas/unit_price";

class UnitPriceModel extends BaseModel {

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

    /**
     * 根据标段获取对应单价数据
     *
     * @param {Number} tenderId
     * @return {Promise}
     */
    async getDataByTenderId(tenderId) {
        tenderId = parseInt(tenderId);
        if (isNaN(tenderId) || tenderId <= 0) {
            return null;
        }

        let unitPriceList = await this.db.model.find({tender_id: tenderId});
        if (unitPriceList.length <= 0) {
            return null;
        }

        // 整理数据
        let result = {};
        for(let tmp of unitPriceList) {
            result[tmp.code] = tmp;
        }

        return result;
    }

}

export default UnitPriceModel;