前一段时间想要用golang读取xls文件里的数据,并写入到xlsx中,xls的读写国内网上的教程鲜有能参考的,因此将自己的处理的过程整理了下供需要的小伙伴们参考,由于是初学golang,水平有限,欢迎大家批评指正。
test.xls里的数据
package main
import (
"log"
"os"
"github.com/extrame/xls"
"github.com/tealeg/xlsx"
)
var xlsxTitle = []string{"字段1", "字段2", "字段3", "字段4"}
var cell *xlsx.Cell
func main() {
pwd, _ := os.Getwd()
targetPath := pwd + `\result.xlsx`
xlsxFile := getXlsxFile(targetPath)
xlsxSheet := xlsxFile.Sheets[0]
dataToInsert := []string{"切片内数据1", "切片内数据2", "切片内数据3", "切片内数据4"}
insertRow(xlsxSheet, &dataToInsert)
xlsPath := pwd + `\test.xls`
xlsFile, closer, err := xls.OpenWithCloser(xlsPath, "utf-8")
if err != nil {
log.Fatal(err)
}
// 获取xls文件的第一个sheet
sheet := xlsFile.GetSheet(0)
// 从第二行开始,遍历xls文件,然后按行调用insertRowFromXls函数
for j := 1; j < int(sheet.MaxRow)+1; j++ {
xlsRow := sheet.Row(j)
rowColCount := xlsRow.LastCol()
insertRowFromXls(xlsxSheet, xlsRow, rowColCount)
}
closer.Close()
xlsxFile.Save(targetPath)
}
// 将xls.Row指针对应的数据插入到xlsx.sheet中
func insertRowFromXls(sheet *xlsx.Sheet, rowDataPtr *xls.Row, rowColCount int) {
row := sheet.AddRow()
for i := 0; i < rowColCount; i++ {
cell = row.AddCell()
cell.Value = rowDataPtr.Col(i)
}
}
// 将一个切片指针对应的数据插入到xlsx.sheet中
func insertRow(sheet *xlsx.Sheet, rowDataPtr *[]string) {
row := sheet.AddRow()
rowData := *rowDataPtr
for _, v := range rowData {
cell := row.AddCell()
cell.Value = v
}
}
// 获取xlsx.File对象的指针,如果文件路径不存在则新建一个文件,并返回其指针
func getXlsxFile(filePath string) *xlsx.File {
var file *xlsx.File
if _, err := os.Stat(filePath); err == nil {
file, err = xlsx.OpenFile(filePath)
if err != nil {
log.Fatal(err)
}
} else {
file = xlsx.NewFile()
sheet, err := file.AddSheet("sheet1")
if err != nil {
log.Fatal(err)
}
insertRow(sheet, &xlsxTitle)
}
return file
}
最终result.xlsx里的数据