JavaScript

超轻量级php框架startmvc

koa上传excel文件并解析的实现方法

更新时间:2020-07-20 06:36:01 作者:startmvc
1.中间键使用koa-bodynpminstallkoa-body--saveconstkoaBody=require('koa-body');app.use(koaBody({multipart:true,formid

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
};

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持脚本之家。

koa上传excel koa上传excel文件并解析