Koa2之解析在线excel文件

涉及Npm模块

• formidable -- 用于接收http请求中上传的文件
• node-xlsx  -- 用于解析excel文件
• fs --操作文件

思路

通过formiadble将获取前端通过http请求上传的excel文件，并将文件保存至本地（因为node-xlsx无法通过虚拟地址解析文件）。然后通过node-xlsx解析并获取excel文件中的数据。将数据获取完后，删除本地excel文件。

代码

文件操作

const uploadExcel = async ctx => {
	let curPath ,tempPath;
	let form = new formidable.IncomingForm();

	form.maxFileSize = 100 * 1024 * 1024; // 文件最大值
	form.keepExtensions = true; // 路径是否显示文件扩展名

	await new Promise(resolve =>{
		// 文件上传处理
		form.parse(ctx.req,function (err,fields,files) {
			tempPath = form.openedFiles[0].path; // 获取文件的虚拟地址

			// 发生错误直接返回
			if(form.error!=null) {
				fs.unlinkSync(tempPath);
				ctx.body = Result.ErrResult(0,form.error.toString());
				return resolve();
			}

			// 判断多个文件，返回
			if(form.openedFiles.length>1){
				ctx.body = Result.ErrResult(0,"请上传单个文件");
				form.openedFiles.forEach(item=>{
					console.log(item.path);
					fs.unlinkSync(item.path);
				});
				return resolve();
			}

			let readStream=fs.createReadStream(tempPath);

			// 生成新的文件名
			let fileName = new Date().valueOf(),
				fileExt = FileUtil.getUploadFileExt(tempPath);

			if(fileExt==undefined||fileExt==""||(fileExt!="xls"&&fileExt!="xlsx")){
				fs.unlinkSync(tempPath);// 异步删除虚拟文件
				ctx.body = Result.ErrResult(0,"文件类型有误");
				return resolve("抛出异常");
			}

			// 无论是什么格式都转换成xlsx格式进行保存,因为node-xlsx只能解析xlsx文件
			curPath = DirPath+fileName+".xlsx";

            // 将文件写入保存至本地
			let writeStream=fs.createWriteStream(curPath);
			readStream.pipe(writeStream);

			// 文件写入完成监听
			readStream.on('end',async function(){
				fs.unlinkSync(tempPath);// 异步删除虚拟文件
				await fileImp.importExcel(curPath);
				fs.unlinkSync(curPath);// 导入结束后,异步删除本地文件
				ctx.body = Result.SuccessResult("导入成功");
				resolve("完成");
			});
		});
	});
};

 

解析excel

const importExcel = async filePath =>{
	let workbook = xlxs.parse(filePath), // 获取excel文件对象
		sheet = workbook[0].data; // 获取第一个sheet的数据

	let importSingleObj ={};

	for(let i = 1,len = sheet.length ;i<len;i++){
		importSingleObj = {};
		importSingleObj.name = sheet[i][0];
		importSingleObj.password = "12311";
		importSingleObj.age = 24;
		importSingleObj.address = sheet[i][3];
		await Mysql.transaction(async t =>{
			await userMapper.AddUser(importSingleObj,t);
		});
	}
};

这个方法可能不是最优的，所以我会继续研究是否有更加好的方法来实现excel导入功能。