调整文件操作位置

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

View File

@ -31,7 +31,7 @@ func (ee *ExcelExporter) Export(sql, pk string) error {
}
ee.count = len(data.Data)
//fmt.Printf("Excel Exporter.Excel, got %v data\n", len(data))
ee.file.Open()
//ee.file.Open()
ee.file.WriteTitle(data.Title)
var last interface{}
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.file.Close()
//ee.file.Close()
return nil
}

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