Golang对excel进行处理

国庆节祝我们的祖国更加繁荣昌盛，想必大家在国庆期间也好好放松了一下，中秋节过节停更了一次，家里面有孩子了，放假期间希望能好好陪陪孩子，请大家谅解。今天发一篇技术文章，帮助大家逐渐回归到正常的工作状态。

之前有关Golang的文章主要是基于Golang的标准库来讲解的，今天我们来看看Golang的一个开源库，它可以用来处理xlsx文件，首先下载包

go get github.com/tealeg/xlsx

我们通过一个简单的例子看看这个包如何使用，这个例子主要实现输入列名，将表格中该列的所有数据进行显示。代码如下所示：

package main

import (
	"bufio"
	"fmt"
	"os"

	"github.com/tealeg/xlsx"
)

func main() {
	if len(os.Args) != 3 {
		fmt.Println("Usage: xlsx pathname sheetname")
		os.Exit(1)
	}

	xlsxFile, err := xlsx.OpenFile(os.Args[1])
	if err != nil {
		fmt.Println(err)
		os.Exit(1)
	}

	sheet := xlsxFile.Sheet[os.Args[2]]
	if sheet == nil {
		fmt.Println("表单名不存在")
		os.Exit(1)
	}

	for {
		title := getStdinInput("请输入列名：")
		if title == "" {
			fmt.Println(title)
			continue
		}

		titleColIndex := findColByTitle(sheet, title)
		if titleColIndex == -1 {
			fmt.Println("列名不存在")
			continue
		}

		rowLen := len(sheet.Rows)
		result := []string{}
		for rowIndex := 1; rowIndex < rowLen; rowIndex++ {
			content := sheet.Cell(rowIndex, titleColIndex).String()
			result = append(result, content)
		}

		fmt.Println(result)
	}
}

func getStdinInput(hint string) string {
	fmt.Print(hint)
	scanner := bufio.NewScanner(os.Stdin)
	if scanner.Scan() {
		return scanner.Text()
	}

	return ""
}

func findColByTitle(sheet *xlsx.Sheet, title string) int {
	titleRow := sheet.Rows[0]
	for titleIndex, col := range titleRow.Cells {
		if col.String() == title {
			return titleIndex
		}
	}

	return -1
}

先看主函数，主函数首先进行命令行参数校验，使用该程序需要使用两个参数，一个是xlsx的路径，一个是需要使用的表单名称。之后打开xlsx文件和对应的表单，通过标准输入读取列名，然后在对应的表单中查找列名，通过遍历所有行，获取该列的所有数据。从标准输入读取数据和查找对应的列索引这里封装了两个函数。

getStdinInput()函数接收一个参数，作为输入的提示语句，该函数基于scanner获取标准输入的文本。

findColByTitle()函数传入两个参数：表单对象的指针和列名。通过遍历所有的标题行中的列，查找匹配的列索引并返回。

可以自己创建一个标准的xlsx文件，第一行是标题行，然后实用程序测试一下，之前有使用Python做Excel处理，但是感觉运行效率还是Golang更好一些。

Golang对excel进行处理

猜你喜欢