Merge pull request #469 from zaddok/patch-1

Fix potential memory leak
2019-08-10 00:07:09 +08:00 · 2019-08-10 00:07:09 +08:00 · 23c8d1ec8a
parent e6b7ac9a03 5107928892
commit 23c8d1ec8a
1 changed files with 2 additions and 0 deletions
--- a/file.go
+++ b/file.go
@ -109,10 +109,12 @@ func (f *File) WriteToBuffer() (*bytes.Buffer, error) {
 	for path, content := range f.XLSX {
 		fi, err := zw.Create(path)
 		if err != nil {
+			zw.Close()
 			return buf, err
 		}
 		_, err = fi.Write(content)
 		if err != nil {
+			zw.Close()
 			return buf, err
 		}
 	}