Calculation/app/service/change_company.js

'use strict';

/**
 *
 *
 * @author Mai
 * @date 2018/8/14
 * @version
 */

const audit = require('../const/audit');

module.exports = app => {
    class ChangeCompany extends app.BaseService {
        /**
         * 构造函数
         *
         * @param {Object} ctx - egg全局变量
         * @return {void}
         */
        constructor(ctx) {
            super(ctx);
            this.tableName = 'change_company';
        }

        /**
         * 队列修改和添加公司信息
         * @param tenderId   标段id
         * @param updateIdArray  更新的公司id
         * @param updateArray   更新的公司名称
         * @param addArray      添加的公司名称
         * @return {Promise.<[*,*]>}  返回新增的公司列表，和替换下拉选择公司列表
         */
        async setCompanyList(data) {
            const tenderId = parseInt(data.tid);
            const updateIdArray = data.uci;
            const updateArray = data.uc;
            const addArray = data.ac;
            let addCompany = new Array();
            let selectCompany = new Array();
            // 初始化事务
            this.transaction = await this.db.beginTransaction();
            try {
                // 更新公司信息
                if (updateIdArray.length !== 0) {
                    for (let index in updateIdArray) {
                        const updateData = {
                            id: updateIdArray[index],
                            name: updateArray[index],
                        };
                        await this.transaction.update(this.tableName, updateData);
                        selectCompany.push(updateData);
                    }
                }

                if (addArray.length !== 0) {
                    for (let name of addArray) {
                        const addData = {
                            tid: tenderId,
                            name: name,
                        };
                        const operate = await this.transaction.insert(this.tableName, addData);
                        if (operate.affectedRows !== 1) {
                            throw '提交数据失败';
                        } else {
                            selectCompany.push({id: operate.insertId, name: name});
                            addCompany.push({id: operate.insertId, name: name});
                        }
                    }
                }
                this.transaction.commit();
            } catch (error) {
                console.log(error);
                // 回滚
                await this.transaction.rollback();
            }
            return [addCompany, selectCompany];
        }
    }

    return ChangeCompany;
};