ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

Golang Gin实践 连载十一 实现导出、导入 Excel

2022-09-15 11:02:11  阅读:291  来源: 互联网

标签:return string err nil Excel Golang tag export Gin


前言

在本节,我们将实现对标签信息的导出、导入功能,这是很标配功能了,希望你掌握基础的使用方式
另外在本文我们使用了 2 个 Excel 的包,excelize 最初的 XML 格式文件的一些结构,是通过 tealeg/xlsx 格式文件结构演化而来的,因此特意在此都展示了,你可以根据自己的场景和喜爱去使用

配置

首先要指定导出的 Excel 文件的存储路径,在 app.ini 中增加配置:

[app]
...
ExportSavePath = export/

修改 setting.go 的 App struct:

type App struct {
    ...
    PrefixUrl string
    ExportSavePath string
}

在这里需增加 ExportSavePath 配置项,另外将先前 ImagePrefixUrl 改为 PrefixUrl 用于支撑两者的 HOST 获取
(注意修改 image.go 的 GetImageFullUrl 方法)

pkg

新建 pkg/export/excel.go 文件,如下:

package export

import "gin_log/pkg/setting"

func GetExcelFullUrl(name string) string {
	return setting.AppSetting.PrefixUrl + "/" + GetExcelPath() + name
}
func GetExcelPath() string {
	return setting.AppSetting.ExportSavePath
}
func GetExcelFullPath() string {
	return setting.AppSetting.RuntimeRootPath + setting.AppSetting.ExportSavePath
}

这里编写了一些常用的方法,以后取值方式如果有变动,直接改内部代码即可,对外不可见

尝试一下标准库

	_ = os.MkdirAll(export.GetExcelFullPath(), os.ModePerm)
	f, err := os.Create(export.GetExcelFullPath() + "test.csv")
	if err != nil {
		fmt.Println(err)
	}
	defer func() {
		_ = f.Close()
	}()
	//_, _ = f.WriteString("\\xEF\\xBB\\xBF")
	w := csv.NewWriter(f)
	data := [][]string{
		{"1", "马亚南", "test1-1"},
		{"2", "test2", "test2-1"},
		{"3", "test3", "test3-1"},
	}
	_ = w.WriteAll(data)

在 Go 提供的标准库 encoding/csv 中,天然的支持 csv 文件的读取和处理,在本段代码中,做了如下工作:

1、os.Create:
创建了一个 test.csv 文件

2、f.WriteString(“\xEF\xBB\xBF”):
\xEF\xBB\xBF 是 UTF-8 BOM 的 16 进制格式,在这里的用处是标识文件的编码格式,通常会出现在文件的开头,因此第一步就要将其写入。如果不标识 UTF-8 的编码格式的话,写入的汉字会显示为乱码

3、csv.NewWriter:

// NewWriter returns a new Writer that writes to w.
func NewWriter(w io.Writer) *Writer {
	return &Writer{
		Comma: ',',
		w:     bufio.NewWriter(w),
	}
}

4、w.WriteAll:

func (w *Writer) WriteAll(records [][]string) error {
	for _, record := range records {
		err := w.Write(record)
		if err != nil {
			return err
		}
	}
	return w.w.Flush()
}

WriteAll 实际是对 Write 的封装,需要注意在最后调用了 w.w.Flush(),这充分了说明了 WriteAll 的使用场景,你可以想想作者的设计用意

导出

下载操作excel的库
go get github.com/tealeg/xlsx

Service方法

打开 service/tag.go,增加 Export 方法,如下:

点击查看代码
func (t *Tag) Export() (string, error) {
	tags := models.GetTags(t.PageNum, t.PageSize, map[string]any{"name": t.Name, "state": t.State})
	if len(tags) == 0 {
		return "", errors.New("暂无导出数据")
	}

	file := xlsx.NewFile()
	sheet, err := file.AddSheet("标签信息")
	if err != nil {
		return "", err
	}

	titles := []string{"ID", "名称", "创建人", "创建时间", "修改人", "修改时间"}
	row := sheet.AddRow()

	var cell *xlsx.Cell
	for _, title := range titles {
		cell = row.AddCell()
		cell.Value = title
	}

	for _, tag := range tags {
		values := []string{
			strconv.Itoa(tag.ID),
			tag.Name,
			tag.CreatedBy,
			strconv.Itoa(int(tag.CreatedOn)),
			tag.ModifiedBy,
			strconv.Itoa(int(tag.ModifiedOn)),
		}
		row = sheet.AddRow()
		for _, value := range values {
			cell = row.AddCell()
			cell.Value = value
		}
	}

	time := strconv.Itoa(int(time2.Now().Unix()))
	filename := "tag-" + time + ".xlsx"

	fullPath := export.GetExcelFullPath() + filename
	err = file.Save(fullPath)
	if err != nil {
		return "", err
	}
	return filename, nil
}

