From d7ffa9757604628b04b979d921a6d8671b757e92 Mon Sep 17 00:00:00 2001
From: "Mr.Li" <lhb2002@qq.com>
Date: Thu, 20 Apr 2023 16:06:55 +0800
Subject: [PATCH] =?UTF-8?q?=E8=AE=B0=E5=BD=95=E8=80=97=E6=97=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 biz/export/csv_exporter.go | 13 +++++++++----
 cmd/cmd/csv.go             |  2 ++
 2 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/biz/export/csv_exporter.go b/biz/export/csv_exporter.go
index 8041647..4ab1756 100644
--- a/biz/export/csv_exporter.go
+++ b/biz/export/csv_exporter.go
@@ -2,7 +2,9 @@ package export
 
 import (
 	"fmt"
+	"log"
 	"sync"
+	"time"
 )
 
 type CsvExporter struct {
@@ -41,6 +43,7 @@ func (ee *CsvExporter) Export(sql, pk string) error {
 	ee.count = len(data.Data)
 	//fmt.Printf("Excel Exporter.Excel, got %v data\n", len(data))
 	go func() {
+		begin := time.Now()
 		ee.file.Open()
 		ee.file.WriteTitle(data.Title)
 
@@ -50,11 +53,13 @@ func (ee *CsvExporter) Export(sql, pk string) error {
 		}
 		ee.file.Close()
 		ee.wg.Done()
+		end := time.Now()
+		log.Println("csv time:%s", end.Sub(begin).String())
 	}()
-
-	last := data.Data[ee.count-1]
-	ee.last = last[ee.getPkIndex(data.Title, pk)]
-
+	if ee.count > 0 {
+		last := data.Data[ee.count-1]
+		ee.last = last[ee.getPkIndex(data.Title, pk)]
+	}
 	return nil
 }
 
diff --git a/cmd/cmd/csv.go b/cmd/cmd/csv.go
index 5e4ebfb..3ad49fa 100644
--- a/cmd/cmd/csv.go
+++ b/cmd/cmd/csv.go
@@ -159,8 +159,10 @@ func (e *Csv) mergeCsv(path string, max int) error {
 }
 
 func (e *Csv) mergeCsvToExcel(path string, max int) error {
+	begin := time.Now()
 	f := excelize.NewFile()
 	defer func() {
+		log.Println("mergeCsvToExcel:耗时 %s", time.Now().Sub(begin).String())
 		if err := f.Close(); err != nil {
 			log.Println(err)
 		}