清除临时目录文件信息

This commit is contained in:
Mr.Li 2023-04-21 17:30:42 +08:00
parent 30227e36bd
commit 4fb1310ce5
2 changed files with 10 additions and 2 deletions

View File

@ -162,5 +162,9 @@ func (e *Csv) mergeCsvToExcel(path string, max int, out string) error {
Reader{Path: path, Index: max},
Writer{File: out, Limit: 500000},
)
return m.Merge()
if err := m.Merge(); err != nil {
return err
}
return m.Clear()
}

View File

@ -156,7 +156,11 @@ func (m *Merge) Save() error {
return m.file.SaveAs(m.writer.GetFileName(m.fileIndex))
}
//GetFileName 获取文件名
func (m *Merge) Clear() error {
return os.RemoveAll(m.reader.Path)
}
// GetFileName 获取文件名
func (w *Writer) GetFileName(fileIndex int) string {
ex := regexp.MustCompile("(\\..*)")
name := ex.ReplaceAllFunc([]byte(w.File), func(b []byte) []byte {