91 lines
2.5 KiB
Go
91 lines
2.5 KiB
Go
package util
|
||
|
||
import (
|
||
"context"
|
||
"ifms/internal/config"
|
||
"ifms/pkg/oss"
|
||
"os"
|
||
|
||
"github.com/xuri/excelize/v2"
|
||
)
|
||
|
||
// ExportDataToOSS 生成带标题的 Excel 文件，上传到 OSS，并删除本地文件，返回 OSS 文件访问 URL。
|
||
// title: Excel 表头（第一行）
|
||
// data: Excel 数据内容（每一行是一个[]string）
|
||
// localFilePath: 本地临时文件路径
|
||
// ossObjectName: OSS 对象名（含路径）
|
||
// 返回 OSS 文件访问 URL 或错误
|
||
func ExportDataToOSS(ctx context.Context, title []string, data [][]string, localFilePath, ossObjectName string) (string, error) {
|
||
// 1. 生成Excel（带标题）
|
||
if err := GenerateExcelFile(title, data, localFilePath); err != nil {
|
||
return "", err
|
||
}
|
||
// 2. 上传并删除本地文件
|
||
url, err := UploadOss(ctx, localFilePath, ossObjectName, true)
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
return url, nil
|
||
}
|
||
|
||
// GenerateExcelFile 生成带标题的 Excel 文件并保存到本地。
|
||
// title: Excel 表头（第一行）
|
||
// data: Excel 数据内容（每一行是一个[]string）
|
||
// filePath: 本地保存路径
|
||
// 返回错误信息
|
||
func GenerateExcelFile(title []string, data [][]string, filePath string) error {
|
||
f := excelize.NewFile()
|
||
sheet := "Sheet1"
|
||
// 写入标题
|
||
for j, cell := range title {
|
||
cellName, _ := excelize.CoordinatesToCellName(j+1, 1)
|
||
f.SetCellValue(sheet, cellName, cell)
|
||
}
|
||
// 写入数据
|
||
for i, row := range data {
|
||
for j, cell := range row {
|
||
cellName, _ := excelize.CoordinatesToCellName(j+1, i+2) // 数据从第2行开始
|
||
f.SetCellValue(sheet, cellName, cell)
|
||
}
|
||
}
|
||
if err := f.SaveAs(filePath); err != nil {
|
||
return err
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// UploadAndDeleteLocalFile 上传本地文件到 OSS，上传成功后删除本地文件。
|
||
// filePath: 本地文件路径
|
||
// objectName: OSS 对象名（含路径）
|
||
// deleteLocalFile: 是否删除本地文件
|
||
// 返回 OSS 文件访问 URL 或错误
|
||
func UploadOss(ctx context.Context, filePath, objectName string, deleteLocalFile bool) (string, error) {
|
||
file, err := os.Open(filePath)
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
defer file.Close()
|
||
|
||
fileInfo, err := file.Stat()
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
fileSize := fileInfo.Size()
|
||
|
||
ossResult, err := oss.Ins.PutObject(ctx, config.C.AliyunOSS.Bucket, objectName, file, fileSize, oss.PutObjectOptions{
|
||
UserMetadata: map[string]string{"name": fileInfo.Name()},
|
||
})
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
|
||
// 上传成功后删除本地文件
|
||
if deleteLocalFile {
|
||
if err := os.Remove(filePath); err != nil {
|
||
return "", err
|
||
}
|
||
}
|
||
|
||
return ossResult.URL, nil
|
||
}
|