go导出excel

入口函数

_, _, list, err := service.BuildingDevice.GetListV2(req)
 if err != nil {
 c.FailJsonExit(r, err.Error())
 }
 excel, err := c.DataToExcel(list)
 if err != nil {
 c.FailJsonExit(r, err.Error())
 }
 // 返回文件流
 filename := "建筑设备信息"
 filename = fmt.Sprintf("%s.xlsx", filename)
 r.Response.Header().Set("Content-Type", "application/octet-stream")
 r.Response.Header().Set("Content-Disposition", "attachment; filename=" filename)
 r.Response.Header().Set("Content-Length", strconv.FormatInt(int64(excel.Len()), 10))
 r.Response.Write(excel)

处理函数

 "github.com/tealeg/xlsx"

func (c *buildingDevice) DataToExcel(data []*dao.BuildingDeviceListRes) (*bytes.Buffer, error) {
 file := xlsx.NewFile()
 sheet, err := file.AddSheet("建筑设备信息")
 if err != nil {
 return nil, err
 }
 row := sheet.AddRow()
 row.AddCell().SetString("设备名称")
 row.AddCell().SetString("设备类型")
 row.AddCell().SetString("设备编号")
 row.AddCell().SetString("规格型号")
 row.AddCell().SetString("区域类型")
 row.AddCell().SetString("所在楼层")
 row.AddCell().SetString("设备位置")
 row.AddCell().SetString("供应商")
 row.AddCell().SetString("生产日期")
 row.AddCell().SetString("启用日期")
 row.AddCell().SetString("合同状态")
 for _, v := range data {
 row := sheet.AddRow()
 row.AddCell().SetString(v.Name)
 switch v.DeviceType {
 case "1":
 row.AddCell().SetString("水表")
 case "2":
 row.AddCell().SetString("电表")
 case "3":
 row.AddCell().SetString("监控")
 case "4":
 row.AddCell().SetString("空调")
 default:
 row.AddCell().SetString("")
 }
 row.AddCell().SetString(v.Code)
 row.AddCell().SetString(v.DeviceModel)
 switch v.AreaType {
 case "1":
 row.AddCell().SetString("室内")
 default:
 row.AddCell().SetString("")
 }
 row.AddCell().SetString(v.TheFloor)
 row.AddCell().SetString(v.DeviceLocation)
 row.AddCell().SetString(v.Supplier)
 if v.ProduceDate.IsZero() {
 row.AddCell().SetString("")
 } else {
 row.AddCell().SetString(v.ProduceDate.Format("2006-01-02"))
 }
 if v.EnableDate.IsZero() {
 row.AddCell().SetString("")
 } else {
 row.AddCell().SetString(v.EnableDate.Format("2006-01-02"))
 }
 switch v.ContractStatus {
 case "1":
 row.AddCell().SetString("正常")
 case "2":
 row.AddCell().SetString("停用")
 default:
 row.AddCell().SetString("")
 }
 }
 buf := new(bytes.Buffer)
 file.Write(buf)
 return buf, nil
}

这篇好文章是转载于：学新通技术网