当前位置：首页 > news >正文

Go Web项目实战：接收上传的Excel文件，处理后再下载（附完整代码）

news 2026/5/31 9:28:26

Go Web实战：Excel文件上传处理与下载全流程解析

在企业级应用开发中，Excel文件处理是常见的业务需求。本文将带你构建一个完整的Go Web服务，实现Excel文件的上传、业务处理与下载功能。我们使用excelize/v2这个高性能库，它特别适合处理大规模数据，相比其他语言的同类库有着显著的性能优势。

1. 项目环境搭建与基础配置

首先确保你的开发环境已经安装Go 1.16或更高版本。创建一个新的项目目录并初始化模块：

mkdir excel-processor && cd excel-processor go mod init github.com/yourusername/excel-processor

安装所需的依赖库：

go get github.com/xuri/excelize/v2

创建一个基本的项目结构：

excel-processor/ ├── main.go # 主程序入口 ├── handlers/ # 请求处理器 │ └── excel.go # Excel处理逻辑 ├── static/ # 静态文件 │ └── index.html # 上传页面 └── go.mod # 模块定义

2. 构建HTTP服务器与文件上传接口

我们先创建一个简单的HTTP服务器，设置文件上传路由：

package main import ( "log" "net/http" "excel-processor/handlers" ) func main() { // 设置路由 http.HandleFunc("/upload", handlers.HandleExcelUpload) http.Handle("/", http.FileServer(http.Dir("./static"))) // 启动服务器 log.Println("Server starting on :8080...") if err := http.ListenAndServe(":8080", nil); err != nil { log.Fatalf("Server failed to start: %v", err) } }

在handlers/excel.go中实现文件上传处理：

package handlers import ( "fmt" "net/http" "github.com/xuri/excelize/v2" ) func HandleExcelUpload(w http.ResponseWriter, r *http.Request) { // 限制上传文件大小 r.ParseMultipartForm(10 << 20) // 10MB file, header, err := r.FormFile("excelFile") if err != nil { http.Error(w, "Error retrieving the file", http.StatusBadRequest) return } defer file.Close() // 验证文件类型 if header.Header.Get("Content-Type") != "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet" { http.Error(w, "Invalid file type, please upload an Excel file", http.StatusBadRequest) return } // 处理Excel文件 processedFile, err := processExcel(file) if err != nil { http.Error(w, fmt.Sprintf("Error processing Excel: %v", err), http.StatusInternalServerError) return } // 设置响应头 w.Header().Set("Content-Disposition", "attachment; filename=processed_"+header.Filename) w.Header().Set("Content-Type", "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet") // 将处理后的文件写入响应 if _, err := processedFile.WriteTo(w); err != nil { http.Error(w, fmt.Sprintf("Error writing response: %v", err), http.StatusInternalServerError) } }

3. Excel文件处理核心逻辑

在processExcel函数中，我们将实现具体的业务处理逻辑。以下是几种常见的Excel处理场景：

3.1 基础数据读取与修改

func processExcel(file io.Reader) (*excelize.File, error) { f, err := excelize.OpenReader(file) if err != nil { return nil, fmt.Errorf("open reader error: %w", err) } // 获取第一个工作表名 sheetName := f.GetSheetName(0) // 读取单元格数据 originalValue, err := f.GetCellValue(sheetName, "A1") if err != nil { return nil, fmt.Errorf("get cell value error: %w", err) } // 修改单元格 if err := f.SetCellValue(sheetName, "A1", originalValue+"_processed"); err != nil { return nil, fmt.Errorf("set cell value error: %w", err) } // 添加新工作表 if _, err := f.NewSheet("Summary"); err != nil { return nil, fmt.Errorf("create new sheet error: %w", err) } // 在新工作表中添加汇总数据 summaryData := [][]interface{}{ {"Item", "Count", "Total"}, {"Products", 150, 4500}, {"Services", 75, 2250}, } for i, row := range summaryData { startCell, _ := excelize.CoordinatesToCellName(1, i+1) if err := f.SetSheetRow("Summary", startCell, &row); err != nil { return nil, fmt.Errorf("set sheet row error: %w", err) } } return f, nil }

3.2 流式处理大规模数据

对于大型Excel文件，使用流式API可以显著降低内存消耗：

func processLargeExcel(f *excelize.File) error { // 创建流式写入器 sw, err := f.NewStreamWriter("LargeData") if err != nil { return fmt.Errorf("create stream writer error: %w", err) } // 流式写入标题行 headers := []interface{}{"ID", "Name", "Value", "Date", "Status"} if err := sw.SetRow("A1", headers); err != nil { return fmt.Errorf("set header row error: %w", err) } // 模拟写入10000行数据 for rowID := 2; rowID <= 10001; rowID++ { row := []interface{}{ rowID - 1, fmt.Sprintf("Product-%d", rowID), rand.Intn(1000), time.Now().AddDate(0, 0, rowID-2).Format("2006-01-02"), []string{"Active", "Inactive", "Pending"}[rand.Intn(3)], } cell, _ := excelize.CoordinatesToCellName(1, rowID) if err := sw.SetRow(cell, row); err != nil { return fmt.Errorf("set row %d error: %w", rowID, err) } } // 结束流式写入 if err := sw.Flush(); err != nil { return fmt.Errorf("flush stream writer error: %w", err) } return nil }

3.3 高级功能：公式、样式与图表

func addAdvancedFeatures(f *excelize.File) error { sheetName := "Advanced" f.NewSheet(sheetName) // 设置样式 styleID, err := f.NewStyle(&excelize.Style{ Font: &excelize.Font{Bold: true, Color: "1a5fb4"}, Fill: excelize.Fill{Type: "pattern", Color: []string{"e5f6ff"}, Pattern: 1}, Alignment: &excelize.Alignment{Horizontal: "center"}, }) if err != nil { return fmt.Errorf("create style error: %w", err) } // 写入带样式的数据 data := [][]interface{}{ {"Region", "Q1", "Q2", "Q3", "Q4", "Total"}, {"North", 1250, 1320, 1420, 1580}, {"South", 980, 1050, 1120, 1240}, {"East", 870, 910, 950, 1020}, {"West", 1150, 1250, 1310, 1400}, } for i, row := range data { cell, _ := excelize.CoordinatesToCellName(1, i+1) if err := f.SetSheetRow(sheetName, cell, &row); err != nil { return fmt.Errorf("set row error: %w", err) } // 应用样式到标题行 if i == 0 { startCell, _ := excelize.CoordinatesToCellName(1, 1) endCell, _ := excelize.CoordinatesToCellName(len(row), 1) f.SetCellStyle(sheetName, startCell, endCell, styleID) } } // 添加公式计算总计 for i := 2; i <= 5; i++ { cell, _ := excelize.CoordinatesToCellName(6, i) formula := fmt.Sprintf("SUM(B%d:E%d)", i, i) f.SetCellFormula(sheetName, cell, formula) } // 添加图表 if err := addChartToSheet(f, sheetName); err != nil { return fmt.Errorf("add chart error: %w", err) } return nil }

4. 前端上传界面与用户体验优化

创建一个简单的HTML上传页面static/index.html：

<!DOCTYPE html> <html lang="en"> <head> <meta charset="UTF-8"> <meta name="viewport" content="width=device-width, initial-scale=1.0"> <title>Excel Processor</title> <style> body { font-family: Arial, sans-serif; max-width: 800px; margin: 0 auto; padding: 20px; } .upload-container { border: 2px dashed #ccc; padding: 40px; text-align: center; border-radius: 5px; margin-bottom: 20px; } .progress { display: none; margin-top: 20px; } #fileInput { display: none; } .btn { background-color: #1a5fb4; color: white; padding: 10px 20px; border: none; border-radius: 4px; cursor: pointer; font-size: 16px; } .btn:hover { background-color: #1a73e8; } </style> </head> <body> <h1>Excel File Processor</h1> <p>Upload an Excel file to process and download the modified version.</p> <div class="upload-container"> <input type="file" id="fileInput" accept=".xlsx, .xls"> <button class="btn" onclick="document.getElementById('fileInput').click()"> Select Excel File </button> <p id="fileName" style="margin-top: 10px;"></p> </div> <button id="uploadBtn" class="btn" disabled>Process File</button> <div class="progress" id="progressContainer"> <p>Processing file... <span id="progressText">0%</span></p> <progress id="progressBar" value="0" max="100"></progress> </div> <script> const fileInput = document.getElementById('fileInput'); const uploadBtn = document.getElementById('uploadBtn'); const fileName = document.getElementById('fileName'); const progressContainer = document.getElementById('progressContainer'); const progressText = document.getElementById('progressText'); const progressBar = document.getElementById('progressBar'); fileInput.addEventListener('change', (e) => { if (e.target.files.length > 0) { fileName.textContent = `Selected file: ${e.target.files[0].name}`; uploadBtn.disabled = false; } }); uploadBtn.addEventListener('click', async () => { const file = fileInput.files[0]; if (!file) return; const formData = new FormData(); formData.append('excelFile', file); // 显示进度条 progressContainer.style.display = 'block'; uploadBtn.disabled = true; try { const xhr = new XMLHttpRequest(); xhr.open('POST', '/upload', true); xhr.responseType = 'blob'; // 进度事件 xhr.upload.onprogress = (e) => { if (e.lengthComputable) { const percent = Math.round((e.loaded / e.total) * 100); progressText.textContent = `${percent}%`; progressBar.value = percent; } }; xhr.onload = () => { if (xhr.status === 200) { // 创建下载链接 const blob = new Blob([xhr.response], {type: 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet'}); const url = window.URL.createObjectURL(blob); const a = document.createElement('a'); a.href = url; a.download = `processed_${file.name}`; document.body.appendChild(a); a.click(); window.URL.revokeObjectURL(url); a.remove(); // 重置UI progressContainer.style.display = 'none'; uploadBtn.disabled = false; fileName.textContent = ''; fileInput.value = ''; } else { alert(`Error: ${xhr.statusText}`); } }; xhr.send(formData); } catch (error) { console.error('Error:', error); alert('An error occurred while processing the file'); progressContainer.style.display = 'none'; uploadBtn.disabled = false; } }); </script> </body> </html>

5. 性能优化与错误处理

5.1 内存优化技巧

处理大型Excel文件时，内存管理至关重要：

使用流式API：对于大数据量操作，始终使用NewStreamWriter
及时关闭资源：确保在所有操作完成后调用Close()方法
限制并发处理：对于高并发场景，实现请求队列或限制并发数

// 带缓冲区的处理池示例 var semaphore = make(chan struct{}, 5) // 限制5个并发处理 func HandleExcelUpload(w http.ResponseWriter, r *http.Request) { // 获取信号量 semaphore <- struct{}{} defer func() { <-semaphore }() // 其余处理逻辑... }

5.2 全面的错误处理

Excel处理可能遇到多种错误情况，需要妥善处理：

func processExcel(file io.Reader) (*excelize.File, error) { f, err := excelize.OpenReader(file) if err != nil { return nil, fmt.Errorf("open reader error: %w", err) } defer func() { if err := f.Close(); err != nil { log.Printf("Warning: error closing file: %v", err) } }() // 检查工作表是否存在 sheetName := f.GetSheetName(0) if sheetName == "" { return nil, fmt.Errorf("no worksheets found in the file") } // 验证工作表是否受保护 protected, err := f.IsProtected(sheetName) if err != nil { return nil, fmt.Errorf("error checking sheet protection: %w", err) } if protected { return nil, fmt.Errorf("protected worksheets are not supported") } // 其余处理逻辑... }