记录耗时

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
 }
 

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)
 		}