快捷導(dǎo)航

koa上傳excel文件并解析的實(shí)現(xiàn)方法

更新時(shí)間：2018年08月09日 13:39:18 作者：小谷xg

這篇文章主要介紹了koa上傳excel文件并解析的實(shí)現(xiàn)方法，小編覺(jué)得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧

1.中間鍵使用 koa-body

npm install koa-body --save

const koaBody = require('koa-body');

app.use(koaBody({
 multipart: true,
 formidable: {
  maxFileSize: 200 * 1024 * 1024 // 設(shè)置上傳文件大小最大限制，默認(rèn)2M
 }
}));

2.書(shū)寫(xiě)路由，croller書(shū)寫(xiě)方法

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, '暫時(shí)不支持多個(gè)sheet存在');
  } else { //得到的是數(shù)組
   const objs = getRes.datas[0];
   ctx.body = {
    status: true,
    msg: '上傳數(shù)據(jù)成功'
   };
  }
 } else {
  errorResult.errorRes(ctx, getRes.msg);
 }
 await next();
};
module.exports = {
 saveData
};

3.處理excel存儲(chǔ)，解析,處理excel用的庫(kù)是 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); // 創(chuàng)建可讀流
 const ext = file.name.split('.').pop(); // 獲取上傳文件擴(kuò)展名
 const filePath = `${downPath}/${Math.random().toString()}.${ext}`;

 const upStream = fs.createWriteStream(filePath); // 創(chuàng)建可寫(xiě)流
 const getRes = await getFile(reader, upStream); //等待數(shù)據(jù)存儲(chǔ)完成

 const datas = []; //可能存在多個(gè)sheet的情況
 if (!getRes) { //沒(méi)有問(wèn)題
  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: '上傳文件錯(cuò)誤'
  };
 }
}

function getFile (reader, upStream) {
 return new Promise(function (result) {
  let stream = reader.pipe(upStream); // 可讀流通過(guò)管道寫(xiě)入可寫(xiě)流
  stream.on('finish', function (err) {
   result(err);
  });
 });
}
module.exports = {
 getExcelObjs
};

以上就是本文的全部?jī)?nèi)容，希望對(duì)大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: