Golang對excel進行處理
阿新 • • 發佈:2018-11-05
國慶節祝我們的祖國更加繁榮昌盛,想必大家在國慶期間也好好放鬆了一下,中秋節過節停更了一次,家裡面有孩子了,放假期間希望能好好陪陪孩子,請大家諒解。今天發一篇技術文章,幫助大家逐漸迴歸到正常的工作狀態。
之前有關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更好一些。