如何用Golang实现将数据库数据导出到Excel文件

1,240 阅读1分钟

使用Golang实现将数据库数据导出到Excel文件,可以使用第三方库如github.com/360EntSecGroup-Skylar/excelize。以下是一个示例代码:

package main

import (
	"database/sql"
	"fmt"
	"log"

	_ "github.com/go-sql-driver/mysql"
	"github.com/360EntSecGroup-Skylar/excelize"
)

func main() {
	// 连接到数据库
	db, err := sql.Open("mysql", "username:password@tcp(localhost:3306)/database")
	if err != nil {
		log.Fatal(err)
	}
	defer db.Close()

	// 执行查询语句
	rows, err := db.Query("SELECT * FROM your_table")
	if err != nil {
		log.Fatal(err)
	}
	defer rows.Close()

	// 创建Excel文件
	file := excelize.NewFile()
	sheetName := "Sheet1"

	// 添加表头
	columns, err := rows.Columns()
	if err != nil {
		log.Fatal(err)
	}
	for i, colName := range columns {
		cell := excelize.ToAlphaString(i) + "1"
		file.SetCellValue(sheetName, cell, colName)
	}

	// 添加数据行
	rowIndex := 2
	values := make([]interface{}, len(columns))
	scanArgs := make([]interface{}, len(values))
	for i := range values {
		scanArgs[i] = &values[i]
	}

	for rows.Next() {
		err := rows.Scan(scanArgs...)
		if err != nil {
			log.Fatal(err)
		}

		for i, value := range values {
			cell := excelize.ToAlphaString(i) + fmt.Sprint(rowIndex)
			file.SetCellValue(sheetName, cell, value)
		}

		rowIndex++
	}

	// 保存Excel文件
	err = file.SaveAs("output.xlsx")
	if err != nil {
		log.Fatal(err)
	}

	fmt.Println("导出成功")
}

请注意,在运行代码之前,您需要先安装excelize库。可以使用以下命令进行安装:

go get github.com/360EntSecGroup-Skylar/excelize

在代码中的sql.Open函数中,将usernamepassworddatabase替换为您实际的数据库连接信息。然后,在db.Query中,将SELECT * FROM your_table替换为您要导出数据的实际查询语句。

运行代码后,将会在当前目录下生成一个名为output.xlsx的Excel文件,其中包含了从数据库导出的数据。