Golang导入导出Excel表格

最近项目开发中有涉及到Excel的导入与导出功能，特别是导出表格时需要特定的格式（单元格合并等），废话不多说，直接上代码了。

首先用到一个第三方库，实测还是很强大很好用的，就是这个https://github.com/qax-os/excelize

引用库

go get github.com/xuri/excelize/v2

导入表格

package main

import (
  "encoding/json"
  "github.com/xuri/excelize/v2"
  "net/http"
  "strings"
)

func fileImport(w http.ResponseWriter, r *http.Request) {
  // 获取请求中的文件名
  formFile, _, err := r.FormFile("filename")
  if err != nil {
    w.Write([]byte("获取文件失败, " + err.Error()))
    return
  }
  // 关闭
  defer formFile.Close()
  //
  reader, err := excelize.OpenReader(formFile)
  if err != nil {
    w.Write([]byte("读取文件失败, " + err.Error()))
    return
  }
  // 关闭
  defer reader.Close()
  rows, err := reader.GetRows("Sheet1")
  if err != nil {
    w.Write([]byte("获取工作表失败, " + err.Error()))
    return
  }
  ret := make([]string, 0, 8)
  for i, row := range rows {
    // 每一行数据的列, 都是从0开始的, 一般0行都是表头
    if i == 0 {
      continue
    }
    value1 := row[0] // 第一列
    value2 := row[1] // 第二列
    // 去除空格
    value1 = strings.Trim(strings.TrimSpace(value1), "\n")
    value2 = strings.Trim(strings.TrimSpace(value2), "\n")
    //
    ret = append(ret, value1+","+value2)
  }
  bytes, _ := json.Marshal(ret)
  w.Write(bytes)
  return
}

func main() {
  // HTTP服务
  http.HandleFunc("/fileImport", fileImport)
  err := http.ListenAndServe(":8192", nil)
  if err != nil {
    panic(err)
  }
}

表格格式

执行后效果

导出表格

package main

import (
  "encoding/json"
  "fmt"
  "github.com/xuri/excelize/v2"
  "net/http"
  "strings"
)

func fileExport(w http.ResponseWriter, r *http.Request) {
  file := excelize.NewFile()
  defer file.Close()

  // 设置页
  sheetName := "Sheet1"
  // 创建
  sheet, err := file.NewSheet(sheetName)
  if err != nil {
    w.Write([]byte("创建失败, " + err.Error()))
    return
  }
  // 设置单元格格式
  style := &excelize.Style{
    Border: nil,
    Fill:   excelize.Fill{},
    Font:   nil,
    Alignment: &excelize.Alignment{
      Horizontal:      "center",
      Indent:          0,
      JustifyLastLine: false,
      ReadingOrder:    0,
      RelativeIndent:  0,
      ShrinkToFit:     false,
      TextRotation:    0,
      Vertical:        "center",
      WrapText:        false,
    },
    Protection:    nil,
    NumFmt:        0,
    DecimalPlaces: nil,
    CustomNumFmt:  nil,
    NegRed:        false,
  }
  styleID, _ := file.NewStyle(style)
  // 设置表头
  _ = file.SetCellValue(sheetName, "A1", "款")
  _ = file.SetCellStyle(sheetName, "A1", "A1", styleID)
  _ = file.SetCellValue(sheetName, "B1", "尺码")
  _ = file.SetCellStyle(sheetName, "B1", "B1", styleID)
  // 设置值
  for i := 0; i < 5; i++ {
    lineStr := fmt.Sprintf("%d", i+2)
    //
    _ = file.SetCellValue(sheetName, "A"+lineStr, "基础款")
    _ = file.SetCellStyle(sheetName, "A"+lineStr, "A"+lineStr, styleID)
    //
    _ = file.SetCellValue(sheetName, "B"+lineStr, "1:2:3:4:5:6")
    _ = file.SetCellStyle(sheetName, "B"+lineStr, "B"+lineStr, styleID)
  }
  //
  file.SetActiveSheet(sheet)
  //
  buffer, err := file.WriteToBuffer()
  if err != nil {
    w.Write([]byte("导出失败, " + err.Error()))
    return
  }
  w.Header().Set("Content-Disposition", fmt.Sprintf("attachment; filename=%s", "导出文件.xlsx"))
  w.Header().Set("Content-Type", "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet")
  w.Write(buffer.Bytes())
}

func main() {
  // HTTP服务
  http.HandleFunc("/fileImport", fileImport)
  http.HandleFunc("/fileExport", fileExport)
  err := http.ListenAndServe(":8192", nil)
  if err != nil {
    panic(err)
  }
}

页面请求 http://127.0.0.1:8192/fileExport 后会直接生成xlsx文件并下载

合并单元格

package main

import (
  "encoding/json"
  "fmt"
  "github.com/xuri/excelize/v2"
  "net/http"
  "strings"
)

