Golang是如何操作excel的?
关键术语介绍
为了方便开源库的快速上手,我们先来了解 excel 中的几个关键术语,如下图所示,①为sheet,也就是表格中的页签;②为row,代表 excel 中的一行;③为cell,代表 excel 中的一个单元格。
正常情况下,创建一个表格的基本流程是打开 wps 点击新建,这时会默认创建一个 sheet,然后在该 sheet 中的第一行填写表头,接下来根据表头逐行填充内容,最后将文件另存为到硬盘的某个位置。这与 Golang 开源库创建 excel 的流程基本相同,下面演示一个极简表格的创建。
创建表格
创建表格前需要先引入 excel 库,我们以比较热门的 tealeg/xlsx 库为例。
go get github.com/tealeg/xlsx
首先创建一个空文件,拿到文件句柄。
file := xlsx.NewFile()
创建一个名为人员信息收集
的 sheet。
sheet, err := file.AddSheet("人员信息收集")
if err != nil {
panic(err.Error())
}
然后为该 sheet 创建一行,这行作为我们的表头。
row := sheet.AddRow()
在该行中创建一个单元格。
cell := row.AddCell()
现在给单元格填充内容,因为是表头,暂且叫姓名
。
cell.Value = "姓名"
如何创建第二个单元格呢?原理相同,此处 cell 变量已定义,再创建新单元格只需赋值即可。
cell = row.AddCell()
cell.Value = "性别"
表头已经设置好了,可以开始创建第二行来填充内容了,方式与上述无差别。
row = sheet.AddRow()
cell = row.AddCell()
cell.Value = "张三"
cell = row.AddCell()
cell.Value = "男"
表格设置完成后,将该文件保存,文件名可自定义。
err = file.Save("demo.xlsx")
if err != nil {
panic(err.Error())
}
跑起来后,可以发现目录中多了一个 demo.xlsx 文件,打开预览内容如下,达到了预期效果。
文件源码
package main
import "github.com/tealeg/xlsx"
func main() {
file := xlsx.NewFile()
sheet, err := file.AddSheet("人员信息收集")
if err != nil {
panic(err.Error())
}
row := sheet.AddRow()
cell := row.AddCell()
cell.Value = "姓名"
cell = row.AddCell()
cell.Value = "性别"
row = sheet.AddRow()
cell = row.AddCell()
cell.Value = "张三"
cell = row.AddCell()
cell.Value = "男"
err = file.Save("demo.xlsx")
if err != nil {
panic(err.Error())
}
}
读取表格
表格的读取比创建简单很多,依然以上文创建的文件为例。
output, err := xlsx.FileToSlice("demo.xlsx")
if err != nil {
panic(err.Error())
}
只需将文件路径传入上述方法,即可自动读取并返回一个三维切片,我们来读取第一个 sheet 的第二行中的第一个单元格。
log.Println(output[0][1][1]) //Output: 男
由此一来就非常容易遍历了。
for rowIndex, row := range output[0] {
for cellIndex, cell := range row {
log.Println(fmt.Sprintf("第%d行,第%d个单元格:%s", rowIndex+1, cellIndex+1, cell))
}
}
2020/10/11 16:15:29 第1行,第1个单元格:姓名
2020/10/11 16:15:29 第1行,第2个单元格:性别
2020/10/11 16:15:29 第2行,第1个单元格:张三
2020/10/11 16:15:29 第2行,第2个单元格:男
文件源码
package main
import (
"fmt"
"github.com/tealeg/xlsx"
"log"
)
func main() {
output, err := xlsx.FileToSlice("demo.xlsx")
if err != nil {
panic(err.Error())
}
log.Println(output[0][1][1])
for rowIndex, row := range output[0] {
for cellIndex, cell := range row {
log.Println(fmt.Sprintf("第%d行,第%d个单元格:%s", rowIndex+1, cellIndex+1, cell))
}
}
}
修改表格
只是读取表格内容可能在特定场景下无法满足需求,有时候需要对表格内容进行更改。
file, err := xlsx.OpenFile("demo.xlsx")
if err != nil {
panic(err.Error())
}
修改表格之前依然需要先读取文件,只是这次并没有直接将其转化为三维切片。拿到文件句柄后,可以直接修改某一行的内容。
file.Sheets[0].Rows[1].Cells[0].Value = "李四"
上述代码将第二行的张三改为了李四,但这还没有结束,接下来需要将文件重新保存。
err = file.Save("demo.xlsx")
if err != nil {
panic(err.Error())
}
打开文件预览,可以看到已经成功将张三改为了李四。
文件源码
package main
import "github.com/tealeg/xlsx"
func main() {
file, err := xlsx.OpenFile("demo.xlsx")
if err != nil {
panic(err.Error())
}
file.Sheets[0].Rows[1].Cells[0].Value = "李四"
err = file.Save("demo.xlsx")
if err != nil {
panic(err.Error())
}
}
样式设置
该开源库不仅支持内容的编辑,还支持表格的样式设置,样式统一由结构体 Style 来负责。
type Style struct {
Border Border
Fill Fill
Font Font
ApplyBorder bool
ApplyFill bool
ApplyFont bool
ApplyAlignment bool
Alignment Alignment
NamedStyleIndex *int
}
拿上述生成的文件为例,假如我要将姓名所在单元格居中,首先要实例化样式对象。
style := xlsx.NewStyle()
赋值居中属性。
style.Alignment = xlsx.Alignment{
Horizontal: "center",
Vertical: "center",
}
给第一行第一个单元格设置样式。
file.Sheets[0].Rows[0].Cells[0].SetStyle(style)
与修改表格处理逻辑相同,最后保存文件。
err = file.Save("demo.xlsx")
if err != nil {
panic(err.Error())
}
打开预览,可以看到文字已经上下左右居中。
同理,可以修改文字颜色和背景,同样通过 style 的属性来设置。
style.Font.Color = xlsx.RGB_Dark_Red
style.Fill.BgColor = xlsx.RGB_Dark_Green
其他还有很多属性可以设置,比如合并单元格、字体、大小等等,大家可以自行测试。
文件源码
package main
import "github.com/tealeg/xlsx"
func main() {
file, err := xlsx.OpenFile("demo.xlsx")
if err != nil {
panic(err.Error())
}
style := xlsx.NewStyle()
style.Font.Color = xlsx.RGB_Dark_Red
style.Fill.BgColor = xlsx.RGB_Dark_Green
style.Alignment = xlsx.Alignment{
Horizontal: "center",
Vertical: "center",
}
file.Sheets[0].Rows[0].Cells[0].SetStyle(style)
err = file.Save("demo.xlsx")
if err != nil {
panic(err.Error())
}
}
我是平也,这有一个专注Gopher技术成长的开源项目「go home」
感谢大家的观看,如果觉得文章对你有所帮助,欢迎关注公众号「平也」,聚焦Go语言与技术原理。