routers 入口

打开 routers/api/v1/tag.go,增加如下方法:

// ExportTag 导出标签
func ExportTag(ctx *gin.Context) {
	appG := app.Gin{C: ctx}
	name := ctx.PostForm("name")
	state := -1
	if arg := ctx.PostForm("state"); arg != "" {
		state = com.StrTo(arg).MustInt()
	}

	tagService := cache_service.Tag{Name: name, State: state, PageNum: util.GetPage(ctx), PageSize: 1000}
	filename, err := tagService.Export()
	if err != nil {
		appG.Response(http.StatusOK, e.ERROR_EXPORT_TAG_FAIL, nil)
		return
	}

	appG.Response(http.StatusOK, e.SUCCESS, map[string]string{
		"export_url": export.GetExcelFullUrl(filename),
		"export_save_url": export.GetExcelPath() + filename,
	})
}

路由

在 routers/router.go 文件中增加路由方法,如下

{
    apiv1.POST("/tags/export", v1.ExportTag)
}

验证接口

http://127.0.0.1:8000/api/v1/tags/export

{
	"code": 200,
	"data": {
		"export_save_url": "export/tag-1663207110.xlsx",
		"export_url": "http://127.0.0.1:8000/export/tag-1663207110.xlsx"
	},
	"msg": "ok"
}

最终通过接口返回了导出文件的地址和保存地址

StaticFS

那你想想,现在直接访问地址肯定是无法下载文件的,那么该如何做呢?
打开 router.go 文件,增加代码如下:

r.StaticFS("/export", http.Dir(export.GetExcelFullPath()))

若你不理解,强烈建议温习下前面的章节,举一反三

验证下载

再次访问上面的 export_url ,如:http://127.0.0.1:8000/export/tags-1528903393.xlsx,是不是成功了呢?

导入

安装操作excel的库
go get github.com/xuri/excelize/v2

Service方法

打开 service/tag.go,增加 Import 方法,如下:

func (t *Tag) Import(r io.Reader) error {
	file, err := excelize.OpenReader(r)
	if err != nil {
		return err
	}
	rows, err := file.GetRows("标签信息")
	if err != nil {
		return err
	}
	for iRow, row := range rows {
		fmt.Println(iRow)
		if iRow > 0 {
			var data []string
			for _, cell := range row {
				data = append(data, cell)
			}
			models.AddTag(data[1], 1, data[2])
		}
	}
	return nil
}

routers入口

打开 routers/api/v1/tag.go,增加如下方法:

// ImportTag 导入标签
func ImportTag(ctx *gin.Context) {
	appG := app.Gin{C: ctx}

	file, _, err := ctx.Request.FormFile("file")
	if err != nil {
		logging.Warn(err)
		appG.Response(http.StatusOK, e.ERROR, nil)
		return
	}

	tag := cache_service.Tag{}
	err = tag.Import(file)
	if err != nil {
		logging.Warn(err)
		appG.Response(http.StatusOK, e.ERROR, nil)
		return
	}

	appG.Response(http.StatusOK, e.SUCCESS, nil)
}

路由

在 routers/router.go 文件中增加路由方法,如下

apiv1.POST("/tags/import", v1.ImportTag)

验证

http://127.0.0.1:8000/api/v1/tags/import
在这里我们将先前导出的 Excel 文件作为入参,访问 http://127.0.0.01:8000/tags/import,检查返回和数据是否正确入库

总结

在本文中,简单介绍了 Excel 的导入、导出的使用方式,使用了以下 2 个包:
https://github.com/tealeg/xlsx
https://github.com/qax-os/excelize

标签:return,string,err,nil,Excel,Golang,tag,export,Gin
来源: https://www.cnblogs.com/mayanan/p/16695251.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有