ysjj-system/app/controller/base.js

'use strict';

const Controller 		= require('egg').Controller;
const path 		 		= require('path'); 
const awaitWriteStream 	= require('await-stream-ready').write;
const sendToWormhole 	= require('stream-wormhole');//管道读入一个虫洞
const fs 				= require('fs');
/**
 * 基类控制器
 */
class BaseController extends Controller {

	/**
	 * [uploadOne 单文件上传功能]
	 * @author 	   szjcomo
	 * @createTime 2020-08-14
	 * @param      {[type]}   savePath [description]
	 * @param      {[type]}   filename [description]
	 * @return     {[type]}            [description]
	 */
	async _uploadOne(savePath,filename) {
		let that = this;
		filename = filename || (that.app.szjcomo.date('YmdHis') + '' + that.app.szjcomo.mt_rand(100000,999999));
		return await that._uploadHandler(await that.ctx.getFileStream(),savePath,filename);
	}
	/**
	 * [uploadAll 批量上传文件功能]
	 * @author 	   szjcomo
	 * @createTime 2020-08-14
	 * @param      {[type]}   savePath [description]
	 * @return     {[type]}            [description]
	 */
	async _uploadAll(savePath) {
		let that = this;
		let part;
		let result = [];
		let fields = {};
		let parts = await that.ctx.multipart();
		while ((part = await parts()) != null) {
			if (part.length) {
				fields[part[0]] = part[1];
			} else {
				// 定义文件名
  				let filename 	= (that.app.szjcomo.date('YmdHis') + '' + that.app.szjcomo.mt_rand(100000,999999));
				// part 是上传的文件流
				result.push(await that._uploadHandler(part,savePath,filename));
			}
		}
		return result;
	}
	/**
	 * [uploadHandler 上传的文件流]
	 * @Author   szjcomo
	 * @DateTime 2019-10-05
	 * @param    {[type]}   stream [description]
	 * @return   {[type]}          [description]
	 */
	async _uploadHandler(stream,savePath,filename){
		let that = this;
	  		let ext 		= path.extname(filename);
	  		if(that.app.szjcomo.empty(ext)) filename += path.extname(stream.filename);
	  		// 生成文件路径
	  		that.app.szjcomo.mkdir(savePath);
	  		// 目标文件
	  		let target 		= path.join(savePath, filename);
	  		// 创建文件流
	  		let writeStream = fs.createWriteStream(target);
	  		try{
	  			//异步把文件流 写入
    			await awaitWriteStream(stream.pipe(writeStream));
	  		} catch(err){
			    //如果出现错误，关闭管道
			    await sendToWormhole(stream);
	  			throw err;
	  		}
	  		await sendToWormhole(stream);
	  		return {save_path:savePath,save_name:filename,file_sha1:that.app.szjcomo.FileSHA1(target),
	  			fields:stream.fields,file_size:writeStream.bytesWritten,mime:stream.mime,encoding:stream.encoding,
	  			origin_name:stream.filename,file_md5:that.app.szjcomo.FileMD5(target),ext:path.extname(stream.filename)}
	}


	/**
	 * [appFileLogs 文件记录位置]
	 * @param  {[type]} fileName [description]
	 * @param  {[type]} err      [description]
	 * @return {[type]}          [description]
	 */
	async logs(fileName,err) {
		let that = this;
		let filePath = `${that.app.baseDir}/logs/${that.app.szjcomo.date('Y-m-d')}`;
		fileName = fileName + '.txt';
		await that.app.szjcomo.mkdir(filePath);
		let isFile = await that.service.base.fileExists(path.join(filePath,fileName));
		let filedata = `=========${that.app.szjcomo.date('Y-m-d H:i:s')}==========\r\n${err.stack}\r\n=========${that.app.szjcomo.date('Y-m-d H:i:s')}==========\r\n\r\n`;
		if(isFile) {
			fs.appendFileSync(path.join(filePath,fileName),filedata);
		} else {
			fs.writeFileSync(path.join(filePath,fileName), filedata);
		}
	}
}
module.exports = BaseController;