使用Node.js将Excel写入文件流并通过API实现在浏览器下载
问题:使用Exceljs生成Excel流并触发浏览器下载失败
需求流程:读取CSV文件解析数据,读取Excel模板改写内容后写入文件流,调用GET API触发浏览器下载(后续集成到Angular应用),无需在服务器留存生成的文件(或下载后立即删除)。当前代码无报错,但无法触发浏览器下载,仅能在服务器生成文件。
原代码核心问题分析
- 异步流程未正确处理:
csvParse依赖stream事件但未返回Promise,导致await csvParse()无法等待CSV解析和Excel生成完成,直接返回了JSON响应。 - 未将Excel流传递给响应:生成的文件流未关联到
res对象,浏览器未收到任何Excel数据。 - 不必要的中间文件写入:先写本地文件再读流的方式既浪费资源,又增加了文件删除的复杂度,Exceljs支持直接写入响应流。
- 响应头不完整:
Content-disposition缺少filename参数,浏览器无法识别下载文件名。 - 缺失模板内容改写逻辑:原代码未将解析后的CSV数据写入Excel模板,生成的文件与模板完全一致。
解决方案:直接写入响应流(无中间文件)
以下是修正后的完整代码,完全避免服务器生成中间文件,直接将Excel内容通过响应流发送给浏览器:
import { Request, Response } from 'express'; import * as csv from 'csv-parse'; import * as fs from 'fs'; import Excel from 'exceljs'; // 存储解析后的CSV数据 const allParsedData: any[] = []; export const generateExcel = async (req: Request, res: Response) => { try { // 等待CSV解析完成 await csvParse(); // 设置完整响应头,指定下载文件名 res.setHeader( 'Content-disposition', `attachment; filename="generated-excel.xlsx"` ); res.contentType( 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet' ); // 读取Excel模板 const workbook = new Excel.Workbook(); await workbook.xlsx.readFile('./utilities/template.xlsx'); // 将解析后的CSV数据写入模板(示例逻辑:从第2行开始追加数据) const worksheet = workbook.getWorksheet(1); if (worksheet) { allParsedData.forEach((row) => { worksheet.addRow(row); }); } // 直接将workbook写入响应流,无需中间文件 await workbook.xlsx.write(res); // 结束响应 res.end(); } catch (err) { console.error(err); return res.status(500).json('生成Excel失败'); } }; const csvParse = () => { return new Promise<void>((resolve, reject) => { allParsedData.length = 0; // 清空历史数据 fs.createReadStream('./content/TestCsv.csv') // 直接指定CSV分隔符,无需手动处理字符串分割 .pipe(csv.parse({ delimiter: ';' })) .on('error', (error) => { console.error('CSV解析错误:', error); reject(error); }) .on('data', (row) => { allParsedData.push(row); }) .on('end', (rowCount) => { console.log(`CSV解析完成,共${rowCount}行`); resolve(); }); }); };
关键改动说明
- 异步流程修复:
csvParse返回Promise,通过stream的end/error事件控制异步流程,确保await能等待解析完成。 - 直接写入响应流:使用
workbook.xlsx.write(res)将Excel内容直接发送给浏览器,完全省略中间文件生成步骤。 - 完善响应头:添加
filename参数,让浏览器明确下载文件的名称。 - 简化CSV解析:通过
csv.parse的delimiter参数直接处理分号分隔的CSV,避免手动字符串分割的错误。 - 补充模板改写逻辑:添加了将解析后的CSV数据写入Excel模板的示例代码,确保生成的Excel包含业务数据。
备选方案:需生成中间文件时的修正(下载后自动删除)
如果因特殊需求必须生成服务器本地文件,可使用以下代码,确保文件在下载完成后自动删除:
export const generateExcel = async (req: Request, res: Response) => { try { await csvParse(); const fileStream = await generateExcelFile(); // 设置响应头 res.setHeader( 'Content-disposition', `attachment; filename="generated-excel.xlsx"` ); res.contentType( 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet' ); // 将文件流pipe到响应 fileStream.pipe(res); // 流传输结束后删除本地文件 fileStream.on('end', () => { fs.unlink('./content/Test.xlsx', (err) => { if (err) console.error('删除临时文件失败:', err); }); }); } catch (err) { console.error(err); return res.status(500).json('生成Excel失败'); } }; const generateExcelFile = async () => { const workbook = new Excel.Workbook(); await workbook.xlsx.readFile('./utilities/template.xlsx'); // 改写模板内容逻辑 const worksheet = workbook.getWorksheet(1); if (worksheet) { allParsedData.forEach((row) => { worksheet.addRow(row); }); } await workbook.xlsx.writeFile('./content/Test.xlsx'); return fs.createReadStream('./content/Test.xlsx'); };
内容的提问来源于stack exchange,提问作者devZ
相关产品推荐
相关产品推荐

