1.中间键使用 koa-body
npm install koa-body --save
const koaBody = require('koa-body');
app.use(koaBody({
multipart: true,
formidable: {
maxFileSize: 200 * 1024 * 1024 // 设置上传文件大小最大限制,默认2M
}
}));
2.书写路由,croller书写方法
uploadData.js
const errorResult = require('../utils/errorResult.js');
const uploadExcelSrv = require('../service/uploadExcelSrv');
const saveData = async function (ctx, next) {
const getRes = await uploadExcelSrv.getExcelObjs(ctx);
if (getRes.status) {
if (getRes.datas.length > 1) {
errorResult.errorRes(ctx, '暂时不支持多个sheet存在');
} else { //得到的是数组
const objs = getRes.datas[0];
ctx.body = {
status: true,
msg: '上传数据成功'
};
}
} else {
errorResult.errorRes(ctx, getRes.msg);
}
await next();
};
module.exports = {
saveData
};
3.处理excel存储,解析,处理excel用的库是 xlsx
npm install xlsx --save
uploadExcelSrv.js
//接收上传的excel文件,保存解析返回objects
const xlsx = require('xlsx');
const fs = require('fs');
const path = require('path');
const downPath = path.resolve(__dirname, '../../fileUpload');
async function getExcelObjs (ctx) {
const file = ctx.request.files.file; // 获取上传文件
const reader = fs.createReadStream(file.path); // 创建可读流
const ext = file.name.split('.').pop(); // 获取上传文件扩展名
const filePath = `${downPath}/${Math.random().toString()}.${ext}`;
const upStream = fs.createWriteStream(filePath); // 创建可写流
const getRes = await getFile(reader, upStream); //等待数据存储完成
const datas = []; //可能存在多个sheet的情况
if (!getRes) { //没有问题
const workbook = xlsx.readFile(filePath);
const sheetNames = workbook.SheetNames; // 返回 ['sheet1', ...]
for (const sheetName of sheetNames) {
const worksheet = workbook.Sheets[sheetName];
const data = xlsx.utils.sheet_to_json(worksheet);
datas.push(data);
}
return {
status: true,
datas
};
} else {
return {
status: false,
msg: '上传文件错误'
};
}
}
function getFile (reader, upStream) {
return new Promise(function (result) {
let stream = reader.pipe(upStream); // 可读流通过管道写入可写流
stream.on('finish', function (err) {
result(err);
});
});
}
module.exports = {
getExcelObjs
};