self-tools
/
excel-export
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

调整文件操作位置

This commit is contained in:
Mr.Li 2023-04-06 21:04:01 +08:00
parent ed692073cf
commit ec94cb1377
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
biz/export/excel_exporter.go
View File

@ -31,7 +31,7 @@ func (ee *ExcelExporter) Export(sql, pk string) error {
} }
ee.count = len(data.Data) ee.count = len(data.Data)
//fmt.Printf("Excel Exporter.Excel, got %v data\n", len(data)) //fmt.Printf("Excel Exporter.Excel, got %v data\n", len(data))
ee.file.Open() //ee.file.Open()
ee.file.WriteTitle(data.Title) ee.file.WriteTitle(data.Title)
var last interface{} var last interface{}
for _, val := range data.Data { for _, val := range data.Data {
@ -43,7 +43,7 @@ func (ee *ExcelExporter) Export(sql, pk string) error {
ee.last = row[getPkIndex(data.Title, pk)] ee.last = row[getPkIndex(data.Title, pk)]
} }
ee.file.Close() //ee.file.Close()
return nil return nil
} }

2
cmd/cmd/export.go
View File

@ -58,6 +58,7 @@ func JobHandler(job config.Job, d export.DataFetcher, params map[string]interfac
func TaskExport(d export.DataFetcher, t config.Task, params map[string]interface{}, f export.FileAdapter, batch int) error { func TaskExport(d export.DataFetcher, t config.Task, params map[string]interface{}, f export.FileAdapter, batch int) error {
//todo 最多分1000个批次进行处理 //todo 最多分1000个批次进行处理
f.Open()
for i := 0; i < 1000; i++ { for i := 0; i < 1000; i++ {
sql := t.GetSql(params) sql := t.GetSql(params)
@ -74,5 +75,6 @@ func TaskExport(d export.DataFetcher, t config.Task, params map[string]interface
params["last"] = last params["last"] = last
time.Sleep(time.Microsecond * 30) time.Sleep(time.Microsecond * 30)
} }
f.Close()
return nil return nil
} }