Calculation/app/service/material_file.js

'use strict';
const archiver = require('archiver');
const path = require('path');
const fs = require('fs');
/**
 * 附件表 数据模型
 * @author LanJianRong
 * @date 2020/6/30
 * @version
 */

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

        /**
         * 获取当前标段(期)所有上传的附件
         * @param {Number} tid 标段id
         * @param {Number?} mid 期id
         * @return {Promise<void>} 数据库查询实例
         */
        async getAllMaterialFiles(tid, mid) {
            const { ctx } = this;
            const where = { tid };
            if (mid) where.mid = mid;
            const result = await this.db.select(this.tableName, {
                where,
                orders: [['upload_time', 'desc']],
            });
            return result.map(item => {
                if (!ctx.helper.canPreview(item.fileext)) {
                    item.filepath = `tender/${ctx.tender.id}/measure/material/${item.s_order}/file/${item.id}/download`;
                }
                return item;
            });
        }


        /**
         * 存储上传的文件信息至数据库
         * @param {Array} payload 载荷
         * @return {Promise<void>} 数据库插入执行实例
         */
        async saveFileMsgToDb(payload) {
            return await this.db.insert(this.tableName, payload);
        }

        /**
         * 获取单个文件信息
         * @param {Number} id 文件id
         * @return {Promise<void>} 数据库查询实例
         */
        async getMaterialFileById(id) {
            return await this.getDataByCondition({ id });
        }

        /**
         * 删除附件
         * @param {Number} id - 附件id
         * @return {void}
         */
        async delete(id) {
            return await this.deleteById(id);
        }

        /**
         * 将文件压缩成zip，并返回zip文件的路径
         * @param {array} fileIds - 文件数组id
         * @param {string} zipPath - 压缩文件存储路径
         * @return {string} 压缩后的zip文件路径
         */
        async compressedFile(fileIds, zipPath) {
            this.initSqlBuilder();
            this.sqlBuilder.setAndWhere('id', {
                value: fileIds,
                operate: 'in',
            });
            const [sql, sqlParam] = this.sqlBuilder.build(this.tableName);
            const files = await this.db.query(sql, sqlParam);
            // const paths = files.map(item => {
            //     return { name: item.filename + item.fileext, path: item.filepath }
            // })
            return new Promise((resolve, reject) => {
                // 每次开一个新的archiver
                const ziparchiver = archiver('zip');
                const outputPath = fs.createWriteStream(path.resolve(this.app.baseDir, zipPath));
                outputPath.on('error', err => {
                    reject(err);
                });

                ziparchiver.pipe(outputPath);
                files.forEach(item => {
                    ziparchiver.file(path.resolve(this.app.baseDir, 'app', item.filepath), { name: item.file_name });
                });

                // 存档警告
                ziparchiver.on('warning', function(err) {
                    if (err.code === 'ENOENT') {
                        console.warn('stat故障和其他非阻塞错误');
                    }
                    reject(err);
                });

                // 存档出错
                ziparchiver.on('error', function(err) {
                    console.log(err);
                    reject(err);
                });
                ziparchiver.finalize();
                outputPath.on('close', () => {
                    resolve(ziparchiver.pointer());
                });
            });
        }
    }
    return MaterialFile;
};