func fileExport(w http.ResponseWriter, r *http.Request) {
  file := excelize.NewFile()
  defer file.Close()

  // 设置页
  sheetName := "Sheet1"
  // 创建
  sheet, err := file.NewSheet(sheetName)
  if err != nil {
    w.Write([]byte("创建失败, " + err.Error()))
    return
  }
  // 设置单元格格式
  style := &excelize.Style{
    Border: nil, // 边框样式
    Fill:   excelize.Fill{},
    Font:   nil, // 字体样式
    Alignment: &excelize.Alignment{ // 位置样式
      Horizontal:      "center",
      Indent:          0,
      JustifyLastLine: false,
      ReadingOrder:    0,
      RelativeIndent:  0,
      ShrinkToFit:     false,
      TextRotation:    0,
      Vertical:        "center",
      WrapText:        false,
    },
    Protection:    nil,
    NumFmt:        0,
    DecimalPlaces: nil,
    CustomNumFmt:  nil,
    NegRed:        false,
  }
  styleID, _ := file.NewStyle(style)
  // 设置表头
  _ = file.MergeCell(sheetName, "A1", "A2") // 合并单元格
  _ = file.SetCellValue(sheetName, "A1", "款")
  _ = file.SetCellStyle(sheetName, "A1", "A2", styleID)

  _ = file.MergeCell(sheetName, "B1", "G1") // 合并单元格
  _ = file.SetCellValue(sheetName, "B1", "尺码")
  _ = file.SetCellStyle(sheetName, "B1", "G1", styleID)
  _ = file.SetCellValue(sheetName, "B2", "XS")
  _ = file.SetCellStyle(sheetName, "B2", "B2", styleID)
  _ = file.SetCellValue(sheetName, "C2", "S")
  _ = file.SetCellStyle(sheetName, "C2", "C2", styleID)
  _ = file.SetCellValue(sheetName, "D2", "M")
  _ = file.SetCellStyle(sheetName, "D2", "D2", styleID)
  _ = file.SetCellValue(sheetName, "E2", "L")
  _ = file.SetCellStyle(sheetName, "E2", "E2", styleID)
  _ = file.SetCellValue(sheetName, "F2", "XL")
  _ = file.SetCellStyle(sheetName, "F2", "F2", styleID)
  _ = file.SetCellValue(sheetName, "G2", "XLL")
  _ = file.SetCellStyle(sheetName, "G2", "G2", styleID)
  // 设置值
  for i := 0; i < 5; i++ {
    lineStr := fmt.Sprintf("%d", i+3)
    //
    _ = file.SetCellValue(sheetName, "A"+lineStr, "基础款")
    _ = file.SetCellStyle(sheetName, "A"+lineStr, "A"+lineStr, styleID)
    //
    split := strings.Split("1:2:3:4:5:6", ":")
    _ = file.SetCellValue(sheetName, "B"+lineStr, split[0])
    _ = file.SetCellStyle(sheetName, "B"+lineStr, "B"+lineStr, styleID)
    _ = file.SetCellValue(sheetName, "C"+lineStr, split[1])
    _ = file.SetCellStyle(sheetName, "C"+lineStr, "C"+lineStr, styleID)
    _ = file.SetCellValue(sheetName, "D"+lineStr, split[2])
    _ = file.SetCellStyle(sheetName, "D"+lineStr, "D"+lineStr, styleID)
    _ = file.SetCellValue(sheetName, "E"+lineStr, split[3])
    _ = file.SetCellStyle(sheetName, "E"+lineStr, "E"+lineStr, styleID)
    _ = file.SetCellValue(sheetName, "F"+lineStr, split[4])
    _ = file.SetCellStyle(sheetName, "F"+lineStr, "F"+lineStr, styleID)
    _ = file.SetCellValue(sheetName, "G"+lineStr, split[5])
    _ = file.SetCellStyle(sheetName, "G"+lineStr, "G"+lineStr, styleID)
  }
  //
  file.SetActiveSheet(sheet)
  //
  buffer, err := file.WriteToBuffer()
  if err != nil {
    w.Write([]byte("导出失败, " + err.Error()))
    return
  }
  // 设置文件名
  w.Header().Set("Content-Disposition", fmt.Sprintf("attachment; filename=%s", "导出文件.xlsx"))
  // 导出的文件格式 xlsx 或者 xsl
  // xlsx application/vnd.openxmlformats-officedocument.spreadsheetml.sheet
  // xls application/vnd.ms-excel
  w.Header().Set("Content-Type", "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet")
  w.Write(buffer.Bytes())
}

func main() {
  // HTTP服务
  http.HandleFunc("/fileImport", fileImport)
  http.HandleFunc("/fileExport", fileExport)
  err := http.ListenAndServe(":8192", nil)
  if err != nil {
    panic(err)
  }
}

页面请求 http://127.0.0.1:8192/fileExport 后会直接生成xlsx文件并下载