mirror of
https://github.com/flipped-aurora/gin-vue-admin.git
synced 2024-04-21 12:32:43 +00:00
6fb6ac2d6c
* 媒体库增加 普通上传、压缩上传按钮，方便媒体库直接上传图片 * 增加数据类型切换后的的校验，避免使用错误的查询条件和字典条件。 * refactor: 重构初始化逻辑 * 媒体库功能丰富 * 修复注入漏洞和路径穿越 * 修复自动化接口获取数据库表失败后未能终止的bug * 微调媒体库样式 Co-authored-by: bypanghu <bypanghu@163.com> Co-authored-by: tesun <36953434+tesun@users.noreply.github.com> Co-authored-by: pnck <hio131@gmail.com> Co-authored-by: task <121913992@qq.com>
112 lines
3.7 KiB
Go
112 lines
3.7 KiB
Go
package example
|
|
|
|
import (
|
|
"github.com/flipped-aurora/gin-vue-admin/server/global"
|
|
"github.com/flipped-aurora/gin-vue-admin/server/model/common/response"
|
|
"github.com/flipped-aurora/gin-vue-admin/server/model/example"
|
|
"github.com/gin-gonic/gin"
|
|
"go.uber.org/zap"
|
|
"os"
|
|
"strings"
|
|
)
|
|
|
|
type ExcelApi struct{}
|
|
|
|
// /excel/importExcel 接口，与upload接口作用类似，只是把文件存到resource/excel目录下，用于导入Excel时存放Excel文件(ExcelImport.xlsx)
|
|
// /excel/loadExcel接口，用于读取resource/excel目录下的文件((ExcelImport.xlsx)并加载为[]model.SysBaseMenu类型的示例数据
|
|
// /excel/exportExcel 接口，用于读取前端传来的tableData，生成Excel文件并返回
|
|
// /excel/downloadTemplate 接口，用于下载resource/excel目录下的 ExcelTemplate.xlsx 文件，作为导入的模板
|
|
|
|
// @Tags excel
|
|
// @Summary 导出Excel
|
|
// @Security ApiKeyAuth
|
|
// @accept application/json
|
|
// @Produce application/octet-stream
|
|
// @Param data body example.ExcelInfo true "导出Excel文件信息"
|
|
// @Success 200
|
|
// @Router /excel/exportExcel [post]
|
|
func (e *ExcelApi) ExportExcel(c *gin.Context) {
|
|
var excelInfo example.ExcelInfo
|
|
_ = c.ShouldBindJSON(&excelInfo)
|
|
if strings.Index(excelInfo.FileName, "..") > -1 {
|
|
response.FailWithMessage("包含非法字符", c)
|
|
return
|
|
}
|
|
filePath := global.GVA_CONFIG.Excel.Dir + excelInfo.FileName
|
|
err := excelService.ParseInfoList2Excel(excelInfo.InfoList, filePath)
|
|
if err != nil {
|
|
global.GVA_LOG.Error("转换Excel失败!", zap.Error(err))
|
|
response.FailWithMessage("转换Excel失败", c)
|
|
return
|
|
}
|
|
c.Writer.Header().Add("success", "true")
|
|
c.File(filePath)
|
|
}
|
|
|
|
// @Tags excel
|
|
// @Summary 导入Excel文件
|
|
// @Security ApiKeyAuth
|
|
// @accept multipart/form-data
|
|
// @Produce application/json
|
|
// @Param file formData file true "导入Excel文件"
|
|
// @Success 200 {object} response.Response{msg=string} "导入Excel文件"
|
|
// @Router /excel/importExcel [post]
|
|
func (e *ExcelApi) ImportExcel(c *gin.Context) {
|
|
_, header, err := c.Request.FormFile("file")
|
|
if err != nil {
|
|
global.GVA_LOG.Error("接收文件失败!", zap.Error(err))
|
|
response.FailWithMessage("接收文件失败", c)
|
|
return
|
|
}
|
|
_ = c.SaveUploadedFile(header, global.GVA_CONFIG.Excel.Dir+"ExcelImport.xlsx")
|
|
response.OkWithMessage("导入成功", c)
|
|
}
|
|
|
|
// @Tags excel
|
|
// @Summary 加载Excel数据
|
|
// @Security ApiKeyAuth
|
|
// @Produce application/json
|
|
// @Success 200 {object} response.Response{data=response.PageResult,msg=string} "加载Excel数据,返回包括列表,总数,页码,每页数量"
|
|
// @Router /excel/loadExcel [get]
|
|
func (e *ExcelApi) LoadExcel(c *gin.Context) {
|
|
menus, err := excelService.ParseExcel2InfoList()
|
|
if err != nil {
|
|
global.GVA_LOG.Error("加载数据失败!", zap.Error(err))
|
|
response.FailWithMessage("加载数据失败", c)
|
|
return
|
|
}
|
|
response.OkWithDetailed(response.PageResult{
|
|
List: menus,
|
|
Total: int64(len(menus)),
|
|
Page: 1,
|
|
PageSize: 999,
|
|
}, "加载数据成功", c)
|
|
}
|
|
|
|
// @Tags excel
|
|
// @Summary 下载模板
|
|
// @Security ApiKeyAuth
|
|
// @accept multipart/form-data
|
|
// @Produce application/json
|
|
// @Param fileName query string true "模板名称"
|
|
// @Success 200
|
|
// @Router /excel/downloadTemplate [get]
|
|
func (e *ExcelApi) DownloadTemplate(c *gin.Context) {
|
|
fileName := c.Query("fileName")
|
|
filePath := global.GVA_CONFIG.Excel.Dir + fileName
|
|
|
|
fi, err := os.Stat(filePath)
|
|
if err != nil {
|
|
global.GVA_LOG.Error("文件不存在!", zap.Error(err))
|
|
response.FailWithMessage("文件不存在", c)
|
|
return
|
|
}
|
|
if fi.IsDir() {
|
|
global.GVA_LOG.Error("不支持下载文件夹!", zap.Error(err))
|
|
response.FailWithMessage("不支持下载文件夹", c)
|
|
return
|
|
}
|
|
c.Writer.Header().Add("success", "true")
|
|
c.File(filePath)
|
|
}
|