1. 程式人生 > >Golang對excel進行處理

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更好一些。