283 lines
7.4 KiB
Go
283 lines
7.4 KiB
Go
|
package test
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"fmt"
|
|||
|
|
"github.com/xuri/excelize/v2"
|
|||
|
|
"ifms/pkg/convert"
|
|||
|
|
"ifms/pkg/oss"
|
|||
|
|
"log"
|
|||
|
|
"os"
|
|||
|
|
"reflect"
|
|||
|
|
"testing"
|
|||
|
|
"time"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func TestExcel(t *testing.T) {
|
|||
|
|
f := excelize.NewFile()
|
|||
|
|
defer func() {
|
|||
|
|
if err := f.Close(); err != nil {
|
|||
|
|
fmt.Println(err)
|
|||
|
|
}
|
|||
|
|
}()
|
|||
|
|
// Upload a new sheet.
|
|||
|
|
index, err := f.NewSheet("Sheet2")
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println(err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
// Set value of a cell.
|
|||
|
|
f.SetCellValue("Sheet2", "A2", "Hello world.")
|
|||
|
|
f.SetCellValue("Sheet1", "B2", 100)
|
|||
|
|
// Set active sheet of the workbook.
|
|||
|
|
f.SetActiveSheet(index)
|
|||
|
|
// Save spreadsheet by the given path.
|
|||
|
|
if err := f.SaveAs("Book1.xlsx"); err != nil {
|
|||
|
|
fmt.Println(err)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func TestExcel2(t *testing.T) {
|
|||
|
|
// 模拟查询出来的列表数据
|
|||
|
|
listData := [][]string{
|
|||
|
|
{"姓名", "年龄", "邮箱"}, // 表头
|
|||
|
|
{"张三", "25", "zhangsan@example.com"},
|
|||
|
|
{"李四", "30", "lisi@example.com"},
|
|||
|
|
{"王五", "28", "wangwu@example.com"},
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 创建一个新的 Excel 文件
|
|||
|
|
f := excelize.NewFile()
|
|||
|
|
// 创建一个工作表
|
|||
|
|
sheetName := "Sheet1"
|
|||
|
|
index, _ := f.NewSheet(sheetName)
|
|||
|
|
f.SetActiveSheet(index)
|
|||
|
|
|
|||
|
|
// 将数据写入 Excel
|
|||
|
|
for rowIdx, rowData := range listData {
|
|||
|
|
for colIdx, cellValue := range rowData {
|
|||
|
|
// 将列索引转换为字母，例如 0 -> 'A', 1 -> 'B', ...
|
|||
|
|
colLetter := string(rune('A' + colIdx))
|
|||
|
|
// 设置单元格的值
|
|||
|
|
cellName := fmt.Sprintf("%s%d", colLetter, rowIdx+1)
|
|||
|
|
f.SetCellValue(sheetName, cellName, cellValue)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 保存 Excel 文件
|
|||
|
|
if err := f.SaveAs("output.xlsx"); err != nil {
|
|||
|
|
fmt.Println("Error saving Excel file:", err)
|
|||
|
|
}
|
|||
|
|
fmt.Println("Excel file created successfully.")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 定义一个结构体，并在字段中添加 excel 标签
|
|||
|
|
type Person struct {
|
|||
|
|
Name string `excel:"姓名"`
|
|||
|
|
Age int `excel:"年龄"`
|
|||
|
|
Email string `excel:"邮箱"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func TestExcel3(t *testing.T) {
|
|||
|
|
// 创建一个包含数据的结构体切片
|
|||
|
|
people := []Person{
|
|||
|
|
{"张三", 25, "zhangsan@example.com"},
|
|||
|
|
{"李四", 30, "lisi@example.com"},
|
|||
|
|
{"王五", 28, "wangwu@example.com"},
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 创建一个新的 Excel 文件
|
|||
|
|
f := excelize.NewFile()
|
|||
|
|
sheetName := "Sheet1"
|
|||
|
|
index, _ := f.NewSheet(sheetName)
|
|||
|
|
f.SetActiveSheet(index)
|
|||
|
|
|
|||
|
|
// 获取结构体的类型
|
|||
|
|
structType := reflect.TypeOf(people[0])
|
|||
|
|
|
|||
|
|
// 写入表头
|
|||
|
|
for colIdx := 0; colIdx < structType.NumField(); colIdx++ {
|
|||
|
|
field := structType.Field(colIdx)
|
|||
|
|
colLetter := string(rune('A' + colIdx))
|
|||
|
|
err := f.SetCellValue(sheetName, fmt.Sprintf("%s1", colLetter), field.Tag.Get("excel"))
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println("Error setting Excel header:", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 写入数据行
|
|||
|
|
for rowIdx, person := range people {
|
|||
|
|
rowNum := rowIdx + 2 // 从第二行开始写入数据（第一行是表头）
|
|||
|
|
for colIdx := 0; colIdx < structType.NumField(); colIdx++ {
|
|||
|
|
colLetter := string(rune('A' + colIdx))
|
|||
|
|
fieldValue := reflect.ValueOf(person).Field(colIdx).Interface()
|
|||
|
|
err := f.SetCellValue(sheetName, fmt.Sprintf("%s%d", colLetter, rowNum), fieldValue)
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println("Error setting Excel header:", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 保存 Excel 文件
|
|||
|
|
if err := f.SaveAs("output_reflect.xlsx"); err != nil {
|
|||
|
|
fmt.Println("Error saving Excel file:", err)
|
|||
|
|
}
|
|||
|
|
fmt.Println("Excel file created successfully.")
|
|||
|
|
|
|||
|
|
ctx := context.Background()
|
|||
|
|
file, err := os.Open("/Users/shaolingjin/source/qk/ifms/test/output_reflect.xlsx")
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println("Failed to open file:", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
defer file.Close()
|
|||
|
|
|
|||
|
|
//获取文件大小
|
|||
|
|
fileInfo, err := file.Stat()
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println("Failed to get file size:", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
fileSize := fileInfo.Size()
|
|||
|
|
|
|||
|
|
name := oss.FormatObjectName("test", "output_reflect.xlsx")
|
|||
|
|
result, err := oss.Ins.PutObject(ctx, "qkkj-ifms-test", name, file, fileSize, oss.PutObjectOptions{
|
|||
|
|
ContentType: "text/plain",
|
|||
|
|
})
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println("Failed to upload object:", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
fmt.Println(result)
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 定义一个结构体，并在字段中添加 excel 和 excel_type 标签
|
|||
|
|
type P struct {
|
|||
|
|
Name string `excel:"姓名" excel_type:"string"`
|
|||
|
|
Age int `excel:"年龄" excel_type:"number"`
|
|||
|
|
Email string `excel:"邮箱" excel_type:"string"`
|
|||
|
|
Birth time.Time `excel:"出生日期" excel_type:"date"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func TestExcel4(t *testing.T) {
|
|||
|
|
// 创建一个包含数据的结构体切片
|
|||
|
|
people := []P{
|
|||
|
|
{"张三", 25, "zhangsan@example.com", time.Now()},
|
|||
|
|
{"李四", 30, "lisi@example.com", time.Now()},
|
|||
|
|
{"王五", 28, "wangwu@example.com", time.Now()},
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 创建一个新的 Excel 文件
|
|||
|
|
f := excelize.NewFile()
|
|||
|
|
sheetName := "Sheet1"
|
|||
|
|
index, _ := f.NewSheet(sheetName)
|
|||
|
|
f.SetActiveSheet(index)
|
|||
|
|
|
|||
|
|
// 获取结构体的类型
|
|||
|
|
structType := reflect.TypeOf(people[0])
|
|||
|
|
|
|||
|
|
// 写入表头
|
|||
|
|
for colIdx := 0; colIdx < structType.NumField(); colIdx++ {
|
|||
|
|
field := structType.Field(colIdx)
|
|||
|
|
colLetter := string(rune('A' + colIdx))
|
|||
|
|
err := f.SetCellValue(sheetName, fmt.Sprintf("%s1", colLetter), field.Tag.Get("excel"))
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println("Error setting Excel header:", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 写入数据行
|
|||
|
|
for rowIdx, person := range people {
|
|||
|
|
rowNum := rowIdx + 2 // 从第二行开始写入数据（第一行是表头）
|
|||
|
|
for colIdx := 0; colIdx < structType.NumField(); colIdx++ {
|
|||
|
|
colLetter := string(rune('A' + colIdx))
|
|||
|
|
fieldValue := reflect.ValueOf(person).Field(colIdx).Interface()
|
|||
|
|
fieldType := structType.Field(colIdx).Tag.Get("excel_type")
|
|||
|
|
|
|||
|
|
f.SetCellValue(sheetName, fmt.Sprintf("%s%d", colLetter, rowNum), fieldValue)
|
|||
|
|
|
|||
|
|
// 根据 excel_type 标签设置单元格格式
|
|||
|
|
switch fieldType {
|
|||
|
|
case "number":
|
|||
|
|
f.SetCellInt(sheetName, fmt.Sprintf("%s%d", colLetter, rowNum), int64(fieldValue.(int)))
|
|||
|
|
case "date":
|
|||
|
|
|
|||
|
|
f.SetCellValue(sheetName, fmt.Sprintf("%s%d", colLetter, rowNum), fieldValue.(string))
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 设置日期格式
|
|||
|
|
style, _ := f.NewStyle(&excelize.Style{
|
|||
|
|
NumFmt: 51,
|
|||
|
|
})
|
|||
|
|
for _, pe := range people {
|
|||
|
|
rowNum := pe.Age + 2
|
|||
|
|
colLetter := string(rune('D' + 0)) // 假设日期列是第四列（D）
|
|||
|
|
f.SetCellStyle(sheetName, fmt.Sprintf("%s%d", colLetter, rowNum), fmt.Sprintf("%s%d", colLetter, rowNum), style)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 保存 Excel 文件
|
|||
|
|
if err := f.SaveAs("output_123.xlsx"); err != nil {
|
|||
|
|
fmt.Println("Error saving Excel file:", err)
|
|||
|
|
}
|
|||
|
|
fmt.Println("Excel file created successfully.")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func TestExcelInput(t *testing.T) {
|
|||
|
|
|
|||
|
|
// 打开 Excel 文件
|
|||
|
|
f, err := excelize.OpenFile("output_123.xlsx")
|
|||
|
|
if err != nil {
|
|||
|
|
log.Fatalf("打开 Excel 文件失败: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取所有 sheet 名称
|
|||
|
|
sheetNames := f.GetSheetMap()
|
|||
|
|
for _, sheetName := range sheetNames {
|
|||
|
|
// 获取指定 sheet 的行数和列数
|
|||
|
|
rows, err := f.GetRows(sheetName)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Fatalf("获取 sheet 数据失败: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 跳过表头行
|
|||
|
|
if len(rows) > 0 {
|
|||
|
|
rows = rows[1:]
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 遍历每一行数据
|
|||
|
|
for _, row := range rows {
|
|||
|
|
// 解析每一行数据到 Factory 结构体
|
|||
|
|
if row[0] == "" {
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
var p P
|
|||
|
|
|
|||
|
|
// 这里根据实际 Excel 表格的列顺序和数据类型进行解析
|
|||
|
|
// 假设 Excel 表格的第一列是 ID，第二列是 TenantId 等
|
|||
|
|
//if id, err := strconv.ParseInt(row[0], 10, 64); err == nil {
|
|||
|
|
// p.ID = id
|
|||
|
|
//}
|
|||
|
|
p.Name = row[0]
|
|||
|
|
p.Age = convert.ToInt(row[1])
|
|||
|
|
p.Email = row[2]
|
|||
|
|
|
|||
|
|
// 处理时间字段
|
|||
|
|
if dt, err := time.Parse(time.DateOnly, row[3]); err == nil {
|
|||
|
|
p.Birth = dt
|
|||
|
|
} else {
|
|||
|
|
println("时间解析失败", err)
|
|||
|
|
}
|
|||
|
|
// 打印解析后的数据
|
|||
|
|
fmt.Printf("P: %+v\n", p)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
}
|