622 lines
20 KiB
Go
622 lines
20 KiB
Go
package main
|
||
|
||
//import (
|
||
// "fmt"
|
||
// "github.com/xuri/excelize/v2"
|
||
// "os"
|
||
// "time"
|
||
//)
|
||
//
|
||
//// ============ main函数 ============
|
||
//
|
||
//func main() {
|
||
// // 创建Excel管理器
|
||
// excelMgr := NewExcelManager()
|
||
// defer excelMgr.CloseAll()
|
||
//
|
||
// fmt.Println("=== Excel文件操作工具 ===")
|
||
// fmt.Println()
|
||
//
|
||
// // 1. 创建测试文件
|
||
// fmt.Println("=== 步骤1: 创建测试文件 ===")
|
||
// createTestFiles()
|
||
// fmt.Println()
|
||
//
|
||
// // 2. 测试基本功能
|
||
// fmt.Println("=== 步骤2: 测试基本功能 ===")
|
||
// testBasicFunctions(excelMgr)
|
||
// fmt.Println()
|
||
//
|
||
// // 3. 测试WriteData函数
|
||
// fmt.Println("=== 步骤3: 测试WriteData函数 ===")
|
||
// testWriteDataFunction(excelMgr)
|
||
// fmt.Println()
|
||
//
|
||
// // 4. 测试SearchRowData函数
|
||
// fmt.Println("=== 步骤4: 测试SearchRowData函数 ===")
|
||
// testSearchRowDataFunction(excelMgr)
|
||
// fmt.Println()
|
||
// //
|
||
// //// 5. 测试CreateAndWrite函数
|
||
// fmt.Println("=== 步骤5: 测试CreateAndWrite函数 ===")
|
||
// testCreateAndWriteFunction(excelMgr)
|
||
// fmt.Println()
|
||
//
|
||
// //// 3. 测试合并功能（包含源文件和序号）
|
||
// //fmt.Println("=== 步骤3: 测试合并功能（包含源文件和序号）===")
|
||
// //start := time.Now()
|
||
// //mergeConfig := MergeConfig{
|
||
// // SourceDir: ".",
|
||
// // OutputFile: "excel/merged_with_info.xlsx",
|
||
// // SheetName: "合并数据",
|
||
// // SourceSheet: "员工数据",
|
||
// // IncludeHeaders: true,
|
||
// // SkipEmptyRows: false,
|
||
// // FilePattern: "excel/data*.xlsx",
|
||
// // AddSourceColumn: false,
|
||
// // AddIndexColumn: false,
|
||
// //}
|
||
// //
|
||
// //err := excelMgr.MergeExcelFiles(mergeConfig)
|
||
// //if err != nil {
|
||
// // fmt.Printf("合并失败: %v\n", err)
|
||
// //} else {
|
||
// // elapsed := time.Since(start)
|
||
// // fmt.Printf("合并完成，耗时: %v\n", elapsed)
|
||
// //}
|
||
// //fmt.Println()
|
||
//
|
||
// //// 4. 测试并行合并
|
||
// //fmt.Println("=== 步骤4: 测试并行合并 ===")
|
||
// //start = time.Now()
|
||
// //parallelConfig := MergeConfig{
|
||
// // SourceDir: ".",
|
||
// // OutputFile: "excel/merged_parallel.xlsx",
|
||
// // SheetName: "并行合并数据",
|
||
// // SourceSheet: "员工数据",
|
||
// // IncludeHeaders: true,
|
||
// // SkipEmptyRows: false,
|
||
// // FilePattern: "excel/data*.xlsx",
|
||
// // AddSourceColumn: false,
|
||
// // AddIndexColumn: false,
|
||
// //}
|
||
// //
|
||
// //err = excelMgr.MergeExcelFilesParallel(parallelConfig, 2)
|
||
// //if err != nil {
|
||
// // fmt.Printf("并行合并失败: %v\n", err)
|
||
// //} else {
|
||
// // elapsed := time.Since(start)
|
||
// // fmt.Printf("并行合并完成，耗时: %v\n", elapsed)
|
||
// //}
|
||
//
|
||
// //// 5. 测试按列合并
|
||
// //fmt.Println("=== 步骤5: 测试按列合并 ===")
|
||
// //start = time.Now()
|
||
// //columnConfig := MergeConfig{
|
||
// // SourceDir: ".",
|
||
// // OutputFile: "merged_by_column.xlsx",
|
||
// // SheetName: "按列合并",
|
||
// // SourceSheet: "员工数据",
|
||
// // IncludeHeaders: false,
|
||
// // MergeByColumn: true,
|
||
// // FilePattern: "data*.xlsx",
|
||
// //}
|
||
// //
|
||
// //err = excelMgr.MergeByColumn(columnConfig)
|
||
// //if err != nil {
|
||
// // fmt.Printf("按列合并失败: %v\n", err)
|
||
// //} else {
|
||
// // elapsed := time.Since(start)
|
||
// // fmt.Printf("按列合并完成，耗时: %v\n", elapsed)
|
||
// //}
|
||
// //fmt.Println()
|
||
//
|
||
// // 6. 测试合并指定文件
|
||
// fmt.Println("=== 步骤6: 测试合并指定文件 ===")
|
||
// start := time.Now()
|
||
// specificConfig := MergeConfig{
|
||
// SpecificFiles: []string{"excel/data1.xlsx", "excel/data3.xlsx"},
|
||
// OutputFile: "excel/merged_specific.xlsx",
|
||
// SheetName: "指定文件合并",
|
||
// SourceSheet: "员工数据",
|
||
// IncludeHeaders: true,
|
||
// SkipEmptyRows: false,
|
||
// AddSourceColumn: false,
|
||
// AddIndexColumn: false,
|
||
// }
|
||
//
|
||
// err := excelMgr.MergeExcelFiles(specificConfig)
|
||
// if err != nil {
|
||
// fmt.Printf("指定文件合并失败: %v\n", err)
|
||
// } else {
|
||
// elapsed := time.Since(start)
|
||
// fmt.Printf("指定文件合并完成，耗时: %v\n", elapsed)
|
||
// }
|
||
// fmt.Println()
|
||
//
|
||
// //// 7. 测试合并多sheet文件
|
||
// //fmt.Println("=== 步骤7: 测试合并多sheet文件 ===")
|
||
// //err = createMultiSheetFile()
|
||
// //if err != nil {
|
||
// // fmt.Printf("创建多sheet文件失败: %v\n", err)
|
||
// // return
|
||
// //}
|
||
// //
|
||
// //start = time.Now()
|
||
// //err = excelMgr.MergeSheetsInSameFile("multi_sheet.xlsx", "merged_sheets.xlsx", "所有Sheet数据")
|
||
// //if err != nil {
|
||
// // fmt.Printf("合并sheet失败: %v\n", err)
|
||
// //} else {
|
||
// // elapsed := time.Since(start)
|
||
// // fmt.Printf("合并sheet完成，耗时: %v\n", elapsed)
|
||
// //}
|
||
// //fmt.Println()
|
||
//
|
||
// // 8. 清理测试文件
|
||
// //fmt.Println("=== 步骤8: 清理测试文件 ===")
|
||
// //cleanupTestFiles()
|
||
//}
|
||
//
|
||
//// 测试基本功能
|
||
//func testBasicFunctions(em *ExcelManager) {
|
||
// // 测试读取数据
|
||
// rows, err := em.ReadData("excel/data1.xlsx", "员工数据")
|
||
// if err != nil {
|
||
// fmt.Printf("读取数据失败: %v\n", err)
|
||
// return
|
||
// }
|
||
// fmt.Printf("data1.xlsx 有 %d 行数据\n", len(rows))
|
||
//
|
||
// // 测试搜索功能
|
||
// results, err := em.SearchByKeyword("excel/data1.xlsx", "员工数据", "员工")
|
||
// if err != nil {
|
||
// fmt.Printf("搜索失败: %v\n", err)
|
||
// } else {
|
||
// fmt.Printf("找到包含'员工'的 %d 个结果\n", len(results))
|
||
// }
|
||
//
|
||
// // 测试追加数据
|
||
// newRow := []interface{}{999, "测试员工", 30, "测试部", "2023-12-01", 20000}
|
||
// err = em.AppendData("excel/data1.xlsx", "员工数据", newRow)
|
||
// if err != nil {
|
||
// fmt.Printf("追加数据失败: %v\n", err)
|
||
// } else {
|
||
// fmt.Println("追加数据成功")
|
||
// }
|
||
//}
|
||
//
|
||
////// 测试WriteData函数 - 从最大行后面加入新数据
|
||
////func testWriteDataFunction(em *ExcelManager) {
|
||
//// // 1. 首先读取现有文件获取最大行数
|
||
//// fmt.Println("1. 读取现有文件获取最大行数:")
|
||
//// rows, err := em.ReadData("excel/data1.xlsx", "员工数据")
|
||
//// if err != nil {
|
||
//// fmt.Printf("读取数据失败: %v\n", err)
|
||
//// return
|
||
//// }
|
||
////
|
||
//// maxRow := len(rows)
|
||
//// fmt.Printf(" 文件当前有 %d 行数据\n", maxRow)
|
||
////
|
||
//// // 显示最后几行数据
|
||
//// if maxRow > 0 {
|
||
//// showRows := 3
|
||
//// if maxRow < showRows {
|
||
//// showRows = maxRow
|
||
//// }
|
||
//// fmt.Printf(" 最后%d行数据预览:\n", showRows)
|
||
//// for i := maxRow - showRows; i < maxRow; i++ {
|
||
//// fmt.Printf(" 第%d行: %v\n", i+1, rows[i])
|
||
//// }
|
||
//// }
|
||
////
|
||
//// // 2. 在最大行后面添加新数据
|
||
//// fmt.Println("\n2. 在最大行后面添加新数据:")
|
||
////
|
||
//// // 计算新数据开始的行号
|
||
//// // 注意：Excel行号从1开始，但rows长度已经包含了所有行
|
||
//// startRow := maxRow + 1
|
||
////
|
||
//// // 准备要添加的新数据（从startRow开始）
|
||
//// newData := make(map[string]interface{})
|
||
////
|
||
//// // 添加第1条新记录（在startRow行）
|
||
//// newData[fmt.Sprintf("A%d", startRow)] = maxRow + 1
|
||
//// newData[fmt.Sprintf("B%d", startRow)] = "新增员工1"
|
||
//// newData[fmt.Sprintf("C%d", startRow)] = 35
|
||
//// newData[fmt.Sprintf("D%d", startRow)] = "研发部"
|
||
//// newData[fmt.Sprintf("E%d", startRow)] = time.Now().Format("2006-01-02")
|
||
//// newData[fmt.Sprintf("F%d", startRow)] = 25000
|
||
////
|
||
//// // 添加第2条新记录（在startRow+1行）
|
||
//// newData[fmt.Sprintf("A%d", startRow+1)] = maxRow + 2
|
||
//// newData[fmt.Sprintf("B%d", startRow+1)] = "新增员工2"
|
||
//// newData[fmt.Sprintf("C%d", startRow+1)] = 28
|
||
//// newData[fmt.Sprintf("D%d", startRow+1)] = "测试部"
|
||
//// newData[fmt.Sprintf("E%d", startRow+1)] = time.Now().Format("2006-01-02")
|
||
//// newData[fmt.Sprintf("F%d", startRow+1)] = 18000
|
||
////
|
||
//// // 添加第3条新记录（在startRow+2行）
|
||
//// newData[fmt.Sprintf("A%d", startRow+2)] = maxRow + 3
|
||
//// newData[fmt.Sprintf("B%d", startRow+2)] = "新增员工3"
|
||
//// newData[fmt.Sprintf("C%d", startRow+2)] = 32
|
||
//// newData[fmt.Sprintf("D%d", startRow+2)] = "运维部"
|
||
//// newData[fmt.Sprintf("E%d", startRow+2)] = time.Now().Format("2006-01-02")
|
||
//// newData[fmt.Sprintf("F%d", startRow+2)] = 22000
|
||
////
|
||
//// fmt.Printf(" 将在第%d行开始添加3条新记录\n", startRow)
|
||
////
|
||
//// // 使用WriteData写入新数据
|
||
//// err = em.WriteData("excel/data1.xlsx", "员工数据", newData)
|
||
//// if err != nil {
|
||
//// fmt.Printf("写入新数据失败: %v\n", err)
|
||
//// return
|
||
//// }
|
||
////
|
||
//// fmt.Println(" 写入新数据成功")
|
||
////
|
||
//// // 3. 验证添加结果
|
||
//// fmt.Println("\n3. 验证添加结果:")
|
||
////
|
||
//// // 重新读取文件
|
||
//// rows, err = em.ReadData("excel/data1.xlsx", "员工数据")
|
||
//// if err != nil {
|
||
//// fmt.Printf("重新读取数据失败: %v\n", err)
|
||
//// return
|
||
//// }
|
||
////
|
||
//// newMaxRow := len(rows)
|
||
//// fmt.Printf(" 添加后文件有 %d 行数据，增加了 %d 行\n", newMaxRow, newMaxRow-maxRow)
|
||
////
|
||
//// // 显示新增的几行数据
|
||
//// if newMaxRow > maxRow {
|
||
//// addedRows := newMaxRow - maxRow
|
||
//// fmt.Printf(" 新增的%d行数据:\n", addedRows)
|
||
//// for i := maxRow; i < newMaxRow; i++ {
|
||
//// fmt.Printf(" 第%d行: %v\n", i+1, rows[i])
|
||
//// }
|
||
//// }
|
||
////
|
||
//// // 4. 测试在空文件的最大行后添加数据
|
||
//// fmt.Println("\n4. 测试在空文件的最大行后添加数据:")
|
||
////
|
||
//// // 创建一个新的空Excel文件
|
||
//// emptyData := [][]interface{}{
|
||
//// {"ID", "Name", "Value"}, // 只有表头
|
||
//// }
|
||
////
|
||
//// // 先创建只有表头的文件
|
||
//// emptyErr := em.CreateAndWrite("excel/empty_test.xlsx", "测试数据", emptyData)
|
||
//// if emptyErr != nil {
|
||
//// fmt.Printf("创建空文件失败: %v\n", emptyErr)
|
||
//// } else {
|
||
//// fmt.Println(" 创建空文件成功")
|
||
////
|
||
//// // 读取空文件
|
||
//// emptyRows, readErr := em.ReadData("excel/empty_test.xlsx", "测试数据")
|
||
//// if readErr != nil {
|
||
//// fmt.Printf("读取空文件失败: %v\n", readErr)
|
||
//// } else {
|
||
//// emptyMaxRow := len(emptyRows)
|
||
//// fmt.Printf(" 空文件有 %d 行数据\n", emptyMaxRow)
|
||
////
|
||
//// // 在空文件的最大行后添加数据
|
||
//// emptyNewData := make(map[string]interface{})
|
||
////
|
||
//// // 由于只有表头，所以从第2行开始添加
|
||
//// emptyStartRow := emptyMaxRow + 1
|
||
////
|
||
//// // 添加测试数据
|
||
//// emptyNewData[fmt.Sprintf("A%d", emptyStartRow)] = 1
|
||
//// emptyNewData[fmt.Sprintf("B%d", emptyStartRow)] = "测试项目1"
|
||
//// emptyNewData[fmt.Sprintf("C%d", emptyStartRow)] = 100.5
|
||
////
|
||
//// emptyNewData[fmt.Sprintf("A%d", emptyStartRow+1)] = 2
|
||
//// emptyNewData[fmt.Sprintf("B%d", emptyStartRow+1)] = "测试项目2"
|
||
//// emptyNewData[fmt.Sprintf("C%d", emptyStartRow+1)] = 200.75
|
||
////
|
||
//// fmt.Printf(" 将在空文件的第%d行开始添加2条记录\n", emptyStartRow)
|
||
////
|
||
//// // 写入数据
|
||
//// writeErr := em.WriteData("excel/empty_test.xlsx", "测试数据", emptyNewData)
|
||
//// if writeErr != nil {
|
||
//// fmt.Printf(" 向空文件写入数据失败: %v\n", writeErr)
|
||
//// } else {
|
||
//// fmt.Println(" 向空文件写入数据成功")
|
||
////
|
||
//// // 验证结果
|
||
//// finalRows, finalErr := em.ReadData("excel/empty_test.xlsx", "测试数据")
|
||
//// if finalErr != nil {
|
||
//// fmt.Printf(" 验证数据失败: %v\n", finalErr)
|
||
//// } else {
|
||
//// finalRowCount := len(finalRows)
|
||
//// fmt.Printf(" 最终文件有 %d 行数据:\n", finalRowCount)
|
||
////
|
||
//// for i, row := range finalRows {
|
||
//// fmt.Printf(" 第%d行: %v\n", i+1, row)
|
||
//// }
|
||
//// }
|
||
//// }
|
||
//// }
|
||
//// }
|
||
////
|
||
//// // 5. 测试替换现有行的数据
|
||
//// fmt.Println("\n5. 测试替换现有行的数据:")
|
||
////
|
||
//// // 替换第2行的部分数据
|
||
//// replaceData := map[string]interface{}{
|
||
//// "B2": "修改后的姓名",
|
||
//// "D2": "修改后的部门",
|
||
//// "F2": 30000, // 修改薪资
|
||
//// }
|
||
////
|
||
//// fmt.Println(" 将修改第2行的数据:")
|
||
//// fmt.Println(" B2: '修改后的姓名'")
|
||
//// fmt.Println(" D2: '修改后的部门'")
|
||
//// fmt.Println(" F2: 30000")
|
||
////
|
||
//// replaceErr := em.WriteData("excel/data1.xlsx", "员工数据", replaceData)
|
||
//// if replaceErr != nil {
|
||
//// fmt.Printf(" 修改数据失败: %v\n", replaceErr)
|
||
//// } else {
|
||
//// fmt.Println(" 修改数据成功")
|
||
////
|
||
//// // 验证修改结果
|
||
//// verifyRows, verifyErr := em.ReadData("excel/data1.xlsx", "员工数据")
|
||
//// if verifyErr != nil {
|
||
//// fmt.Printf(" 验证修改失败: %v\n", verifyErr)
|
||
//// } else if len(verifyRows) >= 2 {
|
||
//// fmt.Printf(" 修改后的第2行数据: %v\n", verifyRows[1])
|
||
//// }
|
||
//// }
|
||
////}
|
||
//
|
||
//// 测试SearchRowData函数
|
||
//func testSearchRowDataFunction(em *ExcelManager) {
|
||
// fmt.Println("1. 测试搜索包含'员工'的行:")
|
||
// rows, err := em.SearchRowData("excel/data1.xlsx", "员工数据", "员工")
|
||
// if err != nil {
|
||
// fmt.Printf("搜索行数据失败: %v\n", err)
|
||
// } else {
|
||
// fmt.Printf("找到 %d 行包含'员工':\n", len(rows))
|
||
// for i, row := range rows {
|
||
// fmt.Printf(" 第%d行: %v\n", i+1, row)
|
||
// }
|
||
// }
|
||
//
|
||
// fmt.Println("\n2. 测试搜索包含'技术部'的行:")
|
||
// rows, err = em.SearchRowData("excel/data1.xlsx", "员工数据", "技术部")
|
||
// if err != nil {
|
||
// fmt.Printf("搜索行数据失败: %v\n", err)
|
||
// } else {
|
||
// fmt.Printf("找到 %d 行包含'技术部':\n", len(rows))
|
||
// for i, row := range rows {
|
||
// fmt.Printf(" 第%d行: %v\n", i+1, row)
|
||
// }
|
||
// }
|
||
//
|
||
// fmt.Println("\n3. 测试搜索包含'30000'的行:")
|
||
// rows, err = em.SearchRowData("excel/data1.xlsx", "员工数据", "30000")
|
||
// if err != nil {
|
||
// fmt.Printf("搜索行数据失败: %v\n", err)
|
||
// } else {
|
||
// fmt.Printf("找到 %d 行包含'30000':\n", len(rows))
|
||
// for i, row := range rows {
|
||
// fmt.Printf(" 第%d行: %v\n", i+1, row)
|
||
// }
|
||
// }
|
||
//
|
||
// fmt.Println("\n4. 测试搜索不存在的关键词:")
|
||
// rows, err = em.SearchRowData("excel/data1.xlsx", "员工数据", "不存在的关键词")
|
||
// if err != nil {
|
||
// fmt.Printf("搜索行数据失败: %v\n", err)
|
||
// } else {
|
||
// fmt.Printf("找到 %d 行包含'不存在的关键词'\n", len(rows))
|
||
// if len(rows) == 0 {
|
||
// fmt.Println(" (正确: 没有找到匹配的行)")
|
||
// }
|
||
// }
|
||
//}
|
||
//
|
||
//// 测试CreateAndWrite函数
|
||
//func testCreateAndWriteFunction(em *ExcelManager) {
|
||
// fmt.Println("1. 测试创建新文件并写入数据:")
|
||
//
|
||
// //// 准备测试数据
|
||
// //testData := [][]string{}{
|
||
// // {"序号", "产品名称", "价格", "库存", "类别"},
|
||
// // {1, "笔记本电脑", 6999.99, 50, "电子产品"},
|
||
// // {2, "智能手机", 3999.99, 100, "电子产品"},
|
||
// // {3, "办公椅", 899.99, 30, "办公家具"},
|
||
// // {4, "台灯", 199.99, 80, "家居用品"},
|
||
// // {5, "书籍", 59.99, 200, "文化用品"},
|
||
// // {6, "水杯", 39.99, 150, "日用品"},
|
||
// // {7, "背包", 299.99, 60, "箱包"},
|
||
// // {8, "鼠标", 99.99, 120, "电子产品"},
|
||
// // {9, "键盘", 199.99, 70, "电子产品"},
|
||
// // {10, "显示器", 1299.99, 40, "电子产品"},
|
||
// //}
|
||
//
|
||
// err := em.CreateAndWrite("excel/test_create.xlsx", "产品数据", testData)
|
||
// if err != nil {
|
||
// fmt.Printf("创建并写入文件失败: %v\n", err)
|
||
// } else {
|
||
// fmt.Println("创建并写入文件成功")
|
||
//
|
||
// // 验证文件
|
||
// if _, err := os.Stat("excel/test_create.xlsx"); err == nil {
|
||
// fmt.Println("文件创建成功: excel/test_create.xlsx")
|
||
//
|
||
// rows, err := em.ReadData("excel/test_create.xlsx", "产品数据")
|
||
// if err != nil {
|
||
// fmt.Printf("读取文件失败: %v\n", err)
|
||
// } else {
|
||
// fmt.Printf("文件有 %d 行数据:\n", len(rows))
|
||
//
|
||
// // 显示前5行
|
||
// limit := 5
|
||
// if len(rows) < limit {
|
||
// limit = len(rows)
|
||
// }
|
||
//
|
||
// fmt.Println(" 前5行数据:")
|
||
// for i := 0; i < limit; i++ {
|
||
// fmt.Printf(" 第%d行: %v\n", i+1, rows[i])
|
||
// }
|
||
//
|
||
// if len(rows) > limit {
|
||
// fmt.Printf(" ... 还有 %d 行数据\n", len(rows)-limit)
|
||
// }
|
||
// }
|
||
//
|
||
// // 测试搜索功能
|
||
// fmt.Println("\n2. 在新建文件中测试搜索:")
|
||
// searchResults, err := em.SearchByKeyword("excel/test_create.xlsx", "产品数据", "电子")
|
||
// if err != nil {
|
||
// fmt.Printf("搜索失败: %v\n", err)
|
||
// } else {
|
||
// fmt.Printf("找到 %d 个包含'电子'的单元格:\n", len(searchResults))
|
||
// for i, result := range searchResults {
|
||
// fmt.Printf(" %d. %s\n", i+1, result)
|
||
// }
|
||
// }
|
||
//
|
||
// // 测试行搜索
|
||
// fmt.Println("\n3. 在新建文件中测试行搜索:")
|
||
// rowResults, err := em.SearchRowData("excel/test_create.xlsx", "产品数据", "电子")
|
||
// if err != nil {
|
||
// fmt.Printf("行搜索失败: %v\n", err)
|
||
// } else {
|
||
// fmt.Printf("找到 %d 行包含'电子':\n", len(rowResults))
|
||
// for i, row := range rowResults {
|
||
// fmt.Printf(" 第%d行: %v\n", i+1, row)
|
||
// }
|
||
// }
|
||
// }
|
||
// }
|
||
//}
|
||
//
|
||
//// 创建测试文件
|
||
//func createTestFiles() {
|
||
// testFiles := []string{
|
||
// "excel/data1.xlsx",
|
||
// "excel/data2.xlsx",
|
||
// "excel/data3.xlsx",
|
||
// }
|
||
//
|
||
// for i, filename := range testFiles {
|
||
// data := [][]interface{}{
|
||
// {"ID", "姓名", "年龄", "部门", "入职日期", "薪资"},
|
||
// {i*100 + 1, fmt.Sprintf("员工%d", i*3+1), 25 + i, "技术部", "2023-01-15", 15000 + i*1000},
|
||
// {i*100 + 2, fmt.Sprintf("员工%d", i*3+2), 28 + i, "市场部", "2023-02-20", 12000 + i*1000},
|
||
// {i*100 + 3, fmt.Sprintf("员工%d", i*3+3), 32 + i, "销售部", "2023-03-10", 18000 + i*1000},
|
||
// {i*100 + 4, fmt.Sprintf("员工%d", i*3+4), 26 + i, "人事部", "2023-04-05", 10000 + i*1000},
|
||
// }
|
||
// err := createExcelFile(filename, "员工数据", data)
|
||
// if err != nil {
|
||
// fmt.Printf("创建文件 %s 失败: %v\n", filename, err)
|
||
// return
|
||
// }
|
||
// fmt.Printf("创建文件: %s (共%d行数据)\n", filename, len(data))
|
||
// }
|
||
//}
|
||
//
|
||
//// 创建多sheet测试文件
|
||
//func createMultiSheetFile() error {
|
||
// file := excelize.NewFile()
|
||
// defer file.Close()
|
||
//
|
||
// // Sheet1
|
||
// file.NewSheet("部门数据")
|
||
// data1 := [][]interface{}{
|
||
// {"部门", "人数", "预算"},
|
||
// {"技术部", 50, 1000000},
|
||
// {"市场部", 30, 800000},
|
||
// {"销售部", 40, 900000},
|
||
// }
|
||
// writeDataToSheet(file, "部门数据", data1)
|
||
//
|
||
// // Sheet2
|
||
// file.NewSheet("项目数据")
|
||
// data2 := [][]interface{}{
|
||
// {"项目", "负责人", "进度", "预算"},
|
||
// {"项目A", "张三", "80%", 500000},
|
||
// {"项目B", "李四", "60%", 300000},
|
||
// {"项目C", "王五", "90%", 400000},
|
||
// }
|
||
// writeDataToSheet(file, "项目数据", data2)
|
||
//
|
||
// // Sheet3
|
||
// file.NewSheet("财务数据")
|
||
// data3 := [][]interface{}{
|
||
// {"月份", "收入", "支出", "利润"},
|
||
// {"1月", 500000, 300000, 200000},
|
||
// {"2月", 550000, 320000, 230000},
|
||
// {"3月", 600000, 350000, 250000},
|
||
// }
|
||
// writeDataToSheet(file, "财务数据", data3)
|
||
//
|
||
// // 删除默认的Sheet1
|
||
// file.DeleteSheet("Sheet1")
|
||
//
|
||
// return file.SaveAs("multi_sheet.xlsx")
|
||
//}
|
||
//
|
||
//// 写入数据到sheet
|
||
//func writeDataToSheet(file *excelize.File, sheet string, data [][]interface{}) {
|
||
// for rowIndex, row := range data {
|
||
// for colIndex, value := range row {
|
||
// cell, _ := excelize.CoordinatesToCellName(colIndex+1, rowIndex+1)
|
||
// file.SetCellValue(sheet, cell, value)
|
||
// }
|
||
// }
|
||
//}
|
||
//
|
||
//// 创建Excel文件
|
||
//func createExcelFile(filename, sheet string, data [][]interface{}) error {
|
||
// file := excelize.NewFile()
|
||
// defer file.Close()
|
||
//
|
||
// // 删除默认的Sheet1
|
||
// file.DeleteSheet("Sheet1")
|
||
//
|
||
// // 创建新sheet
|
||
// sheetIndex, err := file.NewSheet(sheet)
|
||
// if err != nil {
|
||
// return err
|
||
// }
|
||
// file.SetActiveSheet(sheetIndex)
|
||
//
|
||
// // 写入数据
|
||
// for rowIndex, row := range data {
|
||
// for colIndex, value := range row {
|
||
// cell, _ := excelize.CoordinatesToCellName(colIndex+1, rowIndex+1)
|
||
// file.SetCellValue(sheet, cell, value)
|
||
// }
|
||
// }
|
||
//
|
||
// return file.SaveAs(filename)
|
||
//}
|
||
//
|
||
//// 清理测试文件
|
||
//func cleanupTestFiles() {
|
||
// filesToClean := []string{
|
||
// "data1.xlsx",
|
||
// "data2.xlsx",
|
||
// "data3.xlsx",
|
||
// "merged_with_info.xlsx",
|
||
// "merged_parallel.xlsx",
|
||
// "merged_by_column.xlsx",
|
||
// "merged_specific.xlsx",
|
||
// "multi_sheet.xlsx",
|
||
// "merged_sheets.xlsx",
|
||
// }
|
||
//
|
||
// for _, file := range filesToClean {
|
||
// if _, err := os.Stat(file); err == nil {
|
||
// os.Remove(file)
|
||
// fmt.Printf("删除: %s\n", file)
|
||
// }
|
||
// }
|
||
//}